JwtToken 添加判断

performance/Performance.Api/Controllers/AccountController.cs

@@ -123,7 +123,7 @@ public ApiResponse<JwtToken> QuickLogin([FromBody] ResetPwdRequest request)
                 new Claim(JwtClaimTypes.QuickLogin, EQuickLogin.YES.ToString()),
             };
             var roles = _roleService.GetUserRole(user.UserID);
-            if (roles?.Any() == true)
+            if (roles?.Any() == true && UnitTypeUtil.Maps.ContainsKey(roles.First().RoleID))
             {
                 var unitTypes = UnitTypeUtil.Maps.GetOrAdd(roles.First().RoleID, (key) => Array.Empty<string>());
                 claims.Add(new Claim(JwtClaimTypes.UnitType, JsonConvert.SerializeObject(unitTypes)));
@@ -298,7 +298,7 @@ public ApiResponse<JwtToken> DemoUsers(int userId)
                 new Claim(JwtClaimTypes.PersonnelNumber, user.Login ?? ""),
             };
             var roles = _roleService.GetUserRole(user.UserID);
-            if (roles?.Any() == true)
+            if (roles?.Any() == true && UnitTypeUtil.Maps.ContainsKey(roles.First().RoleID))
             {
                 var unitTypes = UnitTypeUtil.Maps.GetOrAdd(roles.First().RoleID, (key) => Array.Empty<string>());
                 claims.Add(new Claim(JwtClaimTypes.UnitType, JsonConvert.SerializeObject(unitTypes)));