sysUserModel.go 22 KB

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