新增客户项目

zxkf-service/src/main/java/api/service/websocket/WebSocket.java

@@ -16,6 +16,7 @@ import me.chanjar.weixin.mp.api.WxMpService;
 import me.chanjar.weixin.mp.bean.kefu.WxMpKefuMessage;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Component;
+import org.springframework.web.bind.annotation.RequestParam;
 
 import javax.annotation.PostConstruct;
 import javax.websocket.*;
@@ -71,11 +72,14 @@ public class WebSocket {
                     servicerQueue.add(userCode);
                 }
             } else {
-                reply=customers.containsKey(userCode);
-                if(!reply) {
+                reply = customers.containsKey(userCode);
+                if (!reply) {
                     customers.put(userCode, source);
                     khsessionPool.put(userCode, session);
-                    addCustomer(userCode,source);
+                    Long pid = null;
+                    List<String> ps = session.getRequestParameterMap().get("project");
+                    if (!ps.isEmpty()) pid = Long.valueOf(ps.get(0));
+                    addCustomer(userCode, source, pid);
                 }
             }
             if(reply) {
@@ -464,7 +468,7 @@ public class WebSocket {
 
         if (!customers.containsKey(sendUser)) {
             customers.put(sendUser, 2);
-            addCustomer(sendUser,2);
+            addCustomer(sendUser,2,null);
         };
 
         String kf = khkfs.get(sendUser);
@@ -491,13 +495,14 @@ public class WebSocket {
         webSocket.messageService.insert(message);
     }
 
-    public void addCustomer(String userCode,Integer source) {
+    public void addCustomer(String userCode,Integer source,Long project) {
         LambdaQueryWrapper<Customer> qw = new LambdaQueryWrapper();
         qw.eq(Customer::getCustomerNo, userCode);
         if (!webSocket.customerService.exists(qw)) {
             Customer customer = new Customer();
             customer.setCustomerNo(userCode);
             customer.setType(Long.valueOf(source));
+            customer.setProject(project);
             customer.setFCreatetime(new Date());
             customer.setFIsdelete(0L);
             webSocket.customerService.insert(customer);