add user row to request context

This commit is contained in:
sentriz
2019-04-18 13:37:29 +01:00
parent c6273c8c12
commit 0932b5c013
3 changed files with 37 additions and 24 deletions

View File

@@ -60,6 +60,25 @@ type Controller struct {
SStore *gormstore.Store
}
func (c *Controller) GetSetting(key string) string {
var setting db.Setting
c.DB.Where("key = ?", key).First(&setting)
return setting.Value
}
func (c *Controller) SetSetting(key, value string) {
c.DB.
Where(db.Setting{Key: key}).
Assign(db.Setting{Value: value}).
FirstOrCreate(&db.Setting{})
}
func (c *Controller) GetUserFromName(name string) *db.User {
var user db.User
c.DB.Where("name = ?", name).First(&user)
return &user
}
type templateData struct {
Flashes []interface{}
User *db.User
@@ -151,7 +170,7 @@ func renderTemplate(w http.ResponseWriter, r *http.Request,
}
data.Flashes = session.Flashes()
session.Save(r, w)
user, ok := session.Values["user"].(*db.User)
user, ok := r.Context().Value("user").(*db.User)
if ok {
data.User = user
}