From 9101357f44fdb3d7972f8e8979bc7bec6eedfebb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=86=B7=E5=86=B7?= <2270033969@qq.com>
Date: Thu, 5 Dec 2024 22:58:22 +0800
Subject: [PATCH] =?UTF-8?q?refactor(=E7=A4=BE=E4=BA=A4=E7=99=BB=E5=BD=95):?=
=?UTF-8?q?=20=E5=89=8D=E5=90=8E=E7=AB=AF=E4=BD=BF=E7=94=A8=E7=BB=9F?=
=?UTF-8?q?=E4=B8=80=E6=9E=9A=E4=B8=BE=EF=BC=8C=E9=81=BF=E5=85=8D=E9=87=8D?=
=?UTF-8?q?=E5=A4=8D=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit
---
.../admin/handler/WeChatCPLoginHandler.java | 113 +++++++++---------
.../service/impl/SysUserServiceImpl.java | 25 ++--
2 files changed, 70 insertions(+), 68 deletions(-)
diff --git a/pigx-upms/pigx-upms-biz/src/main/java/com/pig4cloud/pigx/admin/handler/WeChatCPLoginHandler.java b/pigx-upms/pigx-upms-biz/src/main/java/com/pig4cloud/pigx/admin/handler/WeChatCPLoginHandler.java
index 6d6fff94d..5c55a435b 100644
--- a/pigx-upms/pigx-upms-biz/src/main/java/com/pig4cloud/pigx/admin/handler/WeChatCPLoginHandler.java
+++ b/pigx-upms/pigx-upms-biz/src/main/java/com/pig4cloud/pigx/admin/handler/WeChatCPLoginHandler.java
@@ -27,78 +27,81 @@ import com.pig4cloud.pigx.admin.api.entity.SysUser;
import com.pig4cloud.pigx.admin.mapper.SysSocialDetailsMapper;
import com.pig4cloud.pigx.admin.service.SysUserService;
import com.pig4cloud.pigx.common.core.constant.enums.LoginTypeEnum;
-import lombok.AllArgsConstructor;
+import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
/**
* @author lengleng
* @date 2023-10-14
- *
+ *
* 企业微信登录
*/
@Slf4j
-@Component("CP")
-@AllArgsConstructor
+@Component("WEIXIN_CP")
+@RequiredArgsConstructor
public class WeChatCPLoginHandler extends AbstractLoginHandler {
- private final SysUserService sysUserService;
+ private final SysUserService sysUserService;
- private final SysSocialDetailsMapper sysSocialDetailsMapper;
+ private final SysSocialDetailsMapper sysSocialDetailsMapper;
- /**
- * QQ登录传入code
- *
- * 通过code 调用qq 获取唯一标识
- * @param code
- * @return
- */
- @Override
- public String identify(String code) {
- SysSocialDetails condition = new SysSocialDetails();
- condition.setType(LoginTypeEnum.WEIXIN_CP.getType());
- SysSocialDetails socialDetails = sysSocialDetailsMapper.selectOne(new QueryWrapper<>(condition));
+ /**
+ * 企微登录传入code
+ *
+ * 通过code 调用获取唯一标识
+ *
+ * @param code
+ * @return
+ */
+ @Override
+ public String identify(String code) {
+ SysSocialDetails condition = new SysSocialDetails();
+ condition.setType(LoginTypeEnum.WEIXIN_CP.getType());
+ SysSocialDetails socialDetails = sysSocialDetailsMapper.selectOne(new QueryWrapper<>(condition));
- String getAccessTokenUrl = "https://qyapi.weixin.qq.com/cgi-bin/gettoken?corpid=%s&corpsecret=%s";
+ String getAccessTokenUrl = "https://qyapi.weixin.qq.com/cgi-bin/gettoken?corpid=%s&corpsecret=%s";
- String accessTokenResult = HttpUtil
- .get(String.format(getAccessTokenUrl, socialDetails.getAppId(), socialDetails.getAppSecret()));
- log.debug("获取企业微信Token响应报文：{}", accessTokenResult);
- String accessToken = JSONUtil.parseObj(accessTokenResult).getStr("access_token");
- String getUserIdUrl = "https://qyapi.weixin.qq.com/cgi-bin/auth/getuserinfo?access_token=%s&code=%s";
- String userResult = HttpUtil.get(String.format(getUserIdUrl, accessToken, code));
- log.debug("获取企业微信UserId响应报文:{}", userResult);
- // 企微唯一标识
- return JSONUtil.parseObj(userResult).getStr("userid");
- }
+ String accessTokenResult = HttpUtil
+ .get(String.format(getAccessTokenUrl, socialDetails.getAppId(), socialDetails.getAppSecret()));
+ log.debug("获取企业微信Token响应报文：{}", accessTokenResult);
+ String accessToken = JSONUtil.parseObj(accessTokenResult).getStr("access_token");
+ String getUserIdUrl = "https://qyapi.weixin.qq.com/cgi-bin/auth/getuserinfo?access_token=%s&code=%s";
+ String userResult = HttpUtil.get(String.format(getUserIdUrl, accessToken, code));
+ log.debug("获取企业微信UserId响应报文:{}", userResult);
+ // 企微唯一标识
+ return JSONUtil.parseObj(userResult).getStr("userid");
+ }
- /**
- * openId 获取用户信息
- * @param openId
- * @return
- */
- @Override
- public UserInfo info(String openId) {
- SysUser user = sysUserService.getOne(Wrappers.query().lambda().eq(SysUser::getWxCpUserid, openId));
+ /**
+ * openId 获取用户信息
+ *
+ * @param openId
+ * @return
+ */
+ @Override
+ public UserInfo info(String openId) {
+ SysUser user = sysUserService.getOne(Wrappers.query().lambda().eq(SysUser::getWxCpUserid, openId));
- if (user == null) {
- log.info("企业微信未绑定:{}", openId);
- return null;
- }
- return sysUserService.findUserInfo(user);
- }
+ if (user == null) {
+ log.info("企业微信未绑定:{}", openId);
+ return null;
+ }
+ return sysUserService.findUserInfo(user);
+ }
- /**
- * 绑定逻辑
- * @param user 用户实体
- * @param identify 渠道返回唯一标识
- * @return
- */
- @Override
- public Boolean bind(SysUser user, String identify) {
- user.setWxCpUserid(identify);
- sysUserService.updateById(user);
- return true;
- }
+ /**
+ * 绑定逻辑
+ *
+ * @param user 用户实体
+ * @param identify 渠道返回唯一标识
+ * @return
+ */
+ @Override
+ public Boolean bind(SysUser user, String identify) {
+ user.setWxCpUserid(identify);
+ sysUserService.updateById(user);
+ return true;
+ }
}
diff --git a/pigx-upms/pigx-upms-biz/src/main/java/com/pig4cloud/pigx/admin/service/impl/SysUserServiceImpl.java b/pigx-upms/pigx-upms-biz/src/main/java/com/pig4cloud/pigx/admin/service/impl/SysUserServiceImpl.java
index 78af38628..7114427ea 100644
--- a/pigx-upms/pigx-upms-biz/src/main/java/com/pig4cloud/pigx/admin/service/impl/SysUserServiceImpl.java
+++ b/pigx-upms/pigx-upms-biz/src/main/java/com/pig4cloud/pigx/admin/service/impl/SysUserServiceImpl.java
@@ -42,6 +42,7 @@ import com.pig4cloud.pigx.admin.service.*;
import com.pig4cloud.pigx.common.audit.annotation.Audit;
import com.pig4cloud.pigx.common.core.constant.CacheConstants;
import com.pig4cloud.pigx.common.core.constant.CommonConstants;
+import com.pig4cloud.pigx.common.core.constant.enums.LoginTypeEnum;
import com.pig4cloud.pigx.common.core.exception.ErrorCodes;
import com.pig4cloud.pigx.common.core.util.MsgUtils;
import com.pig4cloud.pigx.common.core.util.R;
@@ -498,32 +499,30 @@ public class SysUserServiceImpl extends ServiceImpl impl
public R unbinding(String type) {
PigxUser user = SecurityUtils.getUser();
LambdaUpdateWrapper wrapper = null;
- if (type.equals("wechat")) {
+
+ // 微信开放平台 （普通用户扫码登录）
+ if (type.equals(LoginTypeEnum.WECHAT.getType())) {
wrapper = Wrappers.lambdaUpdate()
.set(SysUser::getWxOpenid, null)
.eq(SysUser::getUserId, user.getId());
- } else if (type.equals("gitee")) {
+ // 码云登录 （方便申请）
+ } else if (type.equals(LoginTypeEnum.GITEE.getType())) {
wrapper = Wrappers.lambdaUpdate()
.set(SysUser::getGiteeLogin, null)
.eq(SysUser::getUserId, user.getId());
- } else if (type.equals("osc")) {
- wrapper = Wrappers.lambdaUpdate()
- .set(SysUser::getOscId, null)
- .eq(SysUser::getUserId, user.getId());
- } else if (type.equals("tencent")) {
- wrapper = Wrappers.lambdaUpdate()
- .set(SysUser::getQqOpenid, null)
- .eq(SysUser::getUserId, user.getId());
- } else if (type.equals("cp")) {
+ // 企业微信登录
+ } else if (type.equals(LoginTypeEnum.WEIXIN_CP.getType())) {
wrapper = Wrappers.lambdaUpdate()
.set(SysUser::getWxCpUserid, null)
.eq(SysUser::getUserId, user.getId());
- } else if (type.equals("dingding")) {
+ // 钉钉登录
+ } else if (type.equals(LoginTypeEnum.DINGTALK.getType())) {
wrapper = Wrappers.lambdaUpdate()
.set(SysUser::getWxDingUserid, null)
.eq(SysUser::getUserId, user.getId());
}
- if (wrapper == null) {
+
+ if (Objects.isNull(wrapper)) {
return R.failed("解绑账号类型不存在");
}
this.update(wrapper);