mirror of
https://github.com/songquanpeng/one-api.git
synced 2025-09-29 22:56:39 +08:00
18 lines
272 B
Go
18 lines
272 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()
|
|
}
|
|
}
|