modelRT/middleware/token.go

// Package middleware define gin framework middlewares
package middleware

import (
	"context"

	"modelRT/constants"

	"github.com/gin-gonic/gin"
)

// SetTokenMiddleware define a middleware for set token in context
func SetTokenMiddleware(clientToken string) gin.HandlerFunc {
	return func(c *gin.Context) {
		c.Set(constants.ClientTokenContextName, clientToken)
		requestCtx := context.WithValue(c.Request.Context(), constants.CtxKeyClientToken, clientToken)
		c.Request = c.Request.WithContext(requestCtx)
		c.Next()
	}
}