From 2f6341d8233052c3f4424211c93c37d18afdafeb Mon Sep 17 00:00:00 2001
From: lzttt <1803073139@qq.com>
Date: Fri, 25 Oct 2024 18:19:55 +0800
Subject: [PATCH] =?UTF-8?q?=E6=95=91=E6=8F=B4=E5=8F=B8=E6=9C=BA=E7=AB=AF?=
 =?UTF-8?q?=E5=AE=A2=E6=88=B7=E4=BF=A1=E6=81=AF=E5=90=8C=E6=AD=A5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../impl/InspectionInfoServiceImpl.java       | 13 ++++++-
 .../impl/PartnerCustomerInfoServiceImpl.java  |  2 +-
 .../impl/RescueDriverInfoServiceImpl.java     | 38 ++++++++++++++++++-
 3 files changed, 50 insertions(+), 3 deletions(-)

diff --git a/dl-module-inspection/src/main/java/cn/iocoder/yudao/module/inspection/service/impl/InspectionInfoServiceImpl.java b/dl-module-inspection/src/main/java/cn/iocoder/yudao/module/inspection/service/impl/InspectionInfoServiceImpl.java
index 04941ab3..2e68139d 100644
--- a/dl-module-inspection/src/main/java/cn/iocoder/yudao/module/inspection/service/impl/InspectionInfoServiceImpl.java
+++ b/dl-module-inspection/src/main/java/cn/iocoder/yudao/module/inspection/service/impl/InspectionInfoServiceImpl.java
@@ -9,6 +9,8 @@ import cn.hutool.core.date.DateUtil;
 import cn.hutool.core.util.ObjectUtil;
 import cn.iocoder.yudao.framework.security.core.LoginUser;
 import cn.iocoder.yudao.framework.security.core.util.SecurityFrameworkUtils;
+import cn.iocoder.yudao.module.custom.entity.CustomerMain;
+import cn.iocoder.yudao.module.custom.service.CustomerMainService;
 import cn.iocoder.yudao.module.partner.entity.PartnerCustomerInfo;
 import cn.iocoder.yudao.module.partner.service.IPartnerCustomerInfoService;
 import cn.iocoder.yudao.module.partner.service.IPartnerWorkerService;
@@ -38,6 +40,8 @@ import org.springframework.context.annotation.Lazy;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
+import javax.annotation.Resource;
+
 /**
  * 【请填写功能名称】Service业务层处理
  *
@@ -68,7 +72,8 @@ public class InspectionInfoServiceImpl extends ServiceImpl<InspectionInfoMapper,
     private IDelOrderInfoService delInspectionOrderService;
     @Autowired
     private IDelInspectionInfoService delInspectionInfoService;
-
+    @Resource
+    private CustomerMainService customerMainService;
     /**
      * 查询【请填写功能名称】
      *
@@ -195,6 +200,12 @@ public class InspectionInfoServiceImpl extends ServiceImpl<InspectionInfoMapper,
             customerInfo.setCustomerName(user.getNickname());
             customerInfo.setSex("0");
             customerInfo.setUserAge(user.getUserAge());
+
+            // 设置客户来源
+            CustomerMain customerMain = new CustomerMain();
+            customerMain.setDataFrom("01");
+            customerMainService.saveOrUpdate(customerMain);
+
             customerInfoService.insertPartnerCustomerInfo(customerInfo);
         }
         //追加订单明细记录
diff --git a/dl-module-inspection/src/main/java/cn/iocoder/yudao/module/partner/service/impl/PartnerCustomerInfoServiceImpl.java b/dl-module-inspection/src/main/java/cn/iocoder/yudao/module/partner/service/impl/PartnerCustomerInfoServiceImpl.java
index f30e9a68..11155c96 100644
--- a/dl-module-inspection/src/main/java/cn/iocoder/yudao/module/partner/service/impl/PartnerCustomerInfoServiceImpl.java
+++ b/dl-module-inspection/src/main/java/cn/iocoder/yudao/module/partner/service/impl/PartnerCustomerInfoServiceImpl.java
@@ -248,7 +248,7 @@ public class PartnerCustomerInfoServiceImpl extends ServiceImpl<PartnerCustomerI
         customerMain.setIsHangAccount("0");
         customerMain.setTypeCode("01");
         // 设置客户初始来源
-        customerMain.setDataFrom("01");
+        //customerMain.setDataFrom("01");
         // 设置注册方式
         customerMain.setInviterType("01");
 
diff --git a/dl-module-rescue/src/main/java/cn/iocoder/yudao/module/rescue/service/impl/RescueDriverInfoServiceImpl.java b/dl-module-rescue/src/main/java/cn/iocoder/yudao/module/rescue/service/impl/RescueDriverInfoServiceImpl.java
index 6653d5a5..372f0a7a 100644
--- a/dl-module-rescue/src/main/java/cn/iocoder/yudao/module/rescue/service/impl/RescueDriverInfoServiceImpl.java
+++ b/dl-module-rescue/src/main/java/cn/iocoder/yudao/module/rescue/service/impl/RescueDriverInfoServiceImpl.java
@@ -3,9 +3,14 @@ package cn.iocoder.yudao.module.rescue.service.impl;
 import cn.hutool.core.collection.CollectionUtil;
 import cn.hutool.core.util.CoordinateUtil;
 import cn.hutool.http.HttpUtil;
+import cn.iocoder.yudao.module.custom.entity.CustomerMain;
+import cn.iocoder.yudao.module.custom.service.CustomerMainService;
+import cn.iocoder.yudao.module.partner.entity.PartnerCustomerInfo;
+import cn.iocoder.yudao.module.partner.service.IPartnerCustomerInfoService;
 import cn.iocoder.yudao.module.rescue.domain.*;
 import cn.iocoder.yudao.module.rescue.mapper.RescueDriverInfoMapper;
 import cn.iocoder.yudao.module.rescue.utils.RedissonDelayQueue;
+import cn.iocoder.yudao.module.shop.entity.ShopUserCar;
 import cn.iocoder.yudao.module.system.api.dict.DictDataApi;
 import cn.iocoder.yudao.module.system.api.user.AdminUserApi;
 import cn.iocoder.yudao.module.system.api.user.dto.AdminUserRespDTO;
@@ -72,6 +77,10 @@ public class RescueDriverInfoServiceImpl extends ServiceImpl<RescueDriverInfoMap
     private WechatPayConfig wechatPayConfig;
     @Resource
     private RedissonDelayQueue redissonDelayQueue;
+    @Autowired
+    private IPartnerCustomerInfoService customerInfoService;
+    @Resource
+    private CustomerMainService customerMainService;
 
     public static String Redis_Driver_Key = "Rescue:Driver:";
     public static String Redis_Driver_Position_Key = "DriverPosition:";
@@ -581,6 +590,33 @@ public class RescueDriverInfoServiceImpl extends ServiceImpl<RescueDriverInfoMap
         //处理redis
         DriverInfo driverInfo = driverInfoService.getById(rescueInfo.getDriverId());
         AdminUserRespDTO user = userService.getUser(driverInfo.getUserId());
+
+        // 收集客户信息
+        PartnerCustomerInfo customerInfo = new PartnerCustomerInfo();
+        customerInfo.setCustomerName(rescueInfo.getConnectionName());
+        customerInfo.setCustomerPhone(rescueInfo.getConnectionPhone());
+        customerInfo.setUserId(rescueInfo.getUserId());
+
+        // 收集客户车辆信息
+        List<ShopUserCar> userCarList = new ArrayList<>();
+        ShopUserCar userCar = new ShopUserCar();
+        userCar.setCarNo(rescueInfo.getLicenseNum());
+        userCar.setCarModel(rescueInfo.getCarType());
+        userCar.setCarBrand(rescueInfo.getCarBrand());
+        userCar.setUserId(user.getId());
+        userCarList.add(userCar);
+
+        customerInfo.setUserCarList(userCarList);
+
+        // 设置客户来源
+        CustomerMain customerMain = new CustomerMain();
+        customerMain.setDataFrom("02");
+        customerMainService.saveOrUpdate(customerMain);
+
+        // 调用插入客户信息的方法
+        customerInfoService.insertPartnerCustomerInfo(customerInfo);
+
+
         //所在顶级机构
         String redisKey = Redis_Driver_Key + driverInfo.getTenantId() + ":" + rescueInfo.getDriverId();
         Object temp = redisCache.getCacheMapValue(redisKey, "rescueIds");
@@ -588,7 +624,7 @@ public class RescueDriverInfoServiceImpl extends ServiceImpl<RescueDriverInfoMap
             String s = temp.toString();
             if (StringUtils.isNotEmpty(s)) {
                 String resStr = "";
-                String[] rescueIds = temp.toString().split(",");
+                 String[] rescueIds = temp.toString().split(",");
                 for (String tmp : rescueIds) {
                     if (!tmp.equals(rescueId.toString())) {
                         resStr = resStr + "," + tmp;