From caa2b51bb4133f5a2e59eac242e744fb57f864a0 Mon Sep 17 00:00:00 2001 From: 无关风月 <443237572@qq.com> Date: 星期二, 10 九月 2024 13:01:44 +0800 Subject: [PATCH] Merge branch 'master' of http://120.76.84.145:10101/gitblit/r/java/mx_charging_pile --- ruoyi-service/ruoyi-account/src/main/java/com/ruoyi/account/service/impl/TAppUserServiceImpl.java | 103 ++++++++++++++++++++++++++++++++++++++++++++++----- 1 files changed, 92 insertions(+), 11 deletions(-) diff --git a/ruoyi-service/ruoyi-account/src/main/java/com/ruoyi/account/service/impl/TAppUserServiceImpl.java b/ruoyi-service/ruoyi-account/src/main/java/com/ruoyi/account/service/impl/TAppUserServiceImpl.java index 912a5e0..a464df6 100644 --- a/ruoyi-service/ruoyi-account/src/main/java/com/ruoyi/account/service/impl/TAppUserServiceImpl.java +++ b/ruoyi-service/ruoyi-account/src/main/java/com/ruoyi/account/service/impl/TAppUserServiceImpl.java @@ -1,5 +1,8 @@ package com.ruoyi.account.service.impl; +import com.alipay.api.response.AlipaySystemOauthTokenResponse; +import com.alipay.api.response.AlipayUserInfoShareResponse; +import com.alipay.api.response.AlipayUserUserinfoShareResponse; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.toolkit.ObjectUtils; import com.baomidou.mybatisplus.core.toolkit.Wrappers; @@ -7,8 +10,10 @@ import com.ruoyi.account.api.model.TAppUser; import com.ruoyi.account.mapper.TAppUserMapper; import com.ruoyi.account.service.TAppUserService; +import com.ruoyi.account.service.TInviteUserService; import com.ruoyi.account.wx.model.WeixinProperties; import com.ruoyi.account.wx.pojo.AppletUserDecodeData; +import com.ruoyi.common.core.exception.ServiceException; import com.ruoyi.common.security.service.TokenService; import com.ruoyi.system.api.model.LoginUserApplet; import org.springframework.beans.factory.annotation.Autowired; @@ -31,35 +36,99 @@ public class TAppUserServiceImpl extends ServiceImpl<TAppUserMapper, TAppUser> implements TAppUserService { @Autowired private TokenService tokenService; + @Autowired + private TInviteUserService inviteUserService; @Override - public Map<String, Object> login(AppletUserDecodeData appletUserDecodeData) { - // 通过手机号查询用户,是否已存在手动导入用户 + public Map<String, Object> wxLogin(AppletUserDecodeData appletUserDecodeData,Long inviteUserId) { + // 通过手机号查询用户,是否已存在手动导入用户,包含支付宝用户 TAppUser appUser = this.getOne(Wrappers.lambdaQuery(TAppUser.class) .eq(TAppUser::getPhone, appletUserDecodeData.getPhoneNumber()) .isNull(TAppUser::getWxOpenid) .last("LIMIT 1")); - LambdaQueryWrapper<TAppUser> wrapper = Wrappers.lambdaQuery(TAppUser.class) - .eq(TAppUser::getWxOpenid, appletUserDecodeData.getOpenId()); if(Objects.isNull(appUser)){ // 先使用openId和当前手机号进行查询 - wrapper.eq(TAppUser::getPhone, appletUserDecodeData.getPhoneNumber()) - .last("LIMIT 1"); - appUser = this.getOne(wrapper); + appUser = this.getOne(Wrappers.lambdaQuery(TAppUser.class) + .eq(TAppUser::getWxOpenid, appletUserDecodeData.getOpenId()) + .eq(TAppUser::getPhone, appletUserDecodeData.getPhoneNumber()) + .last("LIMIT 1")); if(Objects.isNull(appUser)){ appUser = new TAppUser(); appUser.setPhone(appletUserDecodeData.getPhoneNumber()); } - }else { - wrapper.last("LIMIT 1"); - // 删除小程序原有授权用户 - this.remove(wrapper); } + if(Objects.nonNull(appUser.getStatus())){ + throwInfo(appUser.getStatus()); + } + appUser.setInviteUserId(inviteUserId); appUser.setAvatar(appletUserDecodeData.getAvatarUrl()); appUser.setCity(appletUserDecodeData.getCity()); appUser.setName(appletUserDecodeData.getNickName()); appUser.setProvince(appletUserDecodeData.getProvince()); appUser.setWxOpenid(appletUserDecodeData.getOpenId()); this.saveOrUpdate(appUser); + inviteUserService.saveInviteUser(appUser.getId(), inviteUserId); + return this.getUserInfo(appUser); + } + + @Override + public Map<String, Object> aliLogin(AlipaySystemOauthTokenResponse response,String phone,Long inviteUserId) { + // 通过手机号查询用户,是否已存在手动导入用户,包含微信用户 + TAppUser appUser = this.getOne(Wrappers.lambdaQuery(TAppUser.class) + .eq(TAppUser::getPhone, phone) + .isNull(TAppUser::getAliOpenid) + .last("LIMIT 1")); + if(Objects.isNull(appUser)){ + // 先使用openId和当前手机号进行查询 + appUser = this.getOne(Wrappers.lambdaQuery(TAppUser.class) + .eq(TAppUser::getAliOpenid, response.getOpenId()) + .eq(TAppUser::getPhone, phone) + .last("LIMIT 1")); + if(Objects.isNull(appUser)){ + appUser = new TAppUser(); + appUser.setPhone(phone); + } + } + if(Objects.nonNull(appUser.getStatus())){ + throwInfo(appUser.getStatus()); + } + appUser.setInviteUserId(inviteUserId); + appUser.setAliOpenid(response.getOpenId()); + this.saveOrUpdate(appUser); + inviteUserService.saveInviteUser(appUser.getId(), inviteUserId); + return this.getUserInfo(appUser); + } +// @Override +// public Map<String, Object> aliLogin(AlipaySystemOauthTokenResponse response, AlipayUserInfoShareResponse userInfo) { +// // 通过手机号查询用户,是否已存在手动导入用户,包含微信用户 +// TAppUser appUser = this.getOne(Wrappers.lambdaQuery(TAppUser.class) +// .eq(TAppUser::getPhone, userInfo.getMobile()) +// .isNull(TAppUser::getAliOpenid) +// .last("LIMIT 1")); +// if(Objects.isNull(appUser)){ +// // 先使用openId和当前手机号进行查询 +// appUser = this.getOne(Wrappers.lambdaQuery(TAppUser.class) +// .eq(TAppUser::getAliOpenid, response.getOpenId()) +// .eq(TAppUser::getPhone, userInfo.getMobile()) +// .last("LIMIT 1")); +// if(Objects.isNull(appUser)){ +// appUser = new TAppUser(); +// appUser.setPhone(userInfo.getMobile()); +// } +// } +// if(Objects.nonNull(appUser.getStatus())){ +// throwInfo(appUser.getStatus()); +// } +// appUser.setAvatar(userInfo.getAvatar()); +// appUser.setCity(userInfo.getCity()); +// appUser.setName(userInfo.getNickName()); +// appUser.setProvince(userInfo.getProvince()); +// appUser.setAliOpenid(response.getOpenId()); +// this.saveOrUpdate(appUser); +// return this.getUserInfo(appUser); +// } + + @Override + public Map<String, Object> getUserInfo(TAppUser appUser) { LoginUserApplet loginUserApplet = new LoginUserApplet(); if(ObjectUtils.isNotNull(appUser)){ loginUserApplet.setUserId(appUser.getId()); @@ -73,4 +142,16 @@ tokenInfos.put("info",loginUserApplet); return tokenInfos; } + + @Override + public void throwInfo(Integer status){ + switch (status){ + case 2: + throw new ServiceException("账号被冻结,请联系管理员"); + case 3: +// throw new ServiceException("账号已注销,请重新注册使用"); + default: + break; + } + } } -- Gitblit v1.7.1