optimise codes

src/Infrastructure/Extensions/TokenContextSetup.cs

@@ -10,7 +10,7 @@ public static class TokenContextSetup
     public static IServiceCollection AddDefaultTokenContext(this IServiceCollection services)
     {
         ArgumentNullException.ThrowIfNull(services);
-        services.TryAddSingleton<JwtSecurityTokenHandler>();
+        services.TryAddSingleton<JsonWebTokenHandler>();
         services.TryAddSingleton<DefaultTokenHandler>();
         return services;
     }

src/Infrastructure/GlobalUsing.cs

@@ -11,7 +11,8 @@ global using System.Reflection;
 global using System.Text;
 global using System.Text.Json;
 global using System.Threading;
-global using System.IdentityModel.Tokens.Jwt;
 global using Infrastructure.Security;
 global using Microsoft.Extensions.DependencyInjection;
-global using Microsoft.Extensions.DependencyInjection.Extensions;
+global using Microsoft.Extensions.DependencyInjection.Extensions;
+global using Microsoft.IdentityModel.JsonWebTokens;
+global using Microsoft.IdentityModel.Tokens;

src/Infrastructure/HttpUserContext/UserContext.cs

@@ -3,7 +3,6 @@ using System.Security.Claims;
 using Infrastructure.Utils;
 using Microsoft.AspNetCore.Authentication.JwtBearer;
 using Microsoft.AspNetCore.Http;
-using Microsoft.IdentityModel.Tokens;
 
 namespace Infrastructure.HttpUserContext;
 
@@ -13,13 +12,13 @@ namespace Infrastructure.HttpUserContext;
 /// <param name="httpContextAccessor"></param>
 /// <param name="jwtContext"></param>
 /// <param name="encryptionService"></param>
-/// <param name="jwtSecurityTokenHandler"></param>
+/// <param name="jsonWebTokenHandler"></param>
 /// <typeparam name="TId"></typeparam>
 public class UserContext<TId>(
     IHttpContextAccessor httpContextAccessor,
     JwtContext jwtContext,
     IEncryptionService encryptionService,
-    JwtSecurityTokenHandler jwtSecurityTokenHandler)
+    JsonWebTokenHandler jsonWebTokenHandler)
     : IUserContext<TId> where TId : IEquatable<TId>
 {
     private readonly ClaimsPrincipal principal = httpContextAccessor?.HttpContext?.User;
@@ -48,13 +47,13 @@ public class UserContext<TId>(
 
     public string Username
     {
-        get => _username ??= principal.Claims.First(c => c.Type == JwtRegisteredClaimNames.UniqueName).Value;
+        get => _username ??= GetClaimValue(JwtRegisteredClaimNames.UniqueName);
         set => _username = value;
     }
 
     public string Name
     {
-        get => _name ??= principal.Claims.First(c => c.Type == JwtRegisteredClaimNames.Name).Value;
+        get => _name ??= GetClaimValue(JwtRegisteredClaimNames.Name);
         set => _name = value;
     }
 
@@ -101,14 +100,16 @@ public class UserContext<TId>(
             duration = jwtContext.Duration;
         }
 
-        var securityToken = new JwtSecurityToken(
-            issuer: jwtContext.Issuer,
-            audience: jwtContext.Audience,
-            claims: claims,
-            notBefore: DateTime.Now,
-            expires: DateTime.Now.AddSeconds(jwtContext.Duration),
-            signingCredentials: jwtContext.SigningCredentials);
-        var token = jwtSecurityTokenHandler.WriteToken(securityToken);
+        var tokenDescriptor = new SecurityTokenDescriptor()
+        {
+            Issuer = jwtContext.Issuer,
+            Audience = jwtContext.Audience,
+            Claims = claims?.ToDictionary(c => c.Type, c => (object)c.Value),
+            NotBefore = DateTime.Now,
+            Expires = DateTime.Now.AddSeconds(duration.Value),
+            SigningCredentials = jwtContext.SigningCredentials,
+        };
+        var token = jsonWebTokenHandler.CreateToken(tokenDescriptor);
         token = encryptionService.Encrypt(token);
         return new JwtTokenInfo(token, duration.Value, schemeName);
     }
@@ -142,4 +143,14 @@ public class UserContext<TId>(
         var idClaim = principal.Claims.First(c => c.Type == JwtRegisteredClaimNames.NameId);
         return (TId)Convert.ChangeType(idClaim.Value, typeof(TId));
     }
+
+    private string GetClaimValue(string claimType)
+    {
+        return principal.Claims.First(c => c.Type == claimType).Value;
+    }
+
+    private string[] GetClaimValues(string claimType)
+    {
+        return principal.Claims.Where(c => c.Type == claimType).Select(c => c.Value).ToArray();
+    }
 }

src/Infrastructure/Security/DefaultTokenHandler.cs

@@ -1,19 +1,17 @@
-using Microsoft.IdentityModel.Tokens;
-
 namespace Infrastructure.Security;
 
 /// <summary>
 /// 自定义token解密
 /// </summary>
 /// <param name="encryptionService"></param>
-/// <param name="jwtSecurityTokenHandler"></param>
-public class DefaultTokenHandler(IEncryptionService encryptionService, JwtSecurityTokenHandler jwtSecurityTokenHandler)
+/// <param name="jwtTokenHandler"></param>
+public class DefaultTokenHandler(IEncryptionService encryptionService, JsonWebTokenHandler jwtTokenHandler)
     : TokenHandler
 {
     public override Task<TokenValidationResult> ValidateTokenAsync(string token,
         TokenValidationParameters validationParameters)
     {
         var decodeToken = encryptionService.Decrypt(token);
-        return jwtSecurityTokenHandler.ValidateTokenAsync(decodeToken, validationParameters);
+        return jwtTokenHandler.ValidateTokenAsync(decodeToken, validationParameters);
     }
 }

src/Infrastructure/Security/JwtContext.cs

@@ -1,5 +1,3 @@
-using Microsoft.IdentityModel.Tokens;
-
 namespace Infrastructure.Security;
 
 /// <summary>