17 lines
271 B
Go
17 lines
271 B
Go
package middleware
|
|
|
|
import (
|
|
"one-api/model"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func Distribute() func(c *gin.Context) {
|
|
return func(c *gin.Context) {
|
|
userId := c.GetInt("id")
|
|
userGroup, _ := model.CacheGetUserGroup(userId)
|
|
c.Set("group", userGroup)
|
|
c.Next()
|
|
}
|
|
}
|