From d9e6bd9c7558ccb4bcce44092f3015b331f2f696 Mon Sep 17 00:00:00 2001
From: sxu <602087911@qq.com>
Date: Wed, 25 Dec 2024 14:27:55 +0800
Subject: [PATCH] =?UTF-8?q?=E5=BE=85=E5=8A=9E?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../material/order/service/impl/OrderServiceImpl.java    | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/bonus-modules/bonus-material-mall/src/main/java/com/bonus/material/order/service/impl/OrderServiceImpl.java b/bonus-modules/bonus-material-mall/src/main/java/com/bonus/material/order/service/impl/OrderServiceImpl.java
index 1ca77b6..2c65067 100644
--- a/bonus-modules/bonus-material-mall/src/main/java/com/bonus/material/order/service/impl/OrderServiceImpl.java
+++ b/bonus-modules/bonus-material-mall/src/main/java/com/bonus/material/order/service/impl/OrderServiceImpl.java
@@ -9,6 +9,7 @@ import com.bonus.common.security.utils.SecurityUtils;
 import com.bonus.material.comprehensive.entity.RentDetailDto;
 import com.bonus.material.device.domain.DevInfo;
 import com.bonus.material.device.domain.vo.DevInfoVo;
+import com.bonus.material.device.mapper.DevInfoMapper;
 import com.bonus.material.largeScreen.entity.OrderData;
 import com.bonus.material.lease.domain.LeaseRepair;
 import com.bonus.material.lease.domain.LeaseRepairRecord;
@@ -42,6 +43,9 @@ public class OrderServiceImpl implements OrderService {
     @Resource
     private TmTaskMapper tmTaskMapper;
 
+    @Resource
+    DevInfoMapper devInfoMapper;
+
     @Resource
     private LeaseRepairRecordMapper leaseRepairRecordMapper;
 
@@ -100,7 +104,10 @@ public class OrderServiceImpl implements OrderService {
             tmTask.setCreateBy(SecurityUtils.getLoginUser().getUsername());
             tmTask.setCreateTime(DateUtils.getNowDate());
             tmTask.setBuyerCompanyId(SecurityUtils.getLoginUser().getSysUser().getCompanyId());
-            tmTask.setSellerCompanyId(Long.valueOf(orderInfoDto.getSellerCompany()));
+            if (!CollectionUtils.isEmpty(orderDetailDtos)) {
+                DevInfoVo devInfoVo = devInfoMapper.selectDevInfoByMaId(Long.valueOf(orderDetailDtos.get(0).getMaId()));
+                tmTask.setSellerCompanyId(Long.valueOf(devInfoVo.getCompanyId()));
+            }
             tmTaskMapper.insertSelective(tmTask);
         }
         return j;