mirror of
https://github.com/bufanyun/hotgo.git
synced 2025-02-03 02:54:41 +08:00
50 lines
1.2 KiB
Go
50 lines
1.2 KiB
Go
// Package handler
|
|
// @Link https://github.com/bufanyun/hotgo
|
|
// @Copyright Copyright (c) 2023 HotGo CLI
|
|
// @Author Ms <133814250@qq.com>
|
|
// @License https://github.com/bufanyun/hotgo/blob/master/LICENSE
|
|
package handler
|
|
|
|
import (
|
|
"github.com/gogf/gf/v2/database/gdb"
|
|
"github.com/gogf/gf/v2/text/gstr"
|
|
"hotgo/internal/library/contexts"
|
|
"hotgo/utility/convert"
|
|
)
|
|
|
|
// FilterTenant 过滤多租户数据权限
|
|
// 根据部门类型识别当前租户、商户、用户身份，过滤只属于自己的数据
|
|
func FilterTenant(m *gdb.Model) *gdb.Model {
|
|
var (
|
|
needAuth bool
|
|
filterField string
|
|
fields = convert.EscapeFieldsToSlice(m.GetFieldsStr())
|
|
ctx = m.GetCtx()
|
|
)
|
|
|
|
// 租户
|
|
if contexts.IsTenantDept(ctx) && gstr.InArray(fields, "tenant_id") {
|
|
needAuth = true
|
|
filterField = "tenant_id"
|
|
}
|
|
|
|
// 商户
|
|
if contexts.IsMerchantDept(ctx) && gstr.InArray(fields, "merchant_id") {
|
|
needAuth = true
|
|
filterField = "merchant_id"
|
|
}
|
|
|
|
// 用户
|
|
if contexts.IsUserDept(ctx) && gstr.InArray(fields, "user_id") {
|
|
needAuth = true
|
|
filterField = "user_id"
|
|
}
|
|
|
|
if !needAuth {
|
|
return m
|
|
}
|
|
|
|
m = m.Where(filterField, contexts.GetUserId(ctx))
|
|
return m
|
|
}
|