Chore: Remove untyped data map from macaron context (#39077)

pkg/middleware/auth.go

@@ -111,9 +111,8 @@ func Auth(options *AuthOptions) macaron.Handler {
 		requireLogin := !c.AllowAnonymous || forceLogin || options.ReqNoAnonynmous
 
 		if !c.IsSignedIn && options.ReqSignedIn && requireLogin {
-			lookupTokenErr, hasTokenErr := c.Data["lookupTokenErr"].(error)
 			var revokedErr *models.TokenRevokedError
-			if hasTokenErr && errors.As(lookupTokenErr, &revokedErr) {
+			if errors.As(c.LookupTokenErr, &revokedErr) {
 				tokenRevoked(c, revokedErr)
 				return
 			}

pkg/middleware/logger.go

@@ -19,9 +19,8 @@ import (
 	"net/http"
 	"time"
 
-	"github.com/grafana/grafana/pkg/models"
+	"github.com/grafana/grafana/pkg/services/contexthandler"
 	"github.com/grafana/grafana/pkg/setting"
-	"github.com/prometheus/client_golang/prometheus"
 	cw "github.com/weaveworks/common/middleware"
 	"gopkg.in/macaron.v1"
 )
@@ -29,16 +28,15 @@ import (
 func Logger(cfg *setting.Cfg) macaron.Handler {
 	return func(res http.ResponseWriter, req *http.Request, c *macaron.Context) {
 		start := time.Now()
-		c.Data["perfmon.start"] = start
 
 		rw := res.(macaron.ResponseWriter)
 		c.Next()
 
 		timeTaken := time.Since(start) / time.Millisecond
 
-		if timer, ok := c.Data["perfmon.timer"]; ok {
-			timerTyped := timer.(prometheus.Summary)
-			timerTyped.Observe(float64(timeTaken))
+		ctx := contexthandler.FromContext(c.Req.Context())
+		if ctx != nil && ctx.PerfmonTimer != nil {
+			ctx.PerfmonTimer.Observe(float64(timeTaken))
 		}
 
 		status := rw.Status()
@@ -48,9 +46,7 @@ func Logger(cfg *setting.Cfg) macaron.Handler {
 			}
 		}
 
-		if ctx, ok := c.Data["ctx"]; ok {
-			ctxTyped := ctx.(*models.ReqContext)
-
+		if ctx != nil {
 			logParams := []interface{}{
 				"method", req.Method,
 				"path", req.URL.Path,
@@ -61,15 +57,15 @@ func Logger(cfg *setting.Cfg) macaron.Handler {
 				"referer", req.Referer(),
 			}
 
-			traceID, exist := cw.ExtractTraceID(ctxTyped.Req.Context())
+			traceID, exist := cw.ExtractTraceID(ctx.Req.Context())
 			if exist {
 				logParams = append(logParams, "traceID", traceID)
 			}
 
 			if status >= 500 {
-				ctxTyped.Logger.Error("Request Completed", logParams...)
+				ctx.Logger.Error("Request Completed", logParams...)
 			} else {
-				ctxTyped.Logger.Info("Request Completed", logParams...)
+				ctx.Logger.Info("Request Completed", logParams...)
 			}
 		}
 	}

pkg/middleware/org_redirect.go

@@ -8,6 +8,7 @@ import (
 
 	"github.com/grafana/grafana/pkg/bus"
 	"github.com/grafana/grafana/pkg/models"
+	"github.com/grafana/grafana/pkg/services/contexthandler"
 	"github.com/grafana/grafana/pkg/setting"
 	"gopkg.in/macaron.v1"
 )
@@ -23,8 +24,8 @@ func OrgRedirect(cfg *setting.Cfg) macaron.Handler {
 			return
 		}
 
-		ctx, ok := c.Data["ctx"].(*models.ReqContext)
-		if !ok || !ctx.IsSignedIn {
+		ctx := contexthandler.FromContext(req.Context())
+		if !ctx.IsSignedIn {
 			return
 		}
 

pkg/middleware/recovery.go

@@ -26,7 +26,7 @@ import (
 	"gopkg.in/macaron.v1"
 
 	"github.com/grafana/grafana/pkg/infra/log"
-	"github.com/grafana/grafana/pkg/models"
+	"github.com/grafana/grafana/pkg/services/contexthandler"
 	"github.com/grafana/grafana/pkg/setting"
 )
 
@@ -109,9 +109,9 @@ func Recovery(cfg *setting.Cfg) macaron.Handler {
 			if r := recover(); r != nil {
 				panicLogger := log.Root
 				// try to get request logger
-				if ctx, ok := c.Data["ctx"]; ok {
-					ctxTyped := ctx.(*models.ReqContext)
-					panicLogger = ctxTyped.Logger
+				ctx := contexthandler.FromContext(c.Req.Context())
+				if ctx != nil {
+					panicLogger = ctx.Logger
 				}
 
 				if err, ok := r.(error); ok {
@@ -132,33 +132,34 @@ func Recovery(cfg *setting.Cfg) macaron.Handler {
 					return
 				}
 
-				c.Data["Title"] = "Server Error"
-				c.Data["AppSubUrl"] = cfg.AppSubURL
-				c.Data["Theme"] = cfg.DefaultTheme
+				data := struct {
+					Title     string
+					AppSubUrl string
+					Theme     string
+					ErrorMsg  string
+				}{"Server Error", cfg.AppSubURL, cfg.DefaultTheme, ""}
 
 				if setting.Env == setting.Dev {
 					if err, ok := r.(error); ok {
-						c.Data["Title"] = err.Error()
+						data.Title = err.Error()
 					}
 
-					c.Data["ErrorMsg"] = string(stack)
+					data.ErrorMsg = string(stack)
 				}
 
-				ctx, ok := c.Data["ctx"].(*models.ReqContext)
-
-				if ok && ctx.IsApiRequest() {
+				if ctx != nil && ctx.IsApiRequest() {
 					resp := make(map[string]interface{})
 					resp["message"] = "Internal Server Error - Check the Grafana server logs for the detailed error message."
 
-					if c.Data["ErrorMsg"] != nil {
-						resp["error"] = fmt.Sprintf("%v - %v", c.Data["Title"], c.Data["ErrorMsg"])
+					if data.ErrorMsg != "" {
+						resp["error"] = fmt.Sprintf("%v - %v", data.Title, data.ErrorMsg)
 					} else {
-						resp["error"] = c.Data["Title"]
+						resp["error"] = data.Title
 					}
 
 					c.JSON(500, resp)
 				} else {
-					c.HTML(500, cfg.ErrTemplateName, c.Data)
+					c.HTML(500, cfg.ErrTemplateName, data)
 				}
 			}
 		}()