27 lines
450 B
Go
27 lines
450 B
Go
package v1
|
|
|
|
import (
|
|
"quyun/v2/database/models"
|
|
"quyun/v2/pkg/consts"
|
|
|
|
"github.com/gofiber/fiber/v3"
|
|
)
|
|
|
|
func getUserID(ctx fiber.Ctx) int64 {
|
|
if u := ctx.Locals(consts.CtxKeyUser); u != nil {
|
|
if user, ok := u.(*models.User); ok {
|
|
return user.ID
|
|
}
|
|
}
|
|
return 0
|
|
}
|
|
|
|
func getTenantID(ctx fiber.Ctx) int64 {
|
|
if t := ctx.Locals(consts.CtxKeyTenant); t != nil {
|
|
if tenant, ok := t.(*models.Tenant); ok {
|
|
return tenant.ID
|
|
}
|
|
}
|
|
return 0
|
|
}
|