抽离尝试保存用户信息方法，在用户绑定与发送关键字时执行

Author: imldy

src/main/java/com/github/binarywang/demo/wx/mp/handler/MsgHandler.java

@@ -1,13 +1,16 @@
 package com.github.binarywang.demo.wx.mp.handler;
 
 import com.github.binarywang.demo.wx.mp.builder.TextBuilder;
+import com.github.binarywang.demo.wx.mp.service.WxMpUserService;
 import com.github.binarywang.demo.wx.mp.utils.JsonUtils;
 import me.chanjar.weixin.common.error.WxErrorException;
 import me.chanjar.weixin.common.session.WxSessionManager;
 import me.chanjar.weixin.mp.api.WxMpService;
 import me.chanjar.weixin.mp.bean.message.WxMpXmlMessage;
 import me.chanjar.weixin.mp.bean.message.WxMpXmlOutMessage;
+import me.chanjar.weixin.mp.bean.result.WxMpUser;
 import org.apache.commons.lang3.StringUtils;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Component;
 
 import java.util.Map;
@@ -19,7 +22,8 @@
  */
 @Component
 public class MsgHandler extends AbstractHandler {
-
+    @Autowired
+    public WxMpUserService wxMpUserService;
     @Override
     public WxMpXmlOutMessage handle(WxMpXmlMessage wxMessage,
                                     Map<String, Object> context, WxMpService weixinService,
@@ -29,23 +33,21 @@ public WxMpXmlOutMessage handle(WxMpXmlMessage wxMessage,
             //TODO 可以选择将消息保存到本地
         }
 
+        String resultContent = "欢迎关注";
         //当用户输入关键词如“你好”，“客服”等，并且有客服在线时，把消息转发给在线客服
         try {
-            if (StringUtils.startsWithAny(wxMessage.getContent(), "你好", "客服")
-                && weixinService.getKefuService().kfOnlineList()
-                .getKfOnlineList().size() > 0) {
-                return WxMpXmlOutMessage.TRANSFER_CUSTOMER_SERVICE()
-                    .fromUser(wxMessage.getToUser())
-                    .toUser(wxMessage.getFromUser()).build();
+            if (StringUtils.startsWith(wxMessage.getContent(), "绑定")) {
+                this.logger.info("用户发送发了绑定: {}", wxMessage.getFromUser());
+                WxMpUser userWxInfo = weixinService.getUserService()
+                    .userInfo(wxMessage.getFromUser(), null);
+                boolean result = wxMpUserService.trySave(userWxInfo);
+                resultContent = result ? "绑定推送成功" : "绑定推送失败";
             }
         } catch (WxErrorException e) {
             e.printStackTrace();
         }
 
-        //TODO 组装回复消息
-        String content = "收到信息内容：" + JsonUtils.toJson(wxMessage);
-
-        return new TextBuilder().build(content, wxMessage, weixinService);
+        return new TextBuilder().build(resultContent, wxMessage, weixinService);
 
     }
 

src/main/java/com/github/binarywang/demo/wx/mp/handler/SubscribeHandler.java

@@ -2,8 +2,7 @@
 
 import java.util.Map;
 
-import com.github.binarywang.demo.wx.mp.eneity.UserWxInfoPO;
-import com.github.binarywang.demo.wx.mp.repository.UserWxInfoRepository;
+import com.github.binarywang.demo.wx.mp.service.WxMpUserService;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Component;
 
@@ -21,7 +20,7 @@
 @Component
 public class SubscribeHandler extends AbstractHandler {
     @Autowired
-    public UserWxInfoRepository userWxInfoRepository;
+    public WxMpUserService wxMpUserService;
 
     @Override
     public WxMpXmlOutMessage handle(WxMpXmlMessage wxMessage,
@@ -35,28 +34,8 @@ public WxMpXmlOutMessage handle(WxMpXmlMessage wxMessage,
             WxMpUser userWxInfo = weixinService.getUserService()
                 .userInfo(wxMessage.getFromUser(), null);
             if (userWxInfo != null) {
-                userWxInfo.setUnionId("oKOdI67B5_qwzzLJttiCnA86lBa8");
                 this.logger.debug("获取到用户信息[{}]", userWxInfo.getUnionId());
-                boolean edit = false;
-                UserWxInfoPO userWxInfoPO = userWxInfoRepository.findFirstByUid(userWxInfo.getUnionId());
-                if (userWxInfoPO == null) {
-                    userWxInfoPO = new UserWxInfoPO();
-                    userWxInfoPO.setUid(userWxInfo.getUnionId());
-                    userWxInfoPO.setMpOpenID(userWxInfo.getOpenId());
-                    edit = true;
-                } else {
-                    if (userWxInfoPO.getMpOpenID() == null) {
-                        userWxInfoPO.setMpOpenID(userWxInfo.getOpenId());
-                        edit = true;
-                    }
-                    if (userWxInfoPO.getUid() == null) {
-                        userWxInfoPO.setUid(userWxInfo.getUnionId());
-                        edit = true;
-                    }
-                }
-                if (edit) {
-                    UserWxInfoPO save = userWxInfoRepository.save(userWxInfoPO);
-                }
+                wxMpUserService.trySave(userWxInfo);
             }
         } catch (WxErrorException e) {
             if (e.getError().getErrorCode() == 48001) {

src/main/java/com/github/binarywang/demo/wx/mp/service/WxMpUserService.java

@@ -0,0 +1,41 @@
+package com.github.binarywang.demo.wx.mp.service;
+
+import com.github.binarywang.demo.wx.mp.eneity.UserWxInfoPO;
+import com.github.binarywang.demo.wx.mp.repository.UserWxInfoRepository;
+import me.chanjar.weixin.mp.bean.result.WxMpUser;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Service;
+
+/**
+ * @author imldy
+ * @date 2023/03/04 15:24
+ **/
+@Service
+public class WxMpUserService {
+    @Autowired
+    public UserWxInfoRepository userWxInfoRepository;
+
+    public boolean trySave(WxMpUser userWxInfo) {
+        boolean edit = false;
+        UserWxInfoPO userWxInfoPO = userWxInfoRepository.findFirstByUid(userWxInfo.getUnionId());
+        if (userWxInfoPO == null) {
+            userWxInfoPO = new UserWxInfoPO();
+            userWxInfoPO.setUid(userWxInfo.getUnionId());
+            userWxInfoPO.setMpOpenID(userWxInfo.getOpenId());
+            edit = true;
+        } else {
+            if (userWxInfoPO.getMpOpenID() == null) {
+                userWxInfoPO.setMpOpenID(userWxInfo.getOpenId());
+                edit = true;
+            }
+            if (userWxInfoPO.getUid() == null) {
+                userWxInfoPO.setUid(userWxInfo.getUnionId());
+                edit = true;
+            }
+        }
+        if (edit) {
+            UserWxInfoPO save = userWxInfoRepository.save(userWxInfoPO);
+        }
+        return true;
+    }
+}