签入脚本

internal/model/hooks/hooks.go

@@ -0,0 +1,157 @@
+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})
+}