基于Golang与templ，根据用户登录状态切换导航栏的最优方案

问题背景 #

我用Golang搭建服务器，搭配Templ渲染首页和带“登录”按钮的导航栏，已实现会话认证功能，想了解根据用户登录状态展示不同导航栏的最佳方式。目前已有checkAuth中间件，最初思路是给所有路由绑定该中间件，从Context中获取isAuthenticated并传递给处理器，但觉得每次加载页面都这么做不够合理。

现有代码 #

路由代码

e.GET("/", handlers.HandleHome)

func HandleHome(c echo.Context) error {
    return Render(c, home.Index(c.Get("isAuthenticated").(bool)))
}

中间件代码

func checkAuthMiddleware(next echo.HandlerFunc) echo.HandlerFunc {
    return func(c echo.Context) error {
        session, _ := store.Get(c.Request(), "session")
        isAuthenticated := session.Values["authenticated"] == true
        if !isAuthenticated {
            return c.Redirect(http.StatusUnauthorized, "/")
        }
        c.Set("isAuthenticated", isAuthenticated)
        return next(c)
    }
}

Templ组件代码

home.tmpl

templ Index(isLoggedIn bool) {
    @layouts.Base(isLoggedIn) {

base.tmpl

templ Base(isLoggedIn bool) {
            @components.Navigation(isLoggedIn)
    }

navigation.tmpl

templ Navigation(isLoggedIn bool) {
    <nav>
    if isLoggedIn {
        <div>Welcome back!</div>
    } else {
        <input name="login" type="button" value="Log in"/>
    }
    </nav>
}

优化方案 #

1. 修正中间件逻辑，取消未认证强制跳转 #

现有中间件会在用户未认证时直接跳转，导致首页无法获取未登录状态。修改后只设置认证状态，不强制拦截：

func checkAuthMiddleware(next echo.HandlerFunc) echo.HandlerFunc {
    return func(c echo.Context) error {
        session, err := store.Get(c.Request(), "session")
        // 处理会话获取错误，默认标记为未认证
        if err != nil {
            c.Set("isAuthenticated", false)
            return next(c)
        }
        isAuthenticated := session.Values["authenticated"] == true
        c.Set("isAuthenticated", isAuthenticated)
        return next(c)
    }
}

2. 全局注册中间件，减少重复绑定 #

将中间件注册到Echo全局中间件列表，所有路由自动继承，无需逐个绑定：

e := echo.New()
// 全局注册，所有请求都会经过该中间件
e.Use(checkAuthMiddleware)

e.GET("/", handlers.HandleHome)
// 其他路由定义...

3. 安全获取认证状态，避免panic #

处理器中使用安全类型断言，防止类型不匹配导致的程序崩溃：

func HandleHome(c echo.Context) error {
    isAuthenticated, ok := c.Get("isAuthenticated").(bool)
    // 类型断言失败时默认设为未认证
    if !ok {
        isAuthenticated = false
    }
    return Render(c, home.Index(isAuthenticated))
}

4. 保持Templ组件的层级传参逻辑 #

你当前的Templ组件结构已经很合理：通过home.tmpl传递状态给Base布局，再由Base传递给Navigation组件，最终根据状态渲染不同导航内容。这种组件化的参数传递方式符合Templ的设计理念，无需改动。

进阶优化：封装认证状态获取函数 #

如果多个处理器都需要获取认证状态，可以封装工具函数简化代码：

// 封装全局工具函数，统一处理认证状态获取
func GetAuthStatus(c echo.Context) bool {
    status, ok := c.Get("isAuthenticated").(bool)
    return ok && status
}

使用时直接调用：

func HandleHome(c echo.Context) error {
    return Render(c, home.Index(GetAuthStatus(c)))
}

内容的提问来源于stack exchange，提问作者spotnag