From 3b0516a2959e25576e4f3fda697a3b025d06c8c9 Mon Sep 17 00:00:00 2001
From: zxl <763096477@qq.com>
Date: 星期二, 24 六月 2025 14:09:07 +0800
Subject: [PATCH] 每日录像可用指标添加日志,修改大屏为查看当前月平均

---
 ycl-server/src/main/java/com/ycl/platform/service/impl/TContractServiceImpl.java |  248 +++++++++++++++++++++++++++++--------------------
 1 files changed, 145 insertions(+), 103 deletions(-)

diff --git a/ycl-server/src/main/java/com/ycl/platform/service/impl/TContractServiceImpl.java b/ycl-server/src/main/java/com/ycl/platform/service/impl/TContractServiceImpl.java
index eeb6d56..a3d1960 100644
--- a/ycl-server/src/main/java/com/ycl/platform/service/impl/TContractServiceImpl.java
+++ b/ycl-server/src/main/java/com/ycl/platform/service/impl/TContractServiceImpl.java
@@ -1,38 +1,46 @@
 package com.ycl.platform.service.impl;
 
 import com.alibaba.excel.EasyExcel;
-import com.alibaba.excel.write.merge.LoopMergeStrategy;
+import com.alibaba.excel.read.listener.PageReadListener;
 import com.alibaba.excel.write.style.column.LongestMatchColumnWidthStyleStrategy;
 import com.alibaba.fastjson2.JSON;
-import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.ycl.handler.CommentWriteHandler;
+import com.ycl.handler.CustomCellWriteHandler;
 import com.ycl.handler.CustomSheetWriteHandler;
 import com.ycl.platform.domain.entity.CalculateMoneyRule;
 import com.ycl.platform.domain.entity.CalculateRule;
 import com.ycl.platform.domain.entity.TContract;
+import com.ycl.platform.domain.query.ContractQuery;
+import com.ycl.platform.domain.vo.ContractVO;
 import com.ycl.platform.mapper.TContractMapper;
 import com.ycl.platform.service.ICalculateRuleService;
 import com.ycl.platform.service.ITContractService;
 import com.ycl.system.AjaxResult;
+import com.ycl.system.Result;
+import com.ycl.system.page.PageUtil;
 import com.ycl.utils.DateUtils;
+import com.ycl.utils.SecurityUtils;
 import com.ycl.utils.StringUtils;
+import enumeration.ContractStatus;
 import enumeration.general.RuleDeductCategoryEnum;
 import jakarta.servlet.http.HttpServletResponse;
 import lombok.AllArgsConstructor;
+import lombok.SneakyThrows;
 import org.apache.commons.lang3.ObjectUtils;
 import org.springframework.stereotype.Service;
-import org.springframework.transaction.annotation.Propagation;
 import org.springframework.transaction.annotation.Transactional;
 import org.springframework.transaction.interceptor.TransactionAspectSupport;
+import org.springframework.util.CollectionUtils;
 import org.springframework.web.multipart.MultipartFile;
 
+import java.io.IOException;
 import java.net.URLEncoder;
 import java.nio.charset.StandardCharsets;
-import java.util.ArrayList;
-import java.util.Date;
-import java.util.List;
+import java.util.*;
 import java.util.stream.Collectors;
 
 /**
@@ -47,33 +55,24 @@
 
     private final ICalculateRuleService calculateRuleService;
     private final CalculateMoneyRuleServiceImpl calculateMoneyRuleService;
-    private final YwUnitServiceImpl ywUnitService;
 
     @Override
     public void importTemplate(HttpServletResponse response) {
         try {
-            List<CalculateRule> list = getExcelData();
-            ArrayList<String> arrayList = new ArrayList<>();
-            arrayList.add("鎵f寚瀹氬垎鏁�");
-            arrayList.add("鍒嗘暟涔樹互鏁伴噺");
-            arrayList.add("闄や互鏁伴噺鍚庝箻浠ュ垎鏁�");
             response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
             response.setCharacterEncoding("utf-8");
             String fileName = URLEncoder.encode("瀵煎叆鍚堝悓妯℃澘", StandardCharsets.UTF_8).replace("\\+", "%20");
             response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileName + ".xlsx");
-            // 绗竴鍒椼�佺浜屽垪涓婁笅鍚堝苟3琛�
-            LoopMergeStrategy loopMergeStrategy1 = new LoopMergeStrategy(3, 0);
-            LoopMergeStrategy loopMergeStrategy2 = new LoopMergeStrategy(3, 1);
             EasyExcel.write(response.getOutputStream(), CalculateRule.class)
-                    .registerWriteHandler(loopMergeStrategy1)
-                    .registerWriteHandler(loopMergeStrategy2)
                     // 鑷�傚簲鍒楀
                     .registerWriteHandler(new LongestMatchColumnWidthStyleStrategy())
                     // 涓嬫媺妗�
-                    .registerWriteHandler(new CustomSheetWriteHandler(arrayList))
+                    .registerWriteHandler(new CustomSheetWriteHandler(Arrays.asList("鎵f寚瀹氬垎鏁�", "鍒嗘暟涔樹互鏁伴噺", "闄や互鏁伴噺鍚庝箻浠ュ垎鏁�")))
+                    .registerWriteHandler(new CustomCellWriteHandler())
                     // 鏍囨敞
                     .registerWriteHandler(new CommentWriteHandler())
-                    .sheet("鍚堝悓瀵煎叆妯℃澘").doWrite(list);
+                    .sheet("鍚堝悓瀵煎叆妯℃澘")
+                    .doWrite(getExcelData());
         } catch (Exception e) {
             throw new RuntimeException(e);
         }
@@ -81,68 +80,53 @@
 
     private List<CalculateRule> getExcelData() {
         ArrayList<CalculateRule> list = new ArrayList<>();
-        CalculateRule calculateRule1 = new CalculateRule();
-        calculateRule1.setId(1L);
-        calculateRule1.setRuleName("瑙嗛骞冲潎鍦ㄧ嚎鐜�");
-        calculateRule1.setRuleDesc("锛�1锛夎棰戝湪绾跨巼=瑙嗛鍦ㄧ嚎鏁伴噺梅鍓嶇鏈嶅姟閰嶇疆鏁帮紙搴旀墸闄や笉鍙姉鍔涙垨涓庝腑鏍囦汉鏈嶅姟鏃犲叧閫犳垚鐨勬帀绾块櫎澶栥�傦級");
-        calculateRule1.setRuleCondition("鈮�98%");
-        calculateRule1.setDeductCategory(RuleDeductCategoryEnum.DEDUCT_POINTS);
-        calculateRule1.setCalcFraction(0.00);
-        list.add(calculateRule1);
-        CalculateRule calculateRule2 = new CalculateRule();
-        calculateRule2.setRuleDesc("锛�2锛夎棰戝钩鍧囧湪绾跨巼锛氬綋鏈堟瘡鏃ヨ棰戝湪绾跨巼鐨勫钩鍧囧�笺��");
-        calculateRule2.setRuleCondition("95%鈮よ棰戝钩鍧囧湪绾跨巼锛�98%");
-        calculateRule2.setDeductCategory(RuleDeductCategoryEnum.DEDUCT_POINTS);
-        calculateRule2.setCalcFraction(5.00);
-        list.add(calculateRule2);
-        CalculateRule calculateRule3 = new CalculateRule();
-        calculateRule3.setRuleDesc("锛�3锛夎棰戝湪绾跨巼淇濈暀鏁存暟骞朵笖鍚戜笅鍙栨暣銆�");
-        calculateRule3.setRuleCondition("90%鈮よ棰戝钩鍧囧湪绾跨巼锛�95%");
-        calculateRule3.setDeductCategory(RuleDeductCategoryEnum.DEDUCT_POINTS);
-        calculateRule3.setCalcFraction(10.00);
-        list.add(calculateRule3);
-        CalculateRule calculateRule4 = new CalculateRule();
-        calculateRule4.setId(2L);
-        calculateRule4.setRuleName("鍓嶇鎰熺煡婧愭不鐞嗗伐浣�");
-        calculateRule4.setRuleDesc("鏃堕挓鍚屾锛堣秴杩嚶�3绉掍负涓嶅悎鏍硷級");
-        calculateRule4.setRuleCondition("瑙嗛鐩戞帶鎽勫儚鏈烘椂閽熶笉鍚屾锛�24灏忔椂鍐呮湭淇鐨�");
-        calculateRule4.setDeductCategory(RuleDeductCategoryEnum.MULTIPLY_POINTS_BY_QUANTITY);
-        calculateRule4.setCalcFraction(0.1);
-        list.add(calculateRule4);
-        CalculateRule calculateRule7 = new CalculateRule();
-        calculateRule7.setRuleDesc("鈥︹�︹�︹��");
-        calculateRule7.setRuleCondition("鈥︹�︹�︹��");
-        list.add(calculateRule7);
-        CalculateRule calculateRule6 = new CalculateRule();
-        calculateRule6.setRuleDesc("鈥︹�︹�︹��");
-        calculateRule6.setRuleCondition("鈥︹�︹�︹��");
-        list.add(calculateRule6);
-        CalculateRule calculateRule5 = new CalculateRule();
-        calculateRule5.setId(3L);
-        calculateRule5.setRuleName("鍚庡彴绯荤粺鐨勪繚闅�");
-        calculateRule5.setRuleDesc("鍦ㄤ釜鍒澶囨晠闅滀絾涓嶅奖鍝嶈鍔熻兘妯″潡鏁翠綋搴旂敤鎯呭喌涓嬶紝椤诲湪72灏忔椂浠ュ唴鎺掗櫎鏁呴殰銆�");
-        calculateRule5.setRuleCondition("鍗曟鏁呴殰鏃堕暱鑻ヨ秴鍑�144灏忔椂浠ュ悗锛屾瘡瓒呭嚭12灏忔椂锛堜笉瓒�12灏忔椂鎸�12灏忔椂璁★級");
-        calculateRule5.setDeductCategory(RuleDeductCategoryEnum.MULTIPLY_POINTS_AFTER_DIVIDING_QUANTITY);
-        calculateRule5.setCalcFraction(2.00);
-        calculateRule5.setCalcUnit(12);
-        list.add(calculateRule5);
+        setTemplateRule(list,1,"璁惧骞冲潎鍦ㄧ嚎鐜�","鈮�98%",null,98D,RuleDeductCategoryEnum.DEDUCT_POINTS,0.00);
+        setTemplateRule(list,null,"璁惧骞冲潎鍦ㄧ嚎鐜�","95%鈮よ澶囧钩鍧囧湪绾跨巼锛�98%",97D,95D,RuleDeductCategoryEnum.DEDUCT_POINTS,5.00);
+        setTemplateRule(list,null,"璁惧骞冲潎鍦ㄧ嚎鐜�","90%鈮よ澶囧钩鍧囧湪绾跨巼锛�95%",94D,90D,RuleDeductCategoryEnum.DEDUCT_POINTS,10.00);
+        setTemplateRule(list,null,"璁惧骞冲潎鍦ㄧ嚎鐜�","锛�90%",89D,null,RuleDeductCategoryEnum.DEDUCT_POINTS,20.00);
+
+        setTemplateRule(list,2,"鍓嶇鎰熺煡婧愭不鐞嗗伐浣�","鏃堕挓鍚屾",null,25D,RuleDeductCategoryEnum.MULTIPLY_POINTS_BY_QUANTITY,0.1);
+        setTemplateRule(list,null,"鍓嶇鎰熺煡婧愭不鐞嗗伐浣�","OSD鏍囪瘑",null,25D,RuleDeductCategoryEnum.MULTIPLY_POINTS_BY_QUANTITY,0.1);
+        setTemplateRule(list,null,"鍓嶇鎰熺煡婧愭不鐞嗗伐浣�","涓�鏈轰竴妗�",null,25D,RuleDeductCategoryEnum.MULTIPLY_POINTS_BY_QUANTITY,0.1);
+
+        setTemplateRule(list,3,"瀵逛簬鍓嶇鐐逛綅寮傚父鎯呭喌鐨勫鐞�","闀滃ご鏁呴殰鎴栨薄鏌撴垨鏍戞灊閬尅鎴栨灙鏈鸿瑙掑亸绉绘甯歌搴︽垨琛ュ厜鐏簲浜湭浜�,24灏忔椂鍚庢湭淇鐨�",48D,25D,RuleDeductCategoryEnum.MULTIPLY_POINTS_BY_QUANTITY,0.5);
+        setTemplateRule(list,null,"瀵逛簬鍓嶇鐐逛綅寮傚父鎯呭喌鐨勫鐞�","闀滃ご鏁呴殰鎴栨薄鏌撴垨鏍戞灊閬尅鎴栨灙鏈鸿瑙掑亸绉绘甯歌搴︽垨琛ュ厜鐏簲浜湭浜�,48灏忔椂鍚庢湭淇鐨�",null,49D,RuleDeductCategoryEnum.MULTIPLY_POINTS_BY_QUANTITY,1.0);
+
+        setTemplateRule(list,4,"纭繚褰曞儚瀹屾暣涓嶅畾鏈熷鎵�鏈夌偣浣嶅綍鍍忓畬鏁存�ф娊鏌�","姣忚矾瑙嗛绱涓㈠け10鍒嗛挓锛堝惈锛変互鍐�",10D,null,RuleDeductCategoryEnum.MULTIPLY_POINTS_BY_QUANTITY,0.2);
+        setTemplateRule(list,null,"纭繚褰曞儚瀹屾暣涓嶅畾鏈熷鎵�鏈夌偣浣嶅綍鍍忓畬鏁存�ф娊鏌�","涓㈠け10-60 鍒嗛挓锛堝惈锛�",60D,9D,RuleDeductCategoryEnum.MULTIPLY_POINTS_BY_QUANTITY,0.5);
+        setTemplateRule(list,null,"纭繚褰曞儚瀹屾暣涓嶅畾鏈熷鎵�鏈夌偣浣嶅綍鍍忓畬鏁存�ф娊鏌�","涓㈠け1 灏忔椂-4 灏忔椂锛堝惈锛�",240D,59D,RuleDeductCategoryEnum.MULTIPLY_POINTS_BY_QUANTITY,1.0);
+        setTemplateRule(list,null,"纭繚褰曞儚瀹屾暣涓嶅畾鏈熷鎵�鏈夌偣浣嶅綍鍍忓畬鏁存�ф娊鏌�","涓㈠け4 灏忔椂-12 灏忔椂锛堝惈锛�",720D,539D,RuleDeductCategoryEnum.MULTIPLY_POINTS_BY_QUANTITY,1.5);
+        setTemplateRule(list,null,"纭繚褰曞儚瀹屾暣涓嶅畾鏈熷鎵�鏈夌偣浣嶅綍鍍忓畬鏁存�ф娊鏌�","涓㈠け12 灏忔椂浠ヤ笂",null,719D,RuleDeductCategoryEnum.MULTIPLY_POINTS_BY_QUANTITY,2.0);
+
+        setTemplateRule(list,5,"纭繚鍥剧墖瀹屾暣涓嶅畾鏈熷鎵�鏈変汉鑴歌溅杈嗕互鍙婃櫤鑳藉墠绔姄鎷嶇殑鍥剧墖瀹屾暣鎬ф娊鏌�","鍙戠幇鍚庡彴瀛樺偍涓嶈兘璋冨彇鍓嶇璁惧鍥剧墖",null,null,RuleDeductCategoryEnum.MULTIPLY_POINTS_BY_QUANTITY,2.0);
+
         return list;
     }
-
+    private static void setTemplateRule(ArrayList<CalculateRule> list,Integer id,String ruleName,String condition,Double max,Double min,RuleDeductCategoryEnum deductCategoryEnum,Double calcFraction) {
+        CalculateRule calculateRule = new CalculateRule();
+        calculateRule.setId(id);
+        calculateRule.setRuleName(ruleName);
+        calculateRule.setRuleCondition(condition);
+        calculateRule.setMax(max);
+        calculateRule.setMin(min);
+        calculateRule.setDeductCategory(deductCategoryEnum);
+        calculateRule.setCalcFraction(calcFraction);
+        list.add(calculateRule);
+    }
     @Override
     @Transactional
+    @SneakyThrows
     public AjaxResult importData(MultipartFile file, TContract tContract) {
         // 淇濆瓨鍚堝悓
         tContract.setCreateTime(DateUtils.getNowDate());
-        tContract.setDeleted("0");
         save(tContract);
-        List<CalculateRule> list = calculateRuleService.readExcel(file);
+
+        // 鑾峰彇鍚堝悓Excel瑙勫垯
+        List<CalculateRule> list = new ArrayList<>();
+        EasyExcel.read(file.getInputStream(), CalculateRule.class, new PageReadListener<CalculateRule>(list::addAll)).sheet().doRead();
 
         // 閬嶅巻鐖跺瓙鍏崇郴
         List<CalculateRule> calculateRulesToSave = new ArrayList<>();
-        CalculateRule fu1 = new CalculateRule();
-        CalculateRule fu2 = new CalculateRule();
-        boolean fuNew = true;
         for (CalculateRule calculateRule : list) {
             // 鍒ゆ柇鏁版嵁瀹屾暣鎬�
             if (ObjectUtils.isEmpty(calculateRule.getDeductCategory())) {
@@ -157,35 +141,13 @@
                 TransactionAspectSupport.currentTransactionStatus().setRollbackOnly();
                 return AjaxResult.warn("璇峰~鍐�" + calculateRule.getRuleName() + calculateRule.getRuleCondition() + "鎵e垎鏁板��");
             }
-            // 淇濆瓨鐖惰鍒欒幏鍙栫埗id
+            // 淇濆瓨瑙勫垯
             if (StringUtils.isNotBlank(calculateRule.getRuleName())) {
-                CalculateRule one = new CalculateRule();
-                one.setParentId(0L);
-                one.setContractId(tContract.getId().intValue());
-                one.setRuleName(calculateRule.getRuleName());
-                one.setCreateTime(DateUtils.getNowDate());
-                one.setDeleted(0);
-                calculateRuleService.save(one);
-                fu1 = one;
-                fuNew = true;
+                calculateRule.setContractId(tContract.getId().intValue());
+                calculateRule.setCreateTime(DateUtils.getNowDate());
+                calculateRule.setUpdateTime(DateUtils.getNowDate());
+                calculateRulesToSave.add(calculateRule);
             }
-            // 淇濆瓨绗簩灞傜埗瑙勫垯鑾峰彇绗簩灞傜埗id
-            if (StringUtils.isNotBlank(calculateRule.getRuleDesc())) {
-                CalculateRule two = new CalculateRule();
-                two.setContractId(tContract.getId().intValue());
-                two.setParentId(fu1.getId());
-                two.setRuleDesc(calculateRule.getRuleDesc());
-                two.setCreateTime(DateUtils.getNowDate());
-                two.setDeleted(0);
-                calculateRuleService.save(two);
-                fu2 = two;
-                fuNew = false;
-            }
-            calculateRule.setContractId(tContract.getId().intValue());
-            calculateRule.setCreateTime(DateUtils.getNowDate());
-            calculateRule.setDeleted(0);
-            calculateRule.setParentId(fuNew ? fu1.getId() : fu2.getId());
-            calculateRulesToSave.add(calculateRule);
         }
         // 鎵归噺淇濆瓨瑙勫垯
         calculateRuleService.saveBatch(calculateRulesToSave);
@@ -193,20 +155,100 @@
         calculateMoneyRuleService.saveBatch(JSON.parseArray(tContract.getRuleList(), CalculateMoneyRule.class).stream().peek(
                 calculateMoneyRule -> calculateMoneyRule.setContractId(tContract.getId().intValue())
         ).collect(Collectors.toList()));
+
+
         return AjaxResult.success("鎿嶄綔鎴愬姛");
     }
 
     @Override
-    public List<TContract> selectAll() {
-        return list(new LambdaQueryWrapper<TContract>()
-                .orderByDesc(TContract::getCreateTime))
-                .stream().peek(
-                        tContract -> tContract.setUnitName(ywUnitService.getById(tContract.getUnitId()).getUnitName())
-                ).toList();
+    public AjaxResult importRule(MultipartFile file, TContract tContract) throws IOException {
+
+        // 鑾峰彇鍚堝悓Excel瑙勫垯
+        List<CalculateRule> list = new ArrayList<>();
+        if(file!=null) {
+            EasyExcel.read(file.getInputStream(), CalculateRule.class, new PageReadListener<CalculateRule>(list::addAll)).sheet().doRead();
+
+            // 閬嶅巻鐖跺瓙鍏崇郴
+            List<CalculateRule> calculateRulesToSave = new ArrayList<>();
+            for (CalculateRule calculateRule : list) {
+                // 鍒ゆ柇鏁版嵁瀹屾暣鎬�
+                if (ObjectUtils.isEmpty(calculateRule.getDeductCategory())) {
+                    TransactionAspectSupport.currentTransactionStatus().setRollbackOnly();
+                    return AjaxResult.warn("璇烽�夋嫨" + calculateRule.getRuleName() + calculateRule.getRuleCondition() + "鎵e垎鏂瑰紡");
+                }
+                if (RuleDeductCategoryEnum.MULTIPLY_POINTS_AFTER_DIVIDING_QUANTITY.equals(calculateRule.getDeductCategory()) && ObjectUtils.isEmpty(calculateRule.getCalcUnit())) {
+                    TransactionAspectSupport.currentTransactionStatus().setRollbackOnly();
+                    return AjaxResult.warn("璇峰~鍐�" + calculateRule.getRuleName() + calculateRule.getRuleCondition() + "闄や互鏁伴噺");
+                }
+                if (ObjectUtils.isEmpty(calculateRule.getCalcFraction())) {
+                    TransactionAspectSupport.currentTransactionStatus().setRollbackOnly();
+                    return AjaxResult.warn("璇峰~鍐�" + calculateRule.getRuleName() + calculateRule.getRuleCondition() + "鎵e垎鏁板��");
+                }
+                // 淇濆瓨瑙勫垯
+                if (StringUtils.isNotBlank(calculateRule.getRuleName())) {
+                    calculateRule.setContractId(tContract.getId().intValue());
+                    calculateRule.setCreateTime(DateUtils.getNowDate());
+                    calculateRule.setUpdateTime(DateUtils.getNowDate());
+                    calculateRulesToSave.add(calculateRule);
+                }
+            }
+            if (!CollectionUtils.isEmpty(calculateRulesToSave)) {
+                //鍒犻櫎鍘熻鍒�
+                calculateRuleService.remove(new QueryWrapper<CalculateRule>().eq("contract_id", tContract.getId()));
+                // 鎵归噺淇濆瓨瑙勫垯
+                calculateRuleService.saveBatch(calculateRulesToSave);
+            }
+        }
+        List<CalculateMoneyRule> moneyRule = JSON.parseArray(tContract.getRuleList(), CalculateMoneyRule.class).stream().peek(
+                calculateMoneyRule -> calculateMoneyRule.setContractId(tContract.getId().intValue())
+        ).collect(Collectors.toList());
+        if (!CollectionUtils.isEmpty(moneyRule)) {
+            //鍒犻櫎鍘熻鍒�
+            calculateMoneyRuleService.remove(new QueryWrapper<CalculateMoneyRule>().eq("contract_id",tContract.getId()));
+            //鎵归噺淇濆瓨鑰冩牳缁撴灉搴旂敤瑙勫垯
+            calculateMoneyRuleService.saveBatch(moneyRule);
+        }
+        return AjaxResult.success("鎿嶄綔鎴愬姛");
+    }
+
+    @Override
+    public Result selectAll(ContractQuery query) {
+        IPage<ContractVO> page = PageUtil.getPage(query, ContractVO.class);
+        query.setUnitId(SecurityUtils.getUnitId());
+        baseMapper.getPage(page, query);
+        page.getRecords().stream().forEach(contract -> {
+            Date now = new Date();
+            if (now.before(contract.getStartTime())) {
+                contract.setStatus(ContractStatus.NOT_START);
+            } else if (now.after(contract.getEndTime())) {
+                contract.setStatus(ContractStatus.FINISHED);
+            } else {
+                contract.setStatus(ContractStatus.ACTIVE);
+            }
+        });
+        return Result.ok().data(page.getRecords()).total(page.getTotal());
     }
 
     @Override
     public List<CalculateMoneyRule> selectMoneyRules(Integer contractId) {
         return calculateMoneyRuleService.selectMoneyRules(contractId);
     }
+
+    @Override
+    public List<TContract> selectUsingContract() {
+        return new LambdaQueryChainWrapper<>(baseMapper)
+                .le(TContract::getStartTime, DateUtils.getDate())
+                .ge(TContract::getEndTime, DateUtils.getDate())
+                .list();
+    }
+
+    @Override
+    public Result timeRange(Integer unitId) {
+        List<TContract> unitContractList = new LambdaQueryChainWrapper<>(baseMapper)
+                .select(TContract::getId, TContract::getName, TContract::getStartTime, TContract::getEndTime)
+                .eq(TContract::getUnitId, unitId)
+                .ge(TContract::getEndTime, new Date())
+                .list();
+        return Result.ok().data(unitContractList);
+    }
 }

--
Gitblit v1.8.0