kpt-tmr-group/http/middleware/bus.go

package middleware

import (
	"kpt-tmr-group/dep"

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

const KeyDep = "key:dep"

func WithDependency(s *dep.HttpDependency) gin.HandlerFunc {
	return func(c *gin.Context) {
		c.Set(KeyDep, s)
		c.Next()
	}
}

func Dependency(c *gin.Context) *dep.HttpDependency {
	return c.MustGet(KeyDep).(*dep.HttpDependency)
}