cache/auth 不强依赖Redis

Helpers/SimApiAuth.cs

@@ -1,19 +1,32 @@
 using System;
+using System.Collections.Concurrent;
 using System.Collections.Generic;
 using Microsoft.Extensions.Caching.Distributed;
+using Microsoft.Extensions.DependencyInjection;
 using SimApi.Communications;
 using StackExchange.Redis;
 
 namespace SimApi.Helpers;
 
 /// <summary>
-/// 认证助手
+/// 认证助手（支持 Redis 和 InMemory 两种模式）
 /// </summary>
-public class SimApiAuth(IDistributedCache cache, IConnectionMultiplexer redis)
+public class SimApiAuth
 {
     private const string TokenCacheKey = "SimApi:Auth:Token:{token}";
     private const string TokenSetCacheKey = "SimApi:Auth:User:{userId}";
-    private readonly IDatabase _redisDb = redis.GetDatabase();
+
+    private readonly IDistributedCache _cache;
+    private readonly IDatabase? _redisDb;
+
+    // InMemory 模式：用户 → Token集合
+    private readonly ConcurrentDictionary<string, ConcurrentDictionary<string, byte>> _userTokens = new();
+
+    public SimApiAuth(IDistributedCache cache, IServiceProvider sp)
+    {
+        _cache = cache;
+        _redisDb = sp.GetService<IConnectionMultiplexer>()?.GetDatabase();
+    }
 
     /// <summary>
     /// 登录信息
@@ -28,13 +41,20 @@ public class SimApiAuth(IDistributedCache cache, IConnectionMultiplexer redis)
         token ??= Guid.NewGuid().ToString();
         var cacheKey = TokenCacheKey.Replace("{token}", token);
         var setCacheKey = TokenSetCacheKey.Replace("{userId}", loginItem.Id);
-        _redisDb.SetAdd(setCacheKey, token);
-        cache.SetString(cacheKey, SimApiUtil.Json(loginItem),
-            new DistributedCacheEntryOptions
-            {
-                SlidingExpiration = expireTime
-            });
-        _redisDb.KeyExpire(setCacheKey, expireTime.Value);
+
+        _cache.SetString(cacheKey, SimApiUtil.Json(loginItem),
+            new DistributedCacheEntryOptions { SlidingExpiration = expireTime });
+
+        if (_redisDb != null)
+        {
+            _redisDb.SetAdd(setCacheKey, token);
+            _redisDb.KeyExpire(setCacheKey, expireTime.Value);
+        }
+        else
+        {
+            var tokens = _userTokens.GetOrAdd(loginItem.Id, _ => new ConcurrentDictionary<string, byte>());
+            tokens.TryAdd(token, 0);
+        }
 
         return token;
     }
@@ -48,10 +68,15 @@ public class SimApiAuth(IDistributedCache cache, IConnectionMultiplexer redis)
     public string Update(SimApiLoginItem loginItem, string token)
     {
         var cacheKey = TokenCacheKey.Replace("{token}", token);
-        cache.SetString(cacheKey, SimApiUtil.Json(loginItem));
-        var ttl = _redisDb.KeyTimeToLive(cacheKey);
-        var setCacheKey = TokenSetCacheKey.Replace("{userId}", loginItem.Id);
-        _redisDb.KeyExpire(setCacheKey, ttl);
+        _cache.SetString(cacheKey, SimApiUtil.Json(loginItem));
+
+        if (_redisDb != null)
+        {
+            var ttl = _redisDb.KeyTimeToLive(cacheKey);
+            var setCacheKey = TokenSetCacheKey.Replace("{userId}", loginItem.Id);
+            _redisDb.KeyExpire(setCacheKey, ttl);
+        }
+
         return token;
     }
 
@@ -64,9 +89,10 @@ public class SimApiAuth(IDistributedCache cache, IConnectionMultiplexer redis)
     public SimApiLoginItem? GetLogin(string token)
     {
         var cacheKey = TokenCacheKey.Replace("{token}", token);
-        var login = cache.GetString(cacheKey);
+        var login = _cache.GetString(cacheKey);
         var resp = login != null ? SimApiUtil.FromJson<SimApiLoginItem>(login) : null;
-        if (resp != null)
+
+        if (resp != null && _redisDb != null)
         {
             var ttl = _redisDb.KeyTimeToLive(cacheKey);
             var setCacheKey = TokenSetCacheKey.Replace("{userId}", resp.Id);
@@ -83,26 +109,42 @@ public class SimApiAuth(IDistributedCache cache, IConnectionMultiplexer redis)
     /// <returns></returns>
     public SimApiLoginItem[] GetAllLogins(string userId)
     {
-        var setCacheKey = TokenSetCacheKey.Replace("{userId}", userId);
-        var allLogins = _redisDb.SetMembers(setCacheKey).ToStringArray();
-        var resp = new List<SimApiLoginItem>();
-        foreach (var login in allLogins)
+        if (_redisDb != null)
         {
-            if (login != null)
+            var setCacheKey = TokenSetCacheKey.Replace("{userId}", userId);
+            var allLogins = _redisDb.SetMembers(setCacheKey).ToStringArray();
+            var resp = new List<SimApiLoginItem>();
+            foreach (var login in allLogins)
             {
-                var item = GetLogin(login);
-                if (item != null)
+                if (login != null)
                 {
-                    resp.Add(item);
-                }
-                else
-                {
-                    _redisDb.SetRemove(setCacheKey, login);
+                    var item = GetLogin(login);
+                    if (item != null)
+                        resp.Add(item);
+                    else
+                        _redisDb.SetRemove(setCacheKey, login);
                 }
             }
-        }
 
-        return resp.ToArray();
+            return resp.ToArray();
+        }
+        else
+        {
+            if (!_userTokens.TryGetValue(userId, out var tokens))
+                return [];
+
+            var resp = new List<SimApiLoginItem>();
+            foreach (var token in tokens.Keys)
+            {
+                var item = GetLogin(token);
+                if (item != null)
+                    resp.Add(item);
+                else
+                    tokens.TryRemove(token, out _);
+            }
+
+            return resp.ToArray();
+        }
     }
 
     /// <summary>
@@ -111,18 +153,32 @@ public class SimApiAuth(IDistributedCache cache, IConnectionMultiplexer redis)
     /// <param name="userId"></param>
     public void LogoutAll(string userId)
     {
-        var setCacheKey = TokenSetCacheKey.Replace("{userId}", userId);
-        var allLogins = _redisDb.SetMembers(setCacheKey).ToStringArray();
-        foreach (var login in allLogins)
+        if (_redisDb != null)
         {
-            if (login != null)
+            var setCacheKey = TokenSetCacheKey.Replace("{userId}", userId);
+            var allLogins = _redisDb.SetMembers(setCacheKey).ToStringArray();
+            foreach (var login in allLogins)
             {
-                var cacheKey = TokenCacheKey.Replace("{token}", login);
-                cache.Remove(cacheKey);
+                if (login != null)
+                {
+                    var cacheKey = TokenCacheKey.Replace("{token}", login);
+                    _cache.Remove(cacheKey);
+                }
+            }
+
+            _redisDb.KeyDelete(setCacheKey);
+        }
+        else
+        {
+            if (_userTokens.TryRemove(userId, out var tokens))
+            {
+                foreach (var token in tokens.Keys)
+                {
+                    var cacheKey = TokenCacheKey.Replace("{token}", token);
+                    _cache.Remove(cacheKey);
+                }
             }
         }
-
-        _redisDb.KeyDelete(setCacheKey);
     }
 
     /// <summary>
@@ -132,13 +188,22 @@ public class SimApiAuth(IDistributedCache cache, IConnectionMultiplexer redis)
     public void Logout(string token)
     {
         var item = GetLogin(token);
-        if (item != null)
+
+        if (_redisDb != null)
         {
-            var setCacheKey = TokenSetCacheKey.Replace("{userId}", item.Id);
-            _redisDb.SetRemove(setCacheKey, token);
+            if (item != null)
+            {
+                var setCacheKey = TokenSetCacheKey.Replace("{userId}", item.Id);
+                _redisDb.SetRemove(setCacheKey, token);
+            }
+        }
+        else
+        {
+            if (item != null && _userTokens.TryGetValue(item.Id, out var tokens))
+                tokens.TryRemove(token, out _);
         }
 
         var cacheKey = TokenCacheKey.Replace("{token}", token);
-        cache.Remove(cacheKey);
+        _cache.Remove(cacheKey);
     }
-}
+}