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