sysUserModel.go 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465
  1. package user
  2. import (
  3. "context"
  4. "database/sql"
  5. "errors"
  6. "fmt"
  7. "strings"
  8. "time"
  9. "github.com/zeromicro/go-zero/core/logx"
  10. "github.com/zeromicro/go-zero/core/stores/cache"
  11. "github.com/zeromicro/go-zero/core/stores/sqlx"
  12. )
  13. var ErrUpdateConflict = errors.New("update conflict: data has been modified by another operation")
  14. // ErrTokenVersionMismatch 表示令牌版本与数据库当前版本不一致,刷新令牌失败。
  15. // 典型场景:refreshToken rotation 并发到达 —— 只有持有当前 tokenVersion 的那一次能原子递增成功,
  16. // 其余全部返回该错误,防止两个请求都"换到"新令牌(导致会话劫持)。
  17. var ErrTokenVersionMismatch = errors.New("token version mismatch")
  18. var _ SysUserModel = (*customSysUserModel)(nil)
  19. type (
  20. SysUserModel interface {
  21. sysUserModel
  22. FindListByPage(ctx context.Context, page, pageSize int64) ([]*SysUser, int64, error)
  23. FindListByProductMembers(ctx context.Context, productCode string, page, pageSize int64) ([]*SysUser, map[int64]string, int64, error)
  24. FindByIds(ctx context.Context, ids []int64) ([]*SysUser, error)
  25. FindIdsByDeptId(ctx context.Context, deptId int64) ([]int64, error)
  26. // FindIdsByDeptIdForShareTx 在调用方事务里取目标 deptId 下所有用户 id,并对命中行加 S 锁
  27. // (LOCK IN SHARE MODE)。用于 UpdateDept 收窄(DEV→NORMAL / DEV 部门 Enabled→Disabled /
  28. // NORMAL 部门 Enabled→Disabled)后需要批量递增 tokenVersion 时,阻塞并发
  29. // UpdateProfileWithTx 对同一批 sys_user 行的 X 锁,避免"枚举 userIds 的瞬间有行刚被挪出
  30. // 本部门 / 新行刚被挪入"导致的漏吊销 / 多吊销(审计 L-R16-2)。
  31. // session==nil 返回错误。
  32. FindIdsByDeptIdForShareTx(ctx context.Context, session sqlx.Session, deptId int64) ([]int64, error)
  33. // UpdateProfile 更新用户资料字段(昵称 / 邮箱 / 手机 / 备注 / 部门 / 状态),username 仅用于
  34. // 构造旧缓存键 `cacheSysUserUsernamePrefix` 做失效,**不会**被写入 SET 子句。若未来确实需要
  35. // 修改 username,请独立实现 `UpdateUsernameTx`:
  36. // ① 同事务内做 old/new 两份 UsernameKey 的失效;
  37. // ② 捕获 1062 (UNIQUE 冲突) → response.ErrConflict,不得混进本方法的签名(审计 L-R11-3)。
  38. UpdateProfile(ctx context.Context, id int64, username string, nickname, email, phone, remark string, deptId, newStatus int64, statusChanged bool, expectedUpdateTime int64) error
  39. // UpdateProfileWithTx 与 UpdateProfile 的 SET 子句等价,但 UPDATE 执行在调用方传入的事务里;
  40. // 用于"改 deptId → 同事务内先 FOR SHARE sys_dept 目标行"修复 DeleteDept vs UpdateUser
  41. // 的 write skew(审计 M-R11-3)。
  42. // 审计 L-R12-1:本方法**不再**自己做缓存失效——go-zero `m.ExecCtx` 的 DelCache 会在闭包
  43. // 成功返回时立即发生,这里"成功"只代表 session.ExecCtx 收到了行数,事务仍在进行中,
  44. // 并发 FindOne 会把尚未提交的旧值重新灌回 sysUser 低层缓存造成 stale。调用方**必须**在
  45. // TransactCtx 返回(事务 commit)成功之后调用 InvalidateProfileCache(id, username) 做
  46. // post-commit 失效;同模式对任何 `*WithTx` 方法都成立。
  47. UpdateProfileWithTx(ctx context.Context, session sqlx.Session, id int64, username string, nickname, email, phone, remark string, deptId, newStatus int64, statusChanged bool, expectedUpdateTime int64) error
  48. // InvalidateProfileCache 失效 sysUser 的 id / username 两把低层缓存键。仅应在事务
  49. // commit 成功后由调用方显式调用(见 UpdateProfileWithTx 说明)。best-effort:失效失败
  50. // 只留日志,最终由 TTL 兜底,不影响事务已完成的业务结果(审计 L-R12-1)。
  51. InvalidateProfileCache(ctx context.Context, id int64, username string)
  52. // UpdatePassword 审计 H-R11-1:expectedUpdateTime 必须由调用方用**外层校验旧密码时拿到的
  53. // 那一份 updateTime** 显式透传;禁止函数内部再 FindOne 自对齐乐观锁,否则内层 CAS 等于退化
  54. // 为 last-write-wins,被"旧会话 + 知道旧密码"的攻击者可以把管理员紧急改过的新密码盖回。
  55. // username 仅用于构造 `cacheSysUserUsernamePrefix` 缓存键失效(sysUser.username 唯一,本函数
  56. // 不会修改它)。
  57. UpdatePassword(ctx context.Context, id int64, username string, password string, mustChangePassword, expectedUpdateTime int64) error
  58. // UpdateStatus 审计 M-R11-2:username 由调用方透传,避免仅为构造缓存键而多打一次 FindOne。
  59. UpdateStatus(ctx context.Context, id int64, username string, status int64, expectedUpdateTime int64) error
  60. // IncrementTokenVersion 审计 M-R11-2:username 由调用方透传,避免仅为构造缓存键而多打一次 FindOne。
  61. IncrementTokenVersion(ctx context.Context, id int64, username string) (int64, error)
  62. IncrementTokenVersionIfMatch(ctx context.Context, id int64, username string, expected int64) (int64, error)
  63. // IncrementTokenVersionWithTx 在调用方提供的事务内递增 sys_user.tokenVersion,供 UpdateMember
  64. // 降级 / 禁用等业务把旧 access/refresh token 立刻打到 middleware 的 `claims.TokenVersion !=
  65. // ud.TokenVersion` / refresh CAS 的拒绝分支(审计 M-R15-1 方案 A)。
  66. //
  67. // 与非事务版本 IncrementTokenVersion 的差异:
  68. // - 不在方法内做任何缓存失效;tx 成功提交后由调用方走 InvalidateProfileCache(id, username)
  69. // + UserDetailsLoader.Del/CleanByProduct 的 post-commit 链路(对齐 UpdateProfileWithTx
  70. // 的 L-R12-1 契约;若事务回滚,调用方**不要**触发失效,避免"操作失败但用户被踢下线");
  71. // - session==nil 直接报错——非事务场景请改走 IncrementTokenVersion。
  72. // 返回新的 tokenVersion(SELECT LAST_INSERT_ID())供调用方 log / forensic 比对。
  73. IncrementTokenVersionWithTx(ctx context.Context, session sqlx.Session, id int64) (int64, error)
  74. // BatchIncrementTokenVersionWithTx 批量递增一组 userIds 的 tokenVersion,供 UpdateProduct
  75. // 禁用时吊销该产品全体成员已签发的 access/refresh(审计 L-R15-3)。
  76. //
  77. // 契约:
  78. // - 不做缓存失效;tx 提交后调用方按 (id, username) 对逐个 InvalidateProfileCache,UD 聚合
  79. // 缓存通过 UserDetailsLoader.CleanByProduct 失效;
  80. // - 调用方负责对 ids 去重并控制长度,空切片直接返回 nil;
  81. // - session==nil 返回错误。
  82. BatchIncrementTokenVersionWithTx(ctx context.Context, session sqlx.Session, ids []int64) error
  83. // UpdateSelfInfo 更新用户自身安全字段(昵称/头像/邮箱/手机),不涉及 deptId/status/tokenVersion。
  84. // username 仅用于构造缓存键失效。avatar 使用 sql.NullString 语义写入(空串置 NULL)。
  85. UpdateSelfInfo(ctx context.Context, id int64, username string, nickname, avatar, email, phone string, expectedUpdateTime int64) error
  86. }
  87. customSysUserModel struct {
  88. *defaultSysUserModel
  89. }
  90. )
  91. func NewSysUserModel(conn sqlx.SqlConn, c cache.CacheConf, cachePrefix string, opts ...cache.Option) SysUserModel {
  92. return &customSysUserModel{
  93. defaultSysUserModel: newSysUserModel(conn, c, cachePrefix, opts...),
  94. }
  95. }
  96. func (m *customSysUserModel) FindListByPage(ctx context.Context, page, pageSize int64) ([]*SysUser, int64, error) {
  97. var total int64
  98. countQuery := fmt.Sprintf("SELECT COUNT(*) FROM %s", m.table)
  99. if err := m.QueryRowNoCacheCtx(ctx, &total, countQuery); err != nil {
  100. return nil, 0, err
  101. }
  102. var list []*SysUser
  103. query := fmt.Sprintf("SELECT %s FROM %s ORDER BY id DESC LIMIT ?,?", sysUserRows, m.table)
  104. if err := m.QueryRowsNoCacheCtx(ctx, &list, query, (page-1)*pageSize, pageSize); err != nil {
  105. return nil, 0, err
  106. }
  107. return list, total, nil
  108. }
  109. type UserWithMemberType struct {
  110. SysUser
  111. MemberType string `db:"memberType"`
  112. }
  113. func (m *customSysUserModel) FindListByProductMembers(ctx context.Context, productCode string, page, pageSize int64) ([]*SysUser, map[int64]string, int64, error) {
  114. memberTable := "`sys_product_member`"
  115. var total int64
  116. countQuery := fmt.Sprintf("SELECT COUNT(*) FROM %s u INNER JOIN %s pm ON u.`id` = pm.`userId` WHERE pm.`productCode` = ?", m.table, memberTable)
  117. if err := m.QueryRowNoCacheCtx(ctx, &total, countQuery, productCode); err != nil {
  118. return nil, nil, 0, err
  119. }
  120. var list []*UserWithMemberType
  121. fields := strings.Join(sysUserFieldNames, ",u.")
  122. query := fmt.Sprintf("SELECT u.%s, pm.`memberType` FROM %s u INNER JOIN %s pm ON u.`id` = pm.`userId` WHERE pm.`productCode` = ? ORDER BY u.`id` DESC LIMIT ?,?", fields, m.table, memberTable)
  123. if err := m.QueryRowsNoCacheCtx(ctx, &list, query, productCode, (page-1)*pageSize, pageSize); err != nil {
  124. return nil, nil, 0, err
  125. }
  126. users := make([]*SysUser, len(list))
  127. memberMap := make(map[int64]string, len(list))
  128. for i, item := range list {
  129. users[i] = &item.SysUser
  130. memberMap[item.Id] = item.MemberType
  131. }
  132. return users, memberMap, total, nil
  133. }
  134. func (m *customSysUserModel) FindIdsByDeptId(ctx context.Context, deptId int64) ([]int64, error) {
  135. var ids []int64
  136. query := fmt.Sprintf("SELECT `id` FROM %s WHERE `deptId` = ?", m.table)
  137. if err := m.QueryRowsNoCacheCtx(ctx, &ids, query, deptId); err != nil {
  138. return nil, err
  139. }
  140. return ids, nil
  141. }
  142. // FindIdsByDeptIdForShareTx 见接口注释(审计 L-R16-2)。
  143. func (m *customSysUserModel) FindIdsByDeptIdForShareTx(ctx context.Context, session sqlx.Session, deptId int64) ([]int64, error) {
  144. if session == nil {
  145. return nil, errors.New("FindIdsByDeptIdForShareTx requires a non-nil session")
  146. }
  147. var ids []int64
  148. // ORDER BY id 让 S 锁的申请顺序确定,与并发的 UpdateProfileWithTx(按 id 触发 X 锁)交叉时
  149. // 形成固定加锁偏序,减小死锁概率;同时方便上层日志对照。
  150. query := fmt.Sprintf("SELECT `id` FROM %s WHERE `deptId` = ? ORDER BY `id` LOCK IN SHARE MODE", m.table)
  151. if err := session.QueryRowsCtx(ctx, &ids, query, deptId); err != nil {
  152. return nil, err
  153. }
  154. return ids, nil
  155. }
  156. func (m *customSysUserModel) UpdateProfile(ctx context.Context, id int64, username string, nickname, email, phone, remark string, deptId, newStatus int64, statusChanged bool, expectedUpdateTime int64) error {
  157. sysUserIdKey := fmt.Sprintf("%s%v", cacheSysUserIdPrefix, id)
  158. sysUserUsernameKey := fmt.Sprintf("%s%v", cacheSysUserUsernamePrefix, username)
  159. now := time.Now().Unix()
  160. res, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (sql.Result, error) {
  161. if statusChanged {
  162. query := fmt.Sprintf("UPDATE %s SET `nickname`=?, `email`=?, `phone`=?, `remark`=?, `deptId`=?, `status`=?, `tokenVersion`=`tokenVersion`+1, `updateTime`=? WHERE `id`=? AND `updateTime`=?", m.table)
  163. return conn.ExecCtx(ctx, query, nickname, email, phone, remark, deptId, newStatus, now, id, expectedUpdateTime)
  164. }
  165. query := fmt.Sprintf("UPDATE %s SET `nickname`=?, `email`=?, `phone`=?, `remark`=?, `deptId`=?, `updateTime`=? WHERE `id`=? AND `updateTime`=?", m.table)
  166. return conn.ExecCtx(ctx, query, nickname, email, phone, remark, deptId, now, id, expectedUpdateTime)
  167. }, sysUserIdKey, sysUserUsernameKey)
  168. if err != nil {
  169. return err
  170. }
  171. affected, _ := res.RowsAffected()
  172. if affected == 0 {
  173. return ErrUpdateConflict
  174. }
  175. return nil
  176. }
  177. func (m *customSysUserModel) UpdateSelfInfo(ctx context.Context, id int64, username string, nickname, avatar, email, phone string, expectedUpdateTime int64) error {
  178. sysUserIdKey := fmt.Sprintf("%s%v", cacheSysUserIdPrefix, id)
  179. sysUserUsernameKey := fmt.Sprintf("%s%v", cacheSysUserUsernamePrefix, username)
  180. now := time.Now().Unix()
  181. var avatarVal sql.NullString
  182. if avatar != "" {
  183. avatarVal = sql.NullString{String: avatar, Valid: true}
  184. }
  185. res, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (sql.Result, error) {
  186. query := fmt.Sprintf("UPDATE %s SET `nickname`=?, `avatar`=?, `email`=?, `phone`=?, `updateTime`=? WHERE `id`=? AND `updateTime`=?", m.table)
  187. return conn.ExecCtx(ctx, query, nickname, avatarVal, email, phone, now, id, expectedUpdateTime)
  188. }, sysUserIdKey, sysUserUsernameKey)
  189. if err != nil {
  190. return err
  191. }
  192. affected, _ := res.RowsAffected()
  193. if affected == 0 {
  194. return ErrUpdateConflict
  195. }
  196. return nil
  197. }
  198. // UpdateProfileWithTx 见接口注释(审计 M-R11-3 + L-R12-1)。
  199. // 实现上**绕过** m.ExecCtx 的 pre-commit DelCache 语义——仅调用 session.ExecCtx,缓存失效由
  200. // 调用方在事务 commit 成功后显式走 InvalidateProfileCache。
  201. // session==nil 时直接拒绝(非事务场景必须改走 UpdateProfile)。
  202. func (m *customSysUserModel) UpdateProfileWithTx(ctx context.Context, session sqlx.Session, id int64, username string, nickname, email, phone, remark string, deptId, newStatus int64, statusChanged bool, expectedUpdateTime int64) error {
  203. if session == nil {
  204. return errors.New("UpdateProfileWithTx requires a non-nil session")
  205. }
  206. _ = username // 保留形参以维持调用方契约一致,失效由 InvalidateProfileCache 另行处理
  207. now := time.Now().Unix()
  208. var (
  209. res sql.Result
  210. err error
  211. )
  212. if statusChanged {
  213. query := fmt.Sprintf("UPDATE %s SET `nickname`=?, `email`=?, `phone`=?, `remark`=?, `deptId`=?, `status`=?, `tokenVersion`=`tokenVersion`+1, `updateTime`=? WHERE `id`=? AND `updateTime`=?", m.table)
  214. res, err = session.ExecCtx(ctx, query, nickname, email, phone, remark, deptId, newStatus, now, id, expectedUpdateTime)
  215. } else {
  216. query := fmt.Sprintf("UPDATE %s SET `nickname`=?, `email`=?, `phone`=?, `remark`=?, `deptId`=?, `updateTime`=? WHERE `id`=? AND `updateTime`=?", m.table)
  217. res, err = session.ExecCtx(ctx, query, nickname, email, phone, remark, deptId, now, id, expectedUpdateTime)
  218. }
  219. if err != nil {
  220. return err
  221. }
  222. affected, _ := res.RowsAffected()
  223. if affected == 0 {
  224. return ErrUpdateConflict
  225. }
  226. return nil
  227. }
  228. // InvalidateProfileCache 见接口注释(审计 L-R12-1)。这里不暴露错误——post-commit 阶段
  229. // DB 已为权威,缓存清理失败只会让 sysUser 低层缓存继续提供旧值直至 TTL 过期,而 Logic 层
  230. // 紧接其后的 UserDetailsLoader.Clean 已经失效了上层 UserDetails 聚合缓存;两级缓存一起过期
  231. // 的最坏情形仍然等价于 loadUser 的 cache-miss 路径,不会放大故障面。
  232. func (m *customSysUserModel) InvalidateProfileCache(ctx context.Context, id int64, username string) {
  233. sysUserIdKey := fmt.Sprintf("%s%v", cacheSysUserIdPrefix, id)
  234. sysUserUsernameKey := fmt.Sprintf("%s%v", cacheSysUserUsernamePrefix, username)
  235. if err := m.DelCacheCtx(ctx, sysUserIdKey, sysUserUsernameKey); err != nil {
  236. // 审计 L-R13-5 方案 B:失败原因拆成 ctx 取消 vs 其它两档——前者打独立 audit tag 方便
  237. // 运维按 `cache_invalidation_skipped_due_to_ctx_cancel` 建看板,避免与真正的 Redis 故障
  238. // 混在一起报警;其它错误仍然 Errorf,保持与 sqlc 原生失效路径(Insert/Update 触发的
  239. // DelCache 失败)一致的可观测性口径。
  240. if errors.Is(err, context.Canceled) || errors.Is(err, context.DeadlineExceeded) {
  241. logx.WithContext(ctx).Errorw("cache invalidation skipped: ctx canceled",
  242. logx.Field("audit", "cache_invalidation_skipped_due_to_ctx_cancel"),
  243. logx.Field("scope", "sysUserModel.InvalidateProfileCache"),
  244. logx.Field("id", id),
  245. logx.Field("err", err.Error()),
  246. )
  247. } else {
  248. logx.WithContext(ctx).Errorf("sysUserModel.InvalidateProfileCache failed: id=%d err=%v", id, err)
  249. }
  250. }
  251. }
  252. func (m *customSysUserModel) UpdatePassword(ctx context.Context, id int64, username string, password string, mustChangePassword, expectedUpdateTime int64) error {
  253. sysUserIdKey := fmt.Sprintf("%s%v", cacheSysUserIdPrefix, id)
  254. sysUserUsernameKey := fmt.Sprintf("%s%v", cacheSysUserUsernamePrefix, username)
  255. // 审计 H-R11-1:expectedUpdateTime 必须来自**外层校验旧密码时读到的**快照。不要再内部 FindOne
  256. // 自取 data.UpdateTime —— 那会让 CAS 在本函数内自我对齐,退化为 last-write-wins,让"会话持有 +
  257. // 知道旧密码"的攻击者可以把 admin 紧急改过的新密码盖回到自己手里那一份。
  258. res, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (sql.Result, error) {
  259. query := fmt.Sprintf("UPDATE %s SET `password` = ?, `mustChangePassword` = ?, `tokenVersion` = `tokenVersion` + 1, `updateTime` = ? WHERE `id` = ? AND `updateTime` = ?", m.table)
  260. return conn.ExecCtx(ctx, query, password, mustChangePassword, time.Now().Unix(), id, expectedUpdateTime)
  261. }, sysUserIdKey, sysUserUsernameKey)
  262. if err != nil {
  263. return err
  264. }
  265. if affected, _ := res.RowsAffected(); affected == 0 {
  266. // 行被删除或被并发改过(任何字段,包括 status/password/profile):统一回 ErrUpdateConflict。
  267. return ErrUpdateConflict
  268. }
  269. return nil
  270. }
  271. // UpdateStatus 修改用户 status,并强制递增 tokenVersion 让已签发令牌失效。
  272. // 审计 L-N4:WHERE 必须带 `updateTime=?` 乐观锁,和 UpdateProfile / UpdatePassword 语义对齐。
  273. // 上游 UpdateUserStatusLogic 已经从 ValidateStatusChange 拿到 sysUser,调用方应把
  274. // `sysUser.UpdateTime` 当作 expectedUpdateTime 传入:
  275. // - expectedUpdateTime 不匹配 → ErrUpdateConflict;上层统一回 409 "数据已被其他操作修改"。
  276. // - 避免并发冻结/解冻请求走"last-write-wins",出现两个 admin 同时点"冻结"/"解冻"
  277. // 时后到者覆盖先到者、tokenVersion 被连续加两次把刚刚解冻的用户再次踢下线的诡异现象。
  278. //
  279. // 审计 L-R17-6 · 无条件递增 tokenVersion 是**刻意**设计(不要改成"仅冻结时 +1"的条件递增):
  280. // - Enabled→Disabled(冻结):签发层吊销旧 access/refresh token,JWT middleware 凭
  281. // `claims.tokenVersion < DB.tokenVersion` 一票否决。
  282. // - Disabled→Enabled(解冻):用户已因 Status!=Enabled 无法登录/刷新,+1 在业务上是空动作;
  283. // 但保留 +1 覆盖"冻结 → Redis Clean 失败 → 解冻"这条极窄路径里"旧 access token 凭残留
  284. // UD 复活"的可能,与 UpdateUserStatusLogic 的 L-R17-6 注释同口径。
  285. func (m *customSysUserModel) UpdateStatus(ctx context.Context, id int64, username string, status int64, expectedUpdateTime int64) error {
  286. // 审计 M-R11-2:username 由调用方(ValidateStatusChange 返回的目标用户对象)显式透传,
  287. // 不再内部 FindOne。真实并发安全继续靠 `WHERE updateTime = expectedUpdateTime` 乐观锁兜底。
  288. sysUserIdKey := fmt.Sprintf("%s%v", cacheSysUserIdPrefix, id)
  289. sysUserUsernameKey := fmt.Sprintf("%s%v", cacheSysUserUsernamePrefix, username)
  290. res, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (sql.Result, error) {
  291. query := fmt.Sprintf("UPDATE %s SET `status` = ?, `tokenVersion` = `tokenVersion` + 1, `updateTime` = ? WHERE `id` = ? AND `updateTime` = ?", m.table)
  292. return conn.ExecCtx(ctx, query, status, time.Now().Unix(), id, expectedUpdateTime)
  293. }, sysUserIdKey, sysUserUsernameKey)
  294. if err != nil {
  295. return err
  296. }
  297. if affected, _ := res.RowsAffected(); affected == 0 {
  298. // 行被删除或被并发改过:对外统一回 ErrUpdateConflict,避免对已删除 / 被并发改过的行
  299. // 返回 nil 让上层误判为"冻结生效"(审计 M-2 / L-N4)。
  300. return ErrUpdateConflict
  301. }
  302. return nil
  303. }
  304. // IncrementTokenVersion 强制递增当前用户的 tokenVersion,让**所有**已签发的 access/refresh 立即失效。
  305. //
  306. // WARN: 仅限"强制全量会话失效"场景调用——主动 Logout 或封禁/重置密码。Refresh / Rotate 场景
  307. // 必须调用 IncrementTokenVersionIfMatch 走 CAS 语义,否则会回到 R5 以前的并发 rotate 窗口,
  308. // 两次并发 refresh 都能换到新令牌,等同于会话劫持(见审计 L-2)。
  309. // 调用前请先走 TokenOpLimiter 等限流,避免被反复触发把合法用户 kick 出登录。
  310. func (m *customSysUserModel) IncrementTokenVersion(ctx context.Context, id int64, username string) (int64, error) {
  311. // 审计 M-R11-2:username 由调用方显式透传,不再内部 FindOne 仅为构造缓存键。
  312. // 调用方(Logout 唯一入口)在 UserDetailsLoader.Load 之后天然已经持有 ud.Username。
  313. sysUserIdKey := fmt.Sprintf("%s%v", cacheSysUserIdPrefix, id)
  314. sysUserUsernameKey := fmt.Sprintf("%s%v", cacheSysUserUsernamePrefix, username)
  315. var newVersion int64
  316. err := m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error {
  317. query := fmt.Sprintf("UPDATE %s SET `tokenVersion` = LAST_INSERT_ID(`tokenVersion` + 1), `updateTime` = ? WHERE `id` = ?", m.table)
  318. res, err := session.ExecCtx(ctx, query, time.Now().Unix(), id)
  319. if err != nil {
  320. return err
  321. }
  322. // 审计 L-R10-3:FindOne 与本次 UPDATE 之间若有并发 DELETE,affected=0 仍会返回 nil 让
  323. // SELECT LAST_INSERT_ID() 在全新事务里读出 0,调用方无法区分"真的递增成功"和"目标已被删除"。
  324. // 对外统一回 ErrUpdateConflict,调用方据此做日志/告警或跳过后续 Clean,避免对 tokenVersion=0
  325. // 之类的伪返回值做错误假设(如"版本从 1 起递增"的默认契约踩坑)。
  326. if affected, _ := res.RowsAffected(); affected == 0 {
  327. return ErrUpdateConflict
  328. }
  329. return session.QueryRowCtx(ctx, &newVersion, "SELECT LAST_INSERT_ID()")
  330. })
  331. if err != nil {
  332. return 0, err
  333. }
  334. _ = m.DelCacheCtx(ctx, sysUserIdKey, sysUserUsernameKey)
  335. return newVersion, nil
  336. }
  337. // IncrementTokenVersionIfMatch 原子递增 tokenVersion;仅当 DB 里当前 tokenVersion == expected 时才会生效。
  338. // 这是 refreshToken rotation 的原子 CAS:两个并发的刷新请求只有一个能命中 WHERE tokenVersion=expected,
  339. // 另一个 affected=0 返回 ErrTokenVersionMismatch,从而避免"两边都换到新令牌"的会话劫持窗口。
  340. //
  341. // 由上游透传 username 以便构造 cacheSysUserUsernamePrefix 的缓存键进行失效,避免为此多查一次 FindOne
  342. // (见审计 M-8)。上游通常已经通过 UserDetailsLoader.Load 拿到 username,零额外成本。
  343. func (m *customSysUserModel) IncrementTokenVersionIfMatch(ctx context.Context, id int64, username string, expected int64) (int64, error) {
  344. sysUserIdKey := fmt.Sprintf("%s%v", cacheSysUserIdPrefix, id)
  345. sysUserUsernameKey := fmt.Sprintf("%s%v", cacheSysUserUsernamePrefix, username)
  346. var newVersion int64
  347. err := m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error {
  348. query := fmt.Sprintf("UPDATE %s SET `tokenVersion` = LAST_INSERT_ID(`tokenVersion` + 1), `updateTime` = ? WHERE `id` = ? AND `tokenVersion` = ?", m.table)
  349. res, err := session.ExecCtx(ctx, query, time.Now().Unix(), id, expected)
  350. if err != nil {
  351. return err
  352. }
  353. affected, _ := res.RowsAffected()
  354. if affected == 0 {
  355. return ErrTokenVersionMismatch
  356. }
  357. return session.QueryRowCtx(ctx, &newVersion, "SELECT LAST_INSERT_ID()")
  358. })
  359. if err != nil {
  360. return 0, err
  361. }
  362. _ = m.DelCacheCtx(ctx, sysUserIdKey, sysUserUsernameKey)
  363. return newVersion, nil
  364. }
  365. // IncrementTokenVersionWithTx 见接口注释(审计 M-R15-1 方案 A)。
  366. func (m *customSysUserModel) IncrementTokenVersionWithTx(ctx context.Context, session sqlx.Session, id int64) (int64, error) {
  367. if session == nil {
  368. return 0, errors.New("IncrementTokenVersionWithTx requires a non-nil session")
  369. }
  370. query := fmt.Sprintf("UPDATE %s SET `tokenVersion` = LAST_INSERT_ID(`tokenVersion` + 1), `updateTime` = ? WHERE `id` = ?", m.table)
  371. res, err := session.ExecCtx(ctx, query, time.Now().Unix(), id)
  372. if err != nil {
  373. return 0, err
  374. }
  375. // 与 IncrementTokenVersion 的 L-R10-3 契约一致:affected=0 表示目标已被并发删除,回
  376. // ErrUpdateConflict 让上层业务事务自行 rollback——不递增 tokenVersion 比"哑失败后继续走
  377. // post-commit 失效链"安全(后者会把已不存在的 userId 从缓存里删除,没有副作用但污染日志)。
  378. if affected, _ := res.RowsAffected(); affected == 0 {
  379. return 0, ErrUpdateConflict
  380. }
  381. var newVersion int64
  382. if err := session.QueryRowCtx(ctx, &newVersion, "SELECT LAST_INSERT_ID()"); err != nil {
  383. return 0, err
  384. }
  385. return newVersion, nil
  386. }
  387. // BatchIncrementTokenVersionWithTx 见接口注释(审计 L-R15-3)。
  388. func (m *customSysUserModel) BatchIncrementTokenVersionWithTx(ctx context.Context, session sqlx.Session, ids []int64) error {
  389. if session == nil {
  390. return errors.New("BatchIncrementTokenVersionWithTx requires a non-nil session")
  391. }
  392. if len(ids) == 0 {
  393. return nil
  394. }
  395. placeholders := make([]string, len(ids))
  396. args := make([]interface{}, 0, len(ids)+1)
  397. args = append(args, time.Now().Unix())
  398. for i, id := range ids {
  399. placeholders[i] = "?"
  400. args = append(args, id)
  401. }
  402. // 批量 UPDATE 不再回读每行新 tokenVersion——调用方(UpdateProduct 禁用)只关心"集体递增
  403. // 发生了",不做 forensic 比对;若未来需要逐行返回,请改走 IN(..) + SELECT 的两步模式,
  404. // 不要试图让 LAST_INSERT_ID() 承担 N 行的反向通道(它只会保留最后一次赋值)。
  405. query := fmt.Sprintf("UPDATE %s SET `tokenVersion` = `tokenVersion` + 1, `updateTime` = ? WHERE `id` IN (%s)", m.table, strings.Join(placeholders, ","))
  406. if _, err := session.ExecCtx(ctx, query, args...); err != nil {
  407. return err
  408. }
  409. return nil
  410. }
  411. func (m *customSysUserModel) FindByIds(ctx context.Context, ids []int64) ([]*SysUser, error) {
  412. if len(ids) == 0 {
  413. return nil, nil
  414. }
  415. placeholders := make([]string, len(ids))
  416. args := make([]interface{}, len(ids))
  417. for i, id := range ids {
  418. placeholders[i] = "?"
  419. args[i] = id
  420. }
  421. var list []*SysUser
  422. query := fmt.Sprintf("SELECT %s FROM %s WHERE `id` IN (%s)", sysUserRows, m.table, strings.Join(placeholders, ","))
  423. if err := m.QueryRowsNoCacheCtx(ctx, &list, query, args...); err != nil {
  424. return nil, err
  425. }
  426. return list, nil
  427. }