From ce97969fce786a9a1dc0bc18acad6261bee73a21 Mon Sep 17 00:00:00 2001
From: 李林 <z1292839451@163.com>
Date: 星期二, 02 一月 2024 00:39:12 +0800
Subject: [PATCH] 导出功能

---
 data-server/src/main/java/com/yuanchu/mom/service/impl/DataReportingServiceImpl.java |  124 +++++++++++++++++++++++++++++++++++++----
 1 files changed, 111 insertions(+), 13 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 087b0df..24071b3 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
@@ -1,6 +1,14 @@
 package com.yuanchu.mom.service.impl;
 
+import cn.hutool.core.text.csv.CsvUtil;
+import cn.hutool.core.text.csv.CsvWriter;
+import cn.hutool.core.util.CharsetUtil;
+import cn.hutool.core.util.ObjectUtil;
+import cn.hutool.json.JSONUtil;
+import com.alibaba.fastjson.JSON;
+import com.alibaba.fastjson.JSONObject;
 import com.baomidou.mybatisplus.core.metadata.IPage;
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.yuanchu.mom.common.GetLook;
 import com.yuanchu.mom.common.PrintChina;
@@ -10,24 +18,32 @@
 import com.yuanchu.mom.pojo.DataReporting;
 import com.yuanchu.mom.service.DataReportingService;
 import com.yuanchu.mom.mapper.DataReportingMapper;
+import com.yuanchu.mom.utils.Jwt;
 import com.yuanchu.mom.utils.QueryWrappers;
+import com.yuanchu.mom.utils.ServletUtils;
 import lombok.AllArgsConstructor;
 import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
 
 import java.math.BigDecimal;
 import java.math.RoundingMode;
+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.concurrent.atomic.AtomicInteger;
 
 /**
-* @author Administrator
-* @description 閽堝琛ㄣ�恉ata_reporting(鏁版嵁涓婃姤)銆戠殑鏁版嵁搴撴搷浣淪ervice瀹炵幇
-* @createDate 2023-12-23 10:39:43
-*/
+ * @author Administrator
+ * @description 閽堝琛ㄣ�恉ata_reporting(鏁版嵁涓婃姤)銆戠殑鏁版嵁搴撴搷浣淪ervice瀹炵幇
+ * @createDate 2023-12-23 10:39:43
+ */
 @Service
 @AllArgsConstructor
 public class DataReportingServiceImpl extends ServiceImpl<DataReportingMapper, DataReporting>
-    implements DataReportingService{
+        implements DataReportingService {
 
     private DataReportingMapper dataReportingMapper;
 
@@ -37,7 +53,7 @@
     public Map<String, Object> selectDataReportingList(IPage<DataReportingPageDto> page, DataReportingPageDto dataReportingPageDto) {
         Map<String, Object> map = new HashMap<>();
         Map<String, Integer> map1 = getLook.selectPowerByMethodAndUserId("selectDataReportingList");
-        if(map1.get("look")==1) dataReportingPageDto.setCreateUser(map1.get("userId"));
+        if (map1.get("look") == 1) dataReportingPageDto.setCreateUser(map1.get("userId"));
         map.put("head", PrintChina.printChina(DataReportingPageDto.class));
         map.put("body", dataReportingMapper.selectDataReportingDtoPageList(page, QueryWrappers.queryWrappers(dataReportingPageDto)));
         return map;
@@ -51,9 +67,14 @@
         dataReporting2.setCustomerCosts(dataReporting.getCustomerCosts());
         dataReporting2.setCustomerRebate(dataReporting.getCustomerRebate());
         dataReporting2.setRemark(dataReporting.getRemark());
-        dataReporting2.setRebateConsumption(dataReporting.getAccountConsumption().divide(dataReporting.getCustomerRebate(), 2, RoundingMode.HALF_UP));
-        dataReporting2.setConsumption(dataReporting.getCustomerCosts().multiply(BigDecimal.valueOf(dataReporting.getFansAdd())));
-        System.out.println(dataReporting2);
+        try {
+            dataReporting2.setRebateConsumption(dataReporting.getAccountConsumption().divide(dataReporting.getCustomerRebate(), 2, RoundingMode.HALF_UP));
+        } catch (Exception e) {
+        }
+        try {
+            dataReporting2.setConsumption(dataReporting.getCustomerCosts().multiply(BigDecimal.valueOf(dataReporting.getFansAdd())));
+        } catch (Exception e) {
+        }
         return dataReportingMapper.updateById(dataReporting2);
     }
 
@@ -61,8 +82,8 @@
     public Map<String, Object> selectProductCountDtoPageList(IPage<ProductCountDto> page, ProductCountDto productCountDto) {
         Map<String, Object> map = new HashMap<>();
         map.put("head", PrintChina.printChina(ProductCountDto.class));
-        Map<String, Integer> map1 = getLook.selectPowerByMethodAndUserId("selectDataReportingList");
-        if(map1.get("look")==1) productCountDto.setCreateUser(map1.get("userId"));
+        Map<String, Integer> map1 = getLook.selectPowerByMethodAndUserId("selectProductCountDtoPageList");
+        if (map1.get("look") == 1) productCountDto.setCreateUser(map1.get("userId"));
         map.put("body", dataReportingMapper.selectProductCountDtoPageList(page, QueryWrappers.queryWrappers(productCountDto)));
         return map;
     }
@@ -71,8 +92,8 @@
     public Map<String, Object> selectRegistrantCountDtoPageList(IPage<RegistrantCountDto> page, RegistrantCountDto registrantCountDto) {
         Map<String, Object> map = new HashMap<>();
         map.put("head", PrintChina.printChina(RegistrantCountDto.class));
-        Map<String, Integer> map1 = getLook.selectPowerByMethodAndUserId("selectDataReportingList");
-        if(map1.get("look")==1) registrantCountDto.setCreateUser(map1.get("userId"));
+        Map<String, Integer> map1 = getLook.selectPowerByMethodAndUserId("selectRegistrantCountDtoPageList");
+        if (map1.get("look") == 1) registrantCountDto.setCreateUser(map1.get("userId"));
         map.put("body", dataReportingMapper.selectRegistrantCountDtoPageList(page, QueryWrappers.queryWrappers(registrantCountDto)));
         return map;
     }
@@ -87,6 +108,83 @@
         map.put("data5", dataReportingMapper.getDataFor5());
         return map;
     }
+
+    @Override
+    @Transactional(rollbackFor = Exception.class)
+    public void addDataReporting(List<DataReporting> list) {
+        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()));
+            try {
+                dataReporting.setRebateConsumption(dataReporting.getAccountConsumption().divide(dataReporting.getCustomerRebate(), 2, RoundingMode.HALF_UP));
+            } catch (Exception e) {
+            }
+            try {
+                dataReporting.setConsumption(dataReporting.getCustomerCosts().multiply(new BigDecimal(dataReporting.getFansAdd())));
+            } catch (Exception e) {
+            }
+            try {
+                dataReporting.setDiscountedConsumption(dataReporting.getConsumption().divide(dataReporting.getAgentRebate(), 2, RoundingMode.HALF_UP));
+            } catch (Exception e) {
+            }
+            try {
+                dataReporting.setActualCost(dataReporting.getAccountConsumption().divide(new BigDecimal(dataReporting.getFansAdd()), 2, RoundingMode.HALF_UP));
+            } catch (Exception e) {
+            }
+            try {
+                dataReporting.setProfit(dataReporting.getRebateConsumption().subtract(dataReporting.getConsumption()));
+            } catch (Exception e) {
+            }
+            try {
+                dataReporting.setRemark(dataReporting.getConsumption().subtract(dataReporting.getRebateConsumption()));
+            } catch (Exception e) {
+            }
+            if (ObjectUtil.isEmpty(one)) {
+                dataReporting.setRegistrant(map.get("name"));
+                dataReporting.setDepartment(map.get("department"));
+                dataReportingMapper.insert(dataReporting);
+            } else {
+                dataReporting.setId(one.getId());
+                dataReporting.setName(null);
+                dataReportingMapper.updateById(dataReporting);
+            }
+        }
+    }
+
+    @Override
+    public Integer delDataReporting(Integer id) {
+        return dataReportingMapper.deleteById(id);
+    }
+
+    @Override
+    public String downDataReportingFile(IPage<DataReportingPageDto> page, DataReportingPageDto dataReportingPageDto) {
+        Map<String, Object> map = selectDataReportingList(page, dataReportingPageDto);
+        String time = LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyy_MM_dd HH_mm_ss"));
+        String name = "D:\\Download\\鏁版嵁涓婃姤 " + time + ".csv";
+        CsvWriter writer = CsvUtil.getWriter(name, 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 null;
+    }
 }
 
 

--
Gitblit v1.9.3