199 lines
7.7 KiB
Go
199 lines
7.7 KiB
Go
// +======================================================================
|
|
// | EasyGoAdmin 企业数字化开发平台 [ 赋能开发者，助力企业发展 ]
|
|
// +----------------------------------------------------------------------
|
|
// | Copyright (c) 2019-2026 半城风雨团队 版权所有
|
|
// +----------------------------------------------------------------------
|
|
// | 开源协议: Apache-2.0
|
|
// | 重要声明: 未经书面许可，不得移除或修改本框架相关版权标识。
|
|
// +----------------------------------------------------------------------
|
|
// | 官方网站: https://www.easygoadmin.vip
|
|
// +----------------------------------------------------------------------
|
|
// | 研发团队: 小蚂蚁云团队 荣誉出品
|
|
// +======================================================================
|
|
// | 版权和免责声明:
|
|
// | 一、知识产权声明
|
|
// | 本团队对该软件框架产品（以下简称“本产品”）拥有完整知识产权，包括但不限于商标权、专利权、
|
|
// | 著作权、商业秘密等。前述权利均受中华人民共和国及相关国际法律法规保护。未经本团队书面授权，
|
|
// | 任何个人或单位不得基于本产品本身申请任何形式的知识产权。
|
|
// | 二、使用限制与禁止条款
|
|
// | 您理解并同意，禁止将本产品用于以下任何行为：
|
|
// | 1. 任何违反中华人民共和国法律法规的用途；
|
|
// | 2. 任何侵害他人合法权益（包括但不限于人身权、财产权、知识产权等）的行为；
|
|
// | 3. 任何危害国家安全、公共利益及社会道德风尚的行为。
|
|
// | 三、责任豁免
|
|
// | 因使用本产品而产生的任何损失或法律纠纷（包括但不限于直接损失、间接损失、附带损失、衍生损失
|
|
// | 以及因违约、侵权等引发的责任），本团队均不承担任何法律责任。若因违规使用本产品导致任何第三
|
|
// | 方提起索赔或行政机关进行调查，您应独立承担全部责任。
|
|
// | 四、违约与法律责任
|
|
// | 如发现任何违反本声明的行为，本团队有权：
|
|
// | 1. 无条件配合公安机关及其他有权机关的调查取证工作；
|
|
// | 2. 依法追究相关方的民事责任，并要求赔偿全部损失；
|
|
// | 3. 保留进一步采取刑事控告等法律措施的权利。
|
|
// | 五、许可范围说明
|
|
// | 1、被授权方应视本产品源代码为核心机密信息，负有最高标准的保密与保管义务。禁止以任何形式（包
|
|
// | 括但不限于直接或间接的分享、复制、传播、分发、转售、出租、开源、反向工程、反编译、反汇编或通
|
|
// | 过网络服务提供给任何第三方）泄露、披露或处置源代码的全部或部分内容。任何未经授权的复制、使用、
|
|
// | 传播行为均构成对本团队知识产权的严重侵权，本团队有权依法追究其全部法律责任，并要求赔偿因此造
|
|
// | 成的一切经济损失及商誉损害。
|
|
// | 2、本产品仅限于公司或个人在法律允许范围内，用于内部且合规的软件产品研发。具体权利义务以《软件
|
|
// | 许可协议》附件为准。
|
|
// +======================================================================
|
|
|
|
package hooks
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"reflect"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/gogf/gf/v2/database/gdb"
|
|
|
|
"easygoadmin/internal/consts"
|
|
)
|
|
|
|
// ============================== 全局模型钩子 ==============================
|
|
//
|
|
// 全局钩子是架构红线的执行点：
|
|
// - Select、Update、Delete 的条件过滤在 dao.Ctx 覆写中注入（见 ApplyCtx，进入 SQL 构建前最可靠）
|
|
// - Insert：主键交给数据库自增，自动补齐审计字段（create_user/create_time/del_flag）与租户ID
|
|
// - Update：自动为审计字段（update_user/update_time）盖章
|
|
// - Delete：物理删除一律改写为软删除 UPDATE del_flag=1
|
|
//
|
|
// 所有表均含统一字段 id/create_user/create_time/update_user/update_time/del_flag。
|
|
|
|
// tenantTables 含 tenant_id 列的租户级表
|
|
//
|
|
// 与初始化脚本保持一致，共 9 张
|
|
var tenantTables = map[string]bool{
|
|
"goframe_user": true,
|
|
"goframe_role": true,
|
|
"goframe_dept": true,
|
|
"goframe_level": true,
|
|
"goframe_position": true,
|
|
"goframe_article": true,
|
|
"goframe_category": true,
|
|
"goframe_notice": true,
|
|
"goframe_link": true,
|
|
}
|
|
|
|
// IsTenantTable 判断是否为租户级表
|
|
func IsTenantTable(table string) bool { return tenantTables[table] }
|
|
|
|
// Handler 返回全局钩子
|
|
// 由 dao 扩展模板挂载
|
|
func Handler() gdb.HookHandler {
|
|
return gdb.HookHandler{
|
|
Insert: insertHook,
|
|
Update: updateHook,
|
|
Delete: deleteHook,
|
|
}
|
|
}
|
|
|
|
// ApplyCtx 注入软删除过滤与租户隔离条件
|
|
//
|
|
// 由 dao 扩展模板的 Ctx 覆写方法调用，在模型层（SQL 构建之前）注入：
|
|
//
|
|
// 所有表：del_flag = 0
|
|
// 租户级表：平台级（超管）不过滤，全量可见；租户用户追加 tenant_id IN (0, 自身租户)
|
|
func ApplyCtx(ctx context.Context, m *gdb.Model, table string) *gdb.Model {
|
|
m = m.Where("del_flag", 0)
|
|
if tenantTables[table] && !consts.IsPlatform(ctx) {
|
|
m = m.WhereIn("tenant_id", []int{0, consts.CtxTenantID(ctx)})
|
|
}
|
|
return m
|
|
}
|
|
|
|
// auditUser 获取审计操作人
|
|
//
|
|
// 优先取用户名，保证租户内稳定可追溯
|
|
func auditUser(ctx context.Context) string {
|
|
if u := consts.GetUsername(ctx); u != "" {
|
|
return u
|
|
}
|
|
return consts.GetRealname(ctx)
|
|
}
|
|
|
|
// isBlank 判断插入值是否为空
|
|
//
|
|
// 为空时由钩子补齐
|
|
func isBlank(v any) bool {
|
|
if v == nil {
|
|
return true
|
|
}
|
|
if s, ok := v.(string); ok {
|
|
return s == ""
|
|
}
|
|
rv := reflect.ValueOf(v)
|
|
switch rv.Kind() {
|
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64,
|
|
reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64,
|
|
reflect.Float32, reflect.Float64:
|
|
return rv.IsZero()
|
|
}
|
|
return false
|
|
}
|
|
|
|
// hookTable 归一化钩子回调中的表名
|
|
//
|
|
// gdb 传入的表名带反引号，如 `goframe_user`
|
|
func hookTable(table string) string {
|
|
return strings.Trim(table, "`")
|
|
}
|
|
|
|
// insertHook 插入钩子
|
|
//
|
|
// 自动补齐主键、审计字段与租户ID
|
|
func insertHook(ctx context.Context, in *gdb.HookInsertInput) (sql.Result, error) {
|
|
now := time.Now()
|
|
table := hookTable(in.Table)
|
|
for _, item := range in.Data {
|
|
// 主键交由数据库自增（id 统一 int，不使用雪花算法）
|
|
if isBlank(item["del_flag"]) {
|
|
item["del_flag"] = 0
|
|
}
|
|
if isBlank(item["create_time"]) {
|
|
item["create_time"] = now
|
|
}
|
|
if isBlank(item["create_user"]) {
|
|
item["create_user"] = auditUser(ctx)
|
|
}
|
|
// 租户级表：
|
|
// - 租户用户未指定租户ID时自动携带自身（零值视为未指定，防跨租户逃逸）
|
|
// - 超管未指定租户ID时代入默认租户（tenant.defaultId）；
|
|
// 显式传 0 可创建平台共享行，显式传具体租户从显式值
|
|
if tenantTables[table] {
|
|
v, ok := item["tenant_id"]
|
|
if !consts.IsPlatform(ctx) {
|
|
if !ok || isBlank(v) {
|
|
item["tenant_id"] = consts.CtxTenantID(ctx)
|
|
}
|
|
} else if !ok || v == nil {
|
|
item["tenant_id"] = consts.DefaultTenantID(ctx)
|
|
}
|
|
}
|
|
}
|
|
return in.Next(ctx)
|
|
}
|
|
|
|
// updateHook 更新钩子
|
|
//
|
|
// 自动为审计字段盖章
|
|
func updateHook(ctx context.Context, in *gdb.HookUpdateInput) (sql.Result, error) {
|
|
if data, ok := in.Data.(map[string]any); ok {
|
|
data["update_time"] = time.Now()
|
|
data["update_user"] = auditUser(ctx)
|
|
}
|
|
return in.Next(ctx)
|
|
}
|
|
|
|
// deleteHook 删除钩子
|
|
//
|
|
// 将物理删除改写为软删除，为架构红线；
|
|
// 注意：gdb 的 Model.Update 不接收 ctx，模型已携带上下文设置，
|
|
// 传入 ctx 会被误当作数据导致匹配字段失败
|
|
func deleteHook(ctx context.Context, in *gdb.HookDeleteInput) (sql.Result, error) {
|
|
return in.Model.Update(map[string]any{"del_flag": 1})
|
|
}
|