tbs

handler/index_and_404.go

@@ -32,7 +32,7 @@ func (handler IndexImpl) Handle(router *http.ServeMux) {
 
 func (handler IndexImpl) handleIndexAnd404() http.HandlerFunc {
 	return func(w http.ResponseWriter, r *http.Request) {
-		session := r.Context().Value(middleware.SessionKey).(*service.Session)
+		session := middleware.GetSession(r)
 		var user *service.User
 		if session != nil {
 			user = session.User