From af73f49872f899a1cb301ae6e3ee23ebc28daf20 Mon Sep 17 00:00:00 2001
From: liyong <18434998025@163.com>
Date: 星期一, 23 六月 2025 17:15:07 +0800
Subject: [PATCH] 2025-06-23 设备台账接口对接完成

---
 src/main/java/com/ruoyi/device/service/impl/DeviceLedgerServiceImpl.java |   51 ++++++++++++++++++++++++++++++++++++---------------
 1 files changed, 36 insertions(+), 15 deletions(-)

diff --git a/src/main/java/com/ruoyi/device/service/impl/DeviceLedgerServiceImpl.java b/src/main/java/com/ruoyi/device/service/impl/DeviceLedgerServiceImpl.java
index ee8d002..006ed90 100644
--- a/src/main/java/com/ruoyi/device/service/impl/DeviceLedgerServiceImpl.java
+++ b/src/main/java/com/ruoyi/device/service/impl/DeviceLedgerServiceImpl.java
@@ -5,11 +5,13 @@
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.ruoyi.common.utils.bean.BeanUtils;
 import com.ruoyi.common.utils.poi.ExcelUtil;
-import com.ruoyi.device.dto.DeviceLedgerExeclDto;
+import com.ruoyi.device.dto.DeviceLedgerDto;
+import com.ruoyi.device.execl.DeviceLedgerExeclDto;
 import com.ruoyi.device.mapper.DeviceLedgerMapper;
 import com.ruoyi.device.pojo.DeviceLedger;
 import com.ruoyi.device.service.IDeviceLedgerService;
 import com.ruoyi.framework.web.domain.AjaxResult;
+import com.ruoyi.project.system.mapper.SysUserMapper;
 import lombok.AllArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -27,9 +29,11 @@
 
     @Autowired
     private DeviceLedgerMapper deviceLedgerMapper;
+    @Autowired
+    private SysUserMapper sysUserMapper;
 
     @Override
-    public IPage<DeviceLedger> queryPage(Page page, DeviceLedger deviceLedger) {
+    public IPage<DeviceLedgerDto> queryPage(Page page, DeviceLedgerDto deviceLedger) {
 
         return deviceLedgerMapper.queryPage(page, deviceLedger);
     }
@@ -53,18 +57,35 @@
 
     @Override
     public void export(HttpServletResponse response, Long[] ids) {
-        ArrayList<Long> arrayList = new ArrayList<>();
-        Arrays.stream(ids).map(id -> {
-            return arrayList.add( id);
-        });
-        List<DeviceLedger> supplierManageList = deviceLedgerMapper.selectBatchIds(arrayList);
-        ArrayList<DeviceLedgerExeclDto> deviceLedgerExeclDtos = new ArrayList<>();
-        supplierManageList.stream().forEach(deviceLedger -> {
-            DeviceLedgerExeclDto deviceLedgerExeclDto = new DeviceLedgerExeclDto();
-            BeanUtils.copyProperties(deviceLedger,deviceLedgerExeclDto);
-            deviceLedgerExeclDtos.add(deviceLedgerExeclDto);
-        });
-        ExcelUtil<DeviceLedgerExeclDto> util = new ExcelUtil<DeviceLedgerExeclDto>(DeviceLedgerExeclDto.class);
-        util.exportExcel(response, deviceLedgerExeclDtos, "璁惧鍙拌处瀵煎嚭");
+
+        if (ids == null || ids.length == 0) {
+            List<DeviceLedger> supplierManageList = this.list();
+
+            ArrayList<DeviceLedgerExeclDto> deviceLedgerExeclDtos = new ArrayList<>();
+            supplierManageList.stream().forEach(deviceLedger -> {
+                DeviceLedgerExeclDto deviceLedgerExeclDto = new DeviceLedgerExeclDto();
+                BeanUtils.copyProperties(deviceLedger,deviceLedgerExeclDto);
+                deviceLedgerExeclDto.setCreateUser(sysUserMapper.selectUserById(Long.valueOf(deviceLedger.getCreateUser().toString())).getUserName());
+                deviceLedgerExeclDtos.add(deviceLedgerExeclDto);
+            });
+            ExcelUtil<DeviceLedgerExeclDto> util = new ExcelUtil<DeviceLedgerExeclDto>(DeviceLedgerExeclDto.class);
+            util.exportExcel(response, deviceLedgerExeclDtos, "璁惧鍙拌处瀵煎嚭");
+        }else  {
+            ArrayList<Long> arrayList = new ArrayList<>();
+            Arrays.stream(ids).map(id -> {
+                return arrayList.add( id);
+            });
+            List<DeviceLedger> supplierManageList = deviceLedgerMapper.selectBatchIds(arrayList);
+            ArrayList<DeviceLedgerExeclDto> deviceLedgerExeclDtos = new ArrayList<>();
+            supplierManageList.stream().forEach(deviceLedger -> {
+                DeviceLedgerExeclDto deviceLedgerExeclDto = new DeviceLedgerExeclDto();
+                BeanUtils.copyProperties(deviceLedger,deviceLedgerExeclDto);
+                deviceLedgerExeclDto.setCreateUser(sysUserMapper.selectUserById(Long.valueOf(deviceLedger.getCreateUser().toString())).getUserName());
+                deviceLedgerExeclDtos.add(deviceLedgerExeclDto);
+            });
+            ExcelUtil<DeviceLedgerExeclDto> util = new ExcelUtil<DeviceLedgerExeclDto>(DeviceLedgerExeclDto.class);
+            util.exportExcel(response, deviceLedgerExeclDtos, "璁惧鍙拌处瀵煎嚭");
+        }
+
     }
 }

--
Gitblit v1.9.3