api_key_auth_cache_invalidate.go 1.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748
  1. package service
  2. import "context"
  3. // InvalidateAuthCacheByKey 清除指定 API Key 的认证缓存
  4. func (s *APIKeyService) InvalidateAuthCacheByKey(ctx context.Context, key string) {
  5. if key == "" {
  6. return
  7. }
  8. cacheKey := s.authInvalidationCacheKey(key)
  9. s.deleteAuthCache(ctx, cacheKey)
  10. }
  11. // InvalidateAuthCacheByUserID 清除用户相关的 API Key 认证缓存
  12. func (s *APIKeyService) InvalidateAuthCacheByUserID(ctx context.Context, userID int64) {
  13. if userID <= 0 {
  14. return
  15. }
  16. keys, err := s.apiKeyRepo.ListKeysByUserID(ctx, userID)
  17. if err != nil {
  18. return
  19. }
  20. s.deleteAuthCacheByKeys(ctx, keys)
  21. }
  22. // InvalidateAuthCacheByGroupID 清除分组相关的 API Key 认证缓存
  23. func (s *APIKeyService) InvalidateAuthCacheByGroupID(ctx context.Context, groupID int64) {
  24. if groupID <= 0 {
  25. return
  26. }
  27. keys, err := s.apiKeyRepo.ListKeysByGroupID(ctx, groupID)
  28. if err != nil {
  29. return
  30. }
  31. s.deleteAuthCacheByKeys(ctx, keys)
  32. }
  33. func (s *APIKeyService) deleteAuthCacheByKeys(ctx context.Context, keys []string) {
  34. if len(keys) == 0 {
  35. return
  36. }
  37. for _, key := range keys {
  38. if key == "" {
  39. continue
  40. }
  41. s.deleteAuthCache(ctx, s.authInvalidationCacheKey(key))
  42. }
  43. }