若依 3.0

RuoYi

2020-07-19 b7e78f3e9ae3a7d57cdbfb849064cfeed23ff9e4

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
package com.ruoyi.framework.security.service; 
 
import javax.annotation.Resource; 
import org.springframework.beans.factory.annotation.Autowired; 
import org.springframework.security.authentication.AuthenticationManager; 
import org.springframework.security.authentication.BadCredentialsException; 
import org.springframework.security.authentication.UsernamePasswordAuthenticationToken; 
import org.springframework.security.core.Authentication; 
import org.springframework.stereotype.Component; 
import com.ruoyi.common.constant.Constants; 
import com.ruoyi.common.exception.CustomException; 
import com.ruoyi.common.exception.user.CaptchaException; 
import com.ruoyi.common.exception.user.CaptchaExpireException; 
import com.ruoyi.common.exception.user.UserPasswordNotMatchException; 
import com.ruoyi.common.utils.MessageUtils; 
import com.ruoyi.framework.manager.AsyncManager; 
import com.ruoyi.framework.manager.factory.AsyncFactory; 
import com.ruoyi.framework.redis.RedisCache; 
import com.ruoyi.framework.security.LoginUser; 
 
/** 
 * 登录校验方法 
 *  
 * @author ruoyi 
 */ 
@Component 
public class SysLoginService 
{ 
    @Autowired 
    private TokenService tokenService; 
 
    @Resource 
    private AuthenticationManager authenticationManager; 
 
    @Autowired 
    private RedisCache redisCache; 
 
    /** 
     * 登录验证 
     *  
     * @param username 用户名 
     * @param password 密码 
     * @param code 验证码 
     * @param uuid 唯一标识 
     * @return 结果 
     */ 
    public String login(String username, String password, String code, String uuid) 
    { 
        String verifyKey = Constants.CAPTCHA_CODE_KEY + uuid; 
        String captcha = redisCache.getCacheObject(verifyKey); 
        redisCache.deleteObject(verifyKey); 
        if (captcha == null) 
        { 
            AsyncManager.me().execute(AsyncFactory.recordLogininfor(username, Constants.LOGIN_FAIL, MessageUtils.message("user.jcaptcha.expire"))); 
            throw new CaptchaExpireException(); 
        } 
        if (!code.equalsIgnoreCase(captcha)) 
        { 
            AsyncManager.me().execute(AsyncFactory.recordLogininfor(username, Constants.LOGIN_FAIL, MessageUtils.message("user.jcaptcha.error"))); 
            throw new CaptchaException(); 
        } 
        // 用户验证 
        Authentication authentication = null; 
        try 
        { 
            // 该方法会去调用UserDetailsServiceImpl.loadUserByUsername 
            authentication = authenticationManager 
                    .authenticate(new UsernamePasswordAuthenticationToken(username, password)); 
        } 
        catch (Exception e) 
        { 
            if (e instanceof BadCredentialsException) 
            { 
                AsyncManager.me().execute(AsyncFactory.recordLogininfor(username, Constants.LOGIN_FAIL, MessageUtils.message("user.password.not.match"))); 
                throw new UserPasswordNotMatchException(); 
            } 
            else 
            { 
                AsyncManager.me().execute(AsyncFactory.recordLogininfor(username, Constants.LOGIN_FAIL, e.getMessage())); 
                throw new CustomException(e.getMessage()); 
            } 
        } 
        AsyncManager.me().execute(AsyncFactory.recordLogininfor(username, Constants.LOGIN_SUCCESS, MessageUtils.message("user.login.success"))); 
        LoginUser loginUser = (LoginUser) authentication.getPrincipal(); 
        // 生成token 
        return tokenService.createToken(loginUser); 
    } 
}