From b1935a09641bc915c61f1868d911adf455d79833 Mon Sep 17 00:00:00 2001
From: ZhuDongming <773644075@qq.com>
Date: Tue, 28 Apr 2020 09:33:41 +0800
Subject: [PATCH] update大屏本月平均值查询表

---
 src/main/java/com/moral/security/auth/login/LoginAuthenticationProvider.java |   23 +++++++++++++++++++----
 1 files changed, 19 insertions(+), 4 deletions(-)

diff --git a/src/main/java/com/moral/security/auth/login/LoginAuthenticationProvider.java b/src/main/java/com/moral/security/auth/login/LoginAuthenticationProvider.java
index 307d880..999813a 100644
--- a/src/main/java/com/moral/security/auth/login/LoginAuthenticationProvider.java
+++ b/src/main/java/com/moral/security/auth/login/LoginAuthenticationProvider.java
@@ -1,8 +1,11 @@
 package com.moral.security.auth.login;
 
 import com.moral.entity.Account;
+import com.moral.entity.Role;
+import com.moral.security.exceptions.AccountExpiredBadCredentialsException;
 import com.moral.security.model.UserContext;
 import com.moral.service.AccountService;
+
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.security.authentication.AuthenticationProvider;
 import org.springframework.security.authentication.BadCredentialsException;
@@ -16,8 +19,11 @@
 import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder;
 import org.springframework.stereotype.Component;
 import org.springframework.util.Assert;
+import org.springframework.util.CollectionUtils;
 
+import java.util.Date;
 import java.util.List;
+import java.util.Optional;
 import java.util.stream.Collectors;
 
 /**
@@ -51,15 +57,24 @@
         String password = (String) authentication.getCredentials();
         LoginMode mode = (LoginMode) authentication.getDetails();
         Account account = accountService.queryAccountByName(accountName).orElseThrow(() -> new UsernameNotFoundException("User not found: " + accountName));
+        List<Role> roleList = accountService.getRolesByAccountName(accountName);
+        Date expireTime = Optional.ofNullable(account.getExpireTime())
+                .orElseThrow(
+        ()-> new AccountExpiredBadCredentialsException("Authentication Failed. Account has expired.")
+                );
+        Date nowTime = new Date();
+        if(expireTime.getTime()<nowTime.getTime()){
+            throw new AccountExpiredBadCredentialsException("Authentication Failed. Account has expired.");
+        }
         if (!encoder.matches(password, account.getPassword())) {
             throw new BadCredentialsException("Authentication Failed. Username or Password not valid.");
         }
 
-        if (account.getRoles() == null) {
-            throw new InsufficientAuthenticationException("User has no roles assigned");
+        if (CollectionUtils.isEmpty(roleList)) {
+            throw new InsufficientAuthenticationException("Authentication Failed. User has no roles assigned");
         }
-        List<GrantedAuthority> authorities = account.getRoles().stream()
-                .map(authority -> new SimpleGrantedAuthority(authority.getName()))
+        List<GrantedAuthority> authorities = roleList.stream()
+                .map(authority -> new SimpleGrantedAuthority(authority.getRoleName()))
                 .collect(Collectors.toList());
 
         UserContext userContext = UserContext.create(account.getAccountName(),mode,account.getOrganizationId(),authorities);

--
Gitblit v1.8.0