通过会话变量在Golang layout.tpl中有条件地呈现HTML

Question

我使用Gorilla会话（通过negroni-sessions）将用户会话存储在cookie中。 我还将github.com/unrolled/render用于HTML模板渲染：

main.go：

    package main

    import (
        ...
        "github.com/codegangsta/negroni"
        "github.com/goincremental/negroni-sessions"
        "github.com/goincremental/negroni-sessions/cookiestore"
        "github.com/julienschmidt/httprouter"
        "github.com/unrolled/render"
        ...
    )

    func init() {
        ...
        ren = render.New(render.Options{
            Directory:     "templates",
            Layout:        "layout",
            Extensions:    []string{".html"},
            Funcs:         []template.FuncMap{TemplateHelpers}, 
            IsDevelopment: false,
        })
        ...
    }

    func main() {
        ...
        router := httprouter.New()
        router.GET("/", HomeHandler)

        // Add session store
        store := cookiestore.New([]byte("my password"))
        store.Options(sessions.Options{
            //MaxAge: 1200,
            Domain: "",
            Path:   "/",
        })

        n := negroni.New(
            negroni.NewRecovery(),
            sessions.Sessions("cssession", store),
            negroni.NewStatic(http.Dir("../static")), 
        )

        n.UseHandler(router)
        n.Run(":9000")

    }

正如您在上面看到的，我使用了layout.html主HTML模板，该模板在呈现任何页面（例如我的主页）时都包含在内：

    package main

    import (
        "html/template"
        "github.com/julienschmidt/httprouter"
    )

    func HomeHandler(w http.ResponseWriter, r *http.Request, p httprouter.Params) {

        var model = struct {
            CatalogPicks   []PromotionalModelList
            ClearanceItems []Model
        }{
            CatalogPicks:   GetCatalogPicks(),
            ClearanceItems: GetClearanceItems(),
        }

        ren.HTML(w, http.StatusOK, "home", model)
    }

在我的layout.html主HTML模板中，我想呈现一个管理菜单，但前提是当前用户是admin：

layout.html ：

<!doctype html>
<html>
  <head>
    <meta charset="utf-8">
      <title>{{ template "title" . }}</title>
        ...
  </head>

  <body>
    ...
    <!--Main Menu-->
      <nav class="menu">
         <ul class="catalog">
             <li class="has-submenu">
                {{ RenderMenuCategories }}
             </li>
             <li><a href="javascript:void(0)">Blog</a></li>
             <li><a href="javascript:void(0)">Company</a></li>

             {{ RenderAdminMenu }}

         </ul>
      </nav>

...

我的问题是上述模板帮助器函数RenderAdminMenu()无法访问HTTP Request对象，因此无法访问User会话对象来确定用户是否为admin。

我可以通过Homepage处理程序将User对象传递到模板上下文中，并使用if语句RenderAdminMenu（）函数，如下所示

{{ if .User.IsAdmin }}
   {{ RenderAdminMenu }}
{{ end }}

...但是由于我使用的是主模板，因此必须在网站上的每个网页上都执行此操作。 有没有更有效的方法？

我在想，也许有一种方法可以从包含Request详细信息的RenderAdminMenu() （或layout.html）中访问某种全局Context对象（就像在ASP.NET中一样）

Answer 1

您需要做一些事情来将它们联系在一起。 我将不显示完整的示例，因为它既冗长又可能与您的代码不匹配（您尚未发布）。 但是它将包含基本的构建块：如果遇到问题，请直接回答问题并提供代码段，您将获得更直接的答案:)

1. 在[登录]处理程序中编写一些中间件或逻辑，以在用户登录时将用户数据保存在会话中。用户ID，电子邮件和admin布尔值就足够了。 例如

    // In your login handler, once you've retrieved the user & // matched their password hash (scrypt, of course!) from the DB. session.Values["user"] = &youruserobject err := session.Save(r, w) if err != nil { // Throw a HTTP 500 } 

注意：请记住，如果要存储自己的类型，则需要按照大猩猩/会话文档进行gob.Register(&youruserobject{}) 。

2. 当您从会话中拉出一个类型时，编写一个助手来键入您的类型，例如

    var ErrInvalidUser= errors.New("invalid user stored in session") func GetUser(session *sessions.Session) (*User, error) { // You can make the map key a constant to avoid typos/errors user, ok := session.Values["user"].(*User) if !ok || user == nil { return nil, ErrInvalidUser } return user, nil } // Use it like this in a handler that serves user content session, err := store.Get("yoursessionname", r) if err != nil { // Throw a HTTP 500 } user, err := GetUser(session) if err != nil { // Re-direct back to the login page or // show a HTTP 403 Forbidden, etc. } 

3. 写一些东西来检查返回的用户是否是管理员：

    func IsAdmin(user *User) bool { if user.Admin == true && user.ID != "" && user.Email != "" { return true } return false } 

4. 将其传递给模板：

     err := template.Execute(w, "sometemplate.html", map[string]interface{}{ "admin": IsAdmin(user), "someotherdata": someStructWithData, } // In your template... {{ if .admin }}{{ template "admin_menu" }}{{ end }} 

另外，还要确保为会话cookie（阅读大猩猩文档）设置身份验证密钥（最好是加密密钥），并且还要通过HTTPS为您的网站设置Secure：true标志。

请记住，上述方法也得到了简化：如果在数据库中将用户标记为admin，则只要会话持续，应用程序将继续将其检测为管理员。 默认情况下，这可能是7天，因此，如果您处于危险的环境中，而管理员流失是一个真正的问题，那么为了确保安全，可能需要进行非常短的会话或在IsAdmin函数中访问数据库，这是值得的。 如果这是一个个人博客，而只是您，那就不多了。

---

补充 ：如果要将用户对象直接传递给模板，也可以这样做。 请注意，在处理程序/中间件中执行此操作比在模板逻辑中执行性能更高。 您还可以获得更多错误处理的灵活性，并且可以更早地“保释”选项-即，如果会话不包含任何内容，则可以触发HTTP 500错误，而不用渲染一半的模板，也不必在其中添加大量逻辑用于处理nil数据的模板。

您仍然需要将User对象（或等效对象）存储在会话中，然后从session.Values检索它，然后才能将其传递给模板。

 func GetUser(r *http.Request) *User {
        session, err := store.Get("yoursessionname", r)
        if err != nil {
            // Throw a HTTP 500
        }
        if user, ok := session.Values["user"].(*User); ok {
            return user
        }

        return nil
 }

 // In the handler itself
 err := template.Execute(w, "sometemplate.html", map[string]interface{}{
        "user": GetUser(r),
        "someotherdata": someStructWithData,
        }


 // In your template...
 {{ if .User.admin }}{{ template "admin_menu" }}{{ end }}

Answer 2

看来您毕竟无法从模板或模板帮助器函数访问Request上下文（因此我接受了上面的答案）。 我的解决方案是创建一个Page结构，将其作为每个模板的上下文传递。 它包含作为通用接口的Content ， User对象以及其他有用的参数：

//Page holds the model to be rendered for every HTTP handler.
type Page struct {
     MetaTitle     string
     User          User
     HeaderStyles  string
     HeaderScripts string
     FooterScripts string
     Content       interface{}
 }

 func (pg *Page) Init(r *http.Request) {

    if pg.MetaTitle == "" {
        pg.MetaTitle = "This is the default <title> content for the page!"
    }

    user, _ := GetUserFromSession(r)
    pg.User = *user

    if user.IsAdmin() {
        pg.HeaderStyles += `<link href="/css/libs/summernote/summernote.css" rel="stylesheet">`
        pg.FooterScripts += `<script src="/js/libs/summernote/summernote.min.js"></script>`
    }

}

通过Init方法，我只需要为页面指定Content ，就可以设置默认值并更轻松地使用Page struct：

package main

import (
    "html/template"
    "github.com/julienschmidt/httprouter"
)

func HomeHandler(w http.ResponseWriter, r *http.Request, p httprouter.Params) {

    var model = struct {
        CatalogPicks   []PromotionalModelList
        ClearanceItems []Model
    }{
        CatalogPicks:   GetCatalogPicks(),
        ClearanceItems: GetClearanceItems(),
    }

    pg := &Page{Content: model}
    pg.Init(r)
    ren.HTML(w, http.StatusOK, "home", pg)
}