heimoshuiyu/gonic
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

add some error handing for session getting

Browse Source
This commit is contained in:
sentriz
2020-07-16 23:00:21 +01:00
parent f34a0b48f9
commit ab0d15982a
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
server/ctrladmin/middleware.go
View File

@@ -2,6 +2,7 @@ package ctrladmin
import ( import (
"context" "context"
"fmt"
"net/http" "net/http"
"github.com/gorilla/sessions" "github.com/gorilla/sessions"
@@ -11,7 +12,11 @@ import (
func (c *Controller) WithSession(next http.Handler) http.Handler { func (c *Controller) WithSession(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
session, _ := c.sessDB.Get(r, "gonic") session, err := c.sessDB.Get(r, "gonic")
if err != nil {
http.Error(w, fmt.Sprintf("error getting session: %s", err), 500)
return
}
withSession := context.WithValue(r.Context(), CtxSession, session) withSession := context.WithValue(r.Context(), CtxSession, session)
next.ServeHTTP(w, r.WithContext(withSession)) next.ServeHTTP(w, r.WithContext(withSession))
}) })