From 8f00a10fa10eff8ded7374a87a582f2c2b42569b Mon Sep 17 00:00:00 2001
From: k94314517 <8417338+k94314517@user.noreply.gitee.com>
Date: 星期一, 15 七月 2024 16:33:04 +0800
Subject: [PATCH] 代码提交

---
 server/service/src/main/java/com/doumee/service/business/impl/MemberServiceImpl.java |   84 +++++++++++++++++++++++++++++++-----------
 1 files changed, 62 insertions(+), 22 deletions(-)

diff --git a/server/service/src/main/java/com/doumee/service/business/impl/MemberServiceImpl.java b/server/service/src/main/java/com/doumee/service/business/impl/MemberServiceImpl.java
index 481bbaf..0c3bae8 100644
--- a/server/service/src/main/java/com/doumee/service/business/impl/MemberServiceImpl.java
+++ b/server/service/src/main/java/com/doumee/service/business/impl/MemberServiceImpl.java
@@ -18,11 +18,14 @@
 import com.doumee.core.oss.FileModel;
 import com.doumee.core.utils.CodeVerifyUtils;
 import com.doumee.core.utils.Constants;
+import com.doumee.core.utils.DateUtil;
 import com.doumee.core.utils.Utils;
 import com.doumee.core.wx.WxMiniConfig;
 import com.doumee.dao.business.MemberMapper;
+import com.doumee.dao.business.UsersMapper;
 import com.doumee.dao.business.model.InterfaceLog;
 import com.doumee.dao.business.model.Member;
+import com.doumee.dao.business.model.Users;
 import com.doumee.dao.web.reqeust.EditMemberRequest;
 import com.doumee.dao.web.reqeust.WxPhoneRequest;
 import com.doumee.dao.web.response.AccountResponse;
@@ -35,6 +38,7 @@
 import okhttp3.OkHttpClient;
 import okhttp3.Request;
 import okhttp3.Response;
+import org.apache.catalina.User;
 import org.apache.commons.lang3.StringUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.data.redis.core.RedisTemplate;
@@ -56,6 +60,9 @@
 
     @Autowired
     private MemberMapper memberMapper;
+
+    @Autowired
+    private UsersMapper usersMapper;
 
     @Autowired
     private SystemDictDataBiz systemDictDataBiz;
@@ -365,29 +372,60 @@
 
 
 
-    /**
-     * 鏇存柊涓彴token淇℃伅
-     * @param member
-     */
-    private static void updUserZTToken(ZbomZhongTaiService zbomZhongTaiService,MemberMapper memberMapper,SystemDictDataBiz systemDictDataBiz,Member member){
-        //濡傛灉鏃犳湁鏁堟湡/宸茶繃鏈� 閲嶆柊鑾峰彇
-        if(Objects.isNull(member.getTokenDate()) || member.getTokenDate().getTime()<=System.currentTimeMillis() ){
-            ZTUserGetTokenRequest param = new ZTUserGetTokenRequest();
-            //鏌ヨAPPID
-            String appId = systemDictDataBiz.queryByCode(Constants.WX_MINI_CONFIG,Constants.WX_APPID_CUSTOMER).getCode();
-            param.setAppId(appId);
-            param.setOpenId(member.getOpenid());
-            ZTUserGetTokenResponse ztUserGetTokenResponse = zbomZhongTaiService.getUserToken(param);
-            if(Objects.nonNull(ztUserGetTokenResponse)){
-                member.setToken(ztUserGetTokenResponse.getToken());
-                //TODO 鏆傛棤杩囨湡鏃ユ湡
-                memberMapper.update(new UpdateWrapper<Member>().lambda().set(Member::getToken,ztUserGetTokenResponse.getToken())
-                        .eq(Member::getId,member.getId()));
-            }else{
-                throw new BusinessException(ResponseStatus.NOT_ALLOWED.getCode(),"涓彴TOKEN鑾峰彇澶辫触锛岃鑱旂郴绠$悊鍛�");
+
+    public static String getZTToken(ZbomZhongTaiService zbomZhongTaiService, MemberMapper memberMapper, UsersMapper usersMapper, SystemDictDataBiz systemDictDataBiz, Long id, String userType){
+        if(userType.equals(Constants.CUSTOMER)){
+            Member member = memberMapper.selectById(id);
+            if(Objects.isNull(member)){
+                throw new BusinessException(ResponseStatus.BAD_REQUEST);
             }
-        };
+            //濡傛灉鏃犳湁鏁堟湡/宸茶繃鏈� 閲嶆柊鑾峰彇
+            if(Objects.isNull(member.getTokenDate()) || member.getTokenDate().getTime()<=System.currentTimeMillis() ){
+                ZTUserGetTokenRequest param = new ZTUserGetTokenRequest();
+                //鏌ヨAPPID
+                String appId = systemDictDataBiz.queryByCode(Constants.WX_MINI_CONFIG,Constants.WX_APPID_CUSTOMER).getCode();
+                param.setAppId(appId);
+                param.setOpenId(member.getOpenid());
+                ZTUserGetTokenResponse ztUserGetTokenResponse = zbomZhongTaiService.getUserToken(param);
+                if(Objects.nonNull(ztUserGetTokenResponse)){
+                    member.setToken(ztUserGetTokenResponse.getToken());
+                    //TODO 鏆傛棤杩囨湡鏃ユ湡
+                    memberMapper.update(new UpdateWrapper<Member>().lambda().set(Member::getToken,ztUserGetTokenResponse.getToken())
+                            .eq(Member::getId,member.getId()));
+                    return ztUserGetTokenResponse.getToken();
+                }else{
+                    throw new BusinessException(ResponseStatus.NOT_ALLOWED.getCode(),"涓彴TOKEN鑾峰彇澶辫触锛岃鑱旂郴绠$悊鍛�");
+                }
+            }
+            return member.getToken();
+        }else if(userType.equals(Constants.BUSINESS)){
+            Users users = usersMapper.selectById(id);
+            if(Objects.isNull(users)){
+                throw new BusinessException(ResponseStatus.BAD_REQUEST);
+            }
+            if(Objects.isNull(users.getTokenDate()) || users.getTokenDate().getTime()<=System.currentTimeMillis() ){
+                ZTUserGetTokenRequest param = new ZTUserGetTokenRequest();
+                //鏌ヨAPPID
+                String appId = systemDictDataBiz.queryByCode(Constants.WX_MINI_CONFIG,Constants.WX_APPID_CUSTOMER).getCode();
+                param.setAppId(appId);
+                param.setOpenId(users.getOpenid());
+                ZTUserGetTokenResponse ztUserGetTokenResponse = zbomZhongTaiService.getUserToken(param);
+                if(Objects.nonNull(ztUserGetTokenResponse)){
+                    users.setToken(ztUserGetTokenResponse.getToken());
+                    //TODO 鏆傛棤杩囨湡鏃ユ湡
+                    usersMapper.update(new UpdateWrapper<Users>().lambda().set(Users::getToken,ztUserGetTokenResponse.getToken())
+                            .eq(Users::getId,users.getId()));
+                    return ztUserGetTokenResponse.getToken();
+                }else{
+                    throw new BusinessException(ResponseStatus.NOT_ALLOWED.getCode(),"涓彴TOKEN鑾峰彇澶辫触锛岃鑱旂郴绠$悊鍛�");
+                }
+            }
+            return users.getToken();
+        }else{
+            throw new BusinessException(ResponseStatus.NOT_ALLOWED.getCode(),"鍙傛暟閿欒");
+        }
     }
+
 
 
 
@@ -401,7 +439,7 @@
             throw new BusinessException(ResponseStatus.NOT_ALLOWED.getCode(),"鐢ㄦ埛宸叉敞閿�");
         }
         //鏍¢獙鐢ㄦ埛token淇℃伅
-        MemberServiceImpl.updUserZTToken(zbomZhongTaiService,memberMapper,systemDictDataBiz,member);
+        MemberServiceImpl.getZTToken(zbomZhongTaiService,memberMapper,usersMapper,systemDictDataBiz,member.getId(),Constants.CUSTOMER);
         //璋冭捣涓彴娉ㄩ攢鎺ュ彛
         ZTBaseRequst ztBaseRequst = new ZTBaseRequst();
         ztBaseRequst.setUserType(Constants.CUSTOMER);
@@ -420,4 +458,6 @@
     }
 
 
+
+
 }

--
Gitblit v1.9.3