| 235 | } |
| 236 | |
| 237 | func updateUserCacheFieldAtVersion(userId int, field string, value any, authVersion int64) error { |
| 238 | if !common.RedisEnabled { |
| 239 | return nil |
| 240 | } |
| 241 | if userId <= 0 || authVersion <= 0 { |
| 242 | return fmt.Errorf("invalid user auth version") |
| 243 | } |
| 244 | const script = ` |
| 245 | local incoming = tonumber(ARGV[1]) |
| 246 | local pending = tonumber(redis.call('GET', KEYS[2]) or '0') |
| 247 | local committed = tonumber(redis.call('GET', KEYS[3]) or '0') |
| 248 | local current = tonumber(redis.call('HGET', KEYS[1], 'AuthVersion') or '0') |
| 249 | if pending > incoming or committed > incoming or current > incoming then |
| 250 | return 0 |
| 251 | end |
| 252 | if committed < incoming then |
| 253 | redis.call('SET', KEYS[3], ARGV[1]) |
| 254 | end |
| 255 | if pending > 0 and pending <= incoming then |
| 256 | redis.call('DEL', KEYS[2]) |
| 257 | end |
| 258 | if redis.call('EXISTS', KEYS[1]) == 0 then |
| 259 | return 1 |
| 260 | end |
| 261 | if current ~= incoming then |
| 262 | return 1 |
| 263 | end |
| 264 | redis.call('HSET', KEYS[1], ARGV[2], ARGV[3], 'CacheSchema', ARGV[4]) |
| 265 | return 1` |
| 266 | result, err := common.RDB.Eval(context.Background(), script, |
| 267 | []string{getUserCacheKey(userId), getUserAuthFenceKey(userId), getUserAuthVersionKey(userId)}, |
| 268 | authVersion, field, value, userCacheSchemaVersion, |
| 269 | ).Int() |
| 270 | if err != nil { |
| 271 | return err |
| 272 | } |
| 273 | if result == 0 { |
| 274 | return ErrUserAuthCachePending |
| 275 | } |
| 276 | return nil |
| 277 | } |