From 7cf8ae446e880537a38bd7ac403bd53741a5294f Mon Sep 17 00:00:00 2001
From: 李林 <z1292839451@163.com>
Date: 星期一, 08 一月 2024 09:21:19 +0800
Subject: [PATCH] 修订版2.0

---
 data-server/src/main/java/com/yuanchu/mom/service/impl/DataReportingServiceImpl.java |   60 +++++++++++++++++++++++++++++++++++++++++++++++++++++-------
 1 files changed, 53 insertions(+), 7 deletions(-)

diff --git a/data-server/src/main/java/com/yuanchu/mom/service/impl/DataReportingServiceImpl.java b/data-server/src/main/java/com/yuanchu/mom/service/impl/DataReportingServiceImpl.java
index c978ba5..efe7b8d 100644
--- a/data-server/src/main/java/com/yuanchu/mom/service/impl/DataReportingServiceImpl.java
+++ b/data-server/src/main/java/com/yuanchu/mom/service/impl/DataReportingServiceImpl.java
@@ -19,15 +19,19 @@
 import com.yuanchu.mom.dto.DataReportingPageDto;
 import com.yuanchu.mom.dto.ProductCountDto;
 import com.yuanchu.mom.dto.RegistrantCountDto;
+import com.yuanchu.mom.mapper.FansSubmitMapper;
 import com.yuanchu.mom.pojo.DataReporting;
+import com.yuanchu.mom.pojo.FansSubmit;
 import com.yuanchu.mom.service.DataReportingService;
 import com.yuanchu.mom.mapper.DataReportingMapper;
+import com.yuanchu.mom.service.FansSubmitService;
 import com.yuanchu.mom.utils.Jwt;
 import com.yuanchu.mom.utils.QueryWrappers;
 import com.yuanchu.mom.utils.ServletUtils;
 import lombok.AllArgsConstructor;
 import org.apache.tomcat.util.http.fileupload.IOUtils;
 import org.springframework.beans.factory.annotation.Value;
+import org.springframework.format.annotation.DateTimeFormat;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
@@ -35,12 +39,12 @@
 import java.io.*;
 import java.math.BigDecimal;
 import java.math.RoundingMode;
+import java.text.SimpleDateFormat;
+import java.time.LocalDate;
 import java.time.LocalDateTime;
 import java.time.format.DateTimeFormatter;
-import java.util.Arrays;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
+import java.util.*;
+import java.util.concurrent.ExecutionException;
 import java.util.concurrent.atomic.AtomicInteger;
 
 /**
@@ -60,6 +64,12 @@
 
     @Value("${outPath}")
     private String outPath;
+
+    @Resource
+    private FansSubmitMapper fansSubmitMapper;
+
+    @Resource
+    private FansSubmitService fansSubmitService;
 
     @Override
     public Map<String, Object> selectDataReportingList(IPage<DataReportingPageDto> page, DataReportingPageDto dataReportingPageDto) {
@@ -123,12 +133,15 @@
 
     @Override
     @Transactional(rollbackFor = Exception.class)
-    public void addDataReporting(List<DataReporting> list) {
+    public void addDataReporting(List<DataReporting> list, String date) {
         int userId = Integer.parseInt(JSONUtil.parseObj(new Jwt().readJWT(ServletUtils.getRequest().getHeader("token")).get("data")).get("id") + "");
         Map<String, String> map = dataReportingMapper.selectUser(userId);
         for (DataReporting dataReporting : list) {
-            if (dataReporting.getName() == null) dataReporting.setName(map.get("account"));
-            DataReporting one = dataReportingMapper.selectOne(Wrappers.<DataReporting>lambdaQuery().eq(DataReporting::getProduct, dataReporting.getProduct()).eq(DataReporting::getName, dataReporting.getName()));
+            if (dataReporting.getName() == null || dataReporting.getName().equals("")) dataReporting.setName(map.get("account"));
+            if (date == null) {
+                date = LocalDate.now().minusDays(1).format(DateTimeFormatter.ofPattern("yyyy-MM-dd"));
+            };
+            DataReporting one = dataReportingMapper.selectOne(Wrappers.<DataReporting>lambdaQuery().eq(DataReporting::getProduct, dataReporting.getProduct()).eq(DataReporting::getName, dataReporting.getName()).like(DataReporting::getCreateTime, date));
             try {
                 dataReporting.setRebateConsumption(dataReporting.getAccountConsumption().divide(dataReporting.getCustomerRebate(), 2, RoundingMode.HALF_UP));
             } catch (Exception e) {
@@ -156,6 +169,7 @@
             if (ObjectUtil.isEmpty(one)) {
                 dataReporting.setRegistrant(map.get("name"));
                 dataReporting.setDepartment(map.get("department"));
+                dataReporting.setCreateTime(LocalDateTime.parse(date + " " + new SimpleDateFormat("HH:mm:ss").format(new Date()), DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss")));
                 dataReportingMapper.insert(dataReporting);
             } else {
                 dataReporting.setId(one.getId());
@@ -165,8 +179,11 @@
         }
     }
 
+    @Transactional(rollbackFor = Exception.class)
     @Override
     public Integer delDataReporting(Integer id) {
+        List<FansSubmit> fansSubmits = fansSubmitMapper.selectList(Wrappers.<FansSubmit>lambdaQuery().eq(FansSubmit::getDataId, id).select(FansSubmit::getId));
+        fansSubmitMapper.deleteBatchIds(fansSubmits);
         return dataReportingMapper.deleteById(id);
     }
 
@@ -256,6 +273,35 @@
         writer.close();
         return name;
     }
+
+    @Override
+    public String downFansSubmitFile(IPage<FansSubmit> page, FansSubmit fansSubmit) {
+        Map<String, Object> map = fansSubmitService.selectFansSubmitList(page, fansSubmit);
+        String time = LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyy_MM_dd_HH_mm_ss"));
+        String name = "绮変笣涓婃姤" + time + ".csv";
+        String path = outPath + name;
+        CsvWriter writer = CsvUtil.getWriter(path, CharsetUtil.CHARSET_GBK);
+        writer.write(new String[]{"瀵煎嚭鏃堕棿锛�"+time});
+        writer.write(new String[]{});
+        writer.write(new String[]{});
+        writer.write(new String[]{});
+        writer.write(new String[]{});
+        List<Map<String, String>> head = JSON.parseObject(JSON.toJSONString(map.get("head")), List.class);
+        String[] str = new String[head.size()];
+        for (int i = 0; i < head.size(); i++) {
+            str[i] = head.get(i).get("value");
+        }
+        writer.write(str);
+        JSON.parseObject(JSON.toJSONString(map.get("body")), IPage.class).getRecords().forEach(a->{
+            Map<String, Object> map1 = JSON.parseObject(JSON.toJSONString(a), Map.class);
+            for (int i = 0; i < head.size(); i++) {
+                str[i] = map1.get(head.get(i).get("label"))==null?null:map1.get(head.get(i).get("label"))+"";
+            }
+            writer.write(str);
+        });
+        writer.close();
+        return name;
+    }
 }
 
 

--
Gitblit v1.9.3