提交 | 用户 | 时间
|
71e81e
|
1 |
package cn.stylefeng.guns.oauth2.service.impl; |
懒 |
2 |
|
|
3 |
import cn.stylefeng.guns.base.auth.service.AuthService; |
|
4 |
import cn.stylefeng.guns.base.oauth2.service.OauthUserInfoService; |
|
5 |
import cn.stylefeng.guns.oauth2.service.LoginService; |
|
6 |
import cn.stylefeng.guns.sys.core.exception.oauth.OAuthExceptionEnum; |
|
7 |
import cn.stylefeng.guns.sys.core.exception.oauth.OAuthLoginException; |
|
8 |
import cn.stylefeng.guns.sys.modular.system.entity.User; |
|
9 |
import cn.stylefeng.guns.sys.modular.system.service.UserService; |
|
10 |
import cn.stylefeng.guns.base.oauth2.entity.OauthUserInfo; |
|
11 |
import cn.stylefeng.guns.oauth2.factory.OAuthUserInfoFactory; |
|
12 |
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; |
|
13 |
import me.zhyd.oauth.model.AuthUser; |
|
14 |
import org.springframework.beans.factory.annotation.Autowired; |
|
15 |
import org.springframework.stereotype.Service; |
|
16 |
import org.springframework.transaction.annotation.Transactional; |
|
17 |
|
|
18 |
import java.util.List; |
|
19 |
|
|
20 |
/** |
|
21 |
* 默认第三方登录逻辑 |
|
22 |
* |
|
23 |
* @author fengshuonan |
|
24 |
* @Date 2019/6/9 18:16 |
|
25 |
*/ |
|
26 |
@Service |
|
27 |
public class DefaultLoginService implements LoginService { |
|
28 |
|
|
29 |
@Autowired |
|
30 |
private UserService userService; |
|
31 |
|
|
32 |
@Autowired |
|
33 |
private OauthUserInfoService oauthUserInfoService; |
|
34 |
|
|
35 |
@Autowired |
|
36 |
private AuthService authService; |
|
37 |
|
|
38 |
@Override |
|
39 |
@Transactional(rollbackFor = Exception.class) |
|
40 |
public String oauthLogin(AuthUser oauthUser) { |
|
41 |
|
|
42 |
if (oauthUser == null) { |
|
43 |
throw new OAuthLoginException(OAuthExceptionEnum.OAUTH_RESPONSE_ERROR); |
|
44 |
} |
|
45 |
|
|
46 |
//当前无登录用户,创建用户或根据已有绑定用户的账号登录 |
|
47 |
String account = getOauthUserAccount(oauthUser); |
|
48 |
|
|
49 |
//执行原有系统登录逻辑 |
|
50 |
return authService.login(account); |
|
51 |
} |
|
52 |
|
|
53 |
|
|
54 |
/** |
|
55 |
* 绑定当前用户的source和openId |
|
56 |
* |
|
57 |
* @author fengshuonan |
|
58 |
* @Date 2019/6/9 18:51 |
|
59 |
*/ |
|
60 |
private void bindOAuthUser(Long userId, AuthUser oauthUser) { |
|
61 |
|
|
62 |
//先判断当前系统这个openId有没有人用 |
|
63 |
QueryWrapper<OauthUserInfo> queryWrapper = new QueryWrapper<OauthUserInfo>() |
|
64 |
.eq("source", oauthUser.getSource().name()) |
|
65 |
.and(i -> i.eq("uuid", oauthUser.getUuid())) |
|
66 |
.and(i -> i.ne("user_id", userId)); |
|
67 |
List<OauthUserInfo> oauthUserInfos = this.oauthUserInfoService.list(queryWrapper); |
|
68 |
|
|
69 |
//已有人绑定,抛出异常 |
|
70 |
if (oauthUserInfos != null && oauthUserInfos.size() > 0) { |
|
71 |
throw new OAuthLoginException(OAuthExceptionEnum.OPEN_ID_ALREADY_BIND); |
|
72 |
} else { |
|
73 |
//新建一条绑定记录 |
|
74 |
OauthUserInfo oAuthUserInfo = OAuthUserInfoFactory.createOAuthUserInfo(userId, oauthUser); |
|
75 |
this.oauthUserInfoService.save(oAuthUserInfo); |
|
76 |
} |
|
77 |
|
|
78 |
} |
|
79 |
|
|
80 |
/** |
|
81 |
* 通过第三方登录的信息创建本系统用户 |
|
82 |
* |
|
83 |
* @author fengshuonan |
|
84 |
* @Date 2019/6/9 19:07 |
|
85 |
*/ |
|
86 |
private String getOauthUserAccount(AuthUser oauthUser) { |
|
87 |
|
|
88 |
//先判断当前系统这个openId有没有人用 |
|
89 |
QueryWrapper<OauthUserInfo> queryWrapper = new QueryWrapper<OauthUserInfo>() |
|
90 |
.eq("source", oauthUser.getSource().name()) |
|
91 |
.and(i -> i.eq("uuid", oauthUser.getUuid())); |
|
92 |
OauthUserInfo oauthUserInfos = this.oauthUserInfoService.getOne(queryWrapper); |
|
93 |
|
|
94 |
//已有人绑定,直接返回这个人的账号,进行登录 |
|
95 |
if (oauthUserInfos != null) { |
|
96 |
Long userId = oauthUserInfos.getUserId(); |
|
97 |
return this.userService.getById(userId).getAccount(); |
|
98 |
} else { |
|
99 |
|
|
100 |
//没有人绑定的创建这个人的本系统用户 |
|
101 |
User user = OAuthUserInfoFactory.createOAuthUser(oauthUser); |
|
102 |
this.userService.save(user); |
|
103 |
|
|
104 |
//新建一条oauth2绑定记录 |
|
105 |
OauthUserInfo oAuthUserInfo = OAuthUserInfoFactory.createOAuthUserInfo(user.getUserId(), oauthUser); |
|
106 |
this.oauthUserInfoService.save(oAuthUserInfo); |
|
107 |
|
|
108 |
return user.getAccount(); |
|
109 |
} |
|
110 |
} |
|
111 |
|
|
112 |
} |