From 75c32d6d697a550400d0b4eec95b8570d83b726f Mon Sep 17 00:00:00 2001
From: kimi <kimi42345@gmail.com>
Date: 星期五, 27 十一月 2020 19:20:32 +0800
Subject: [PATCH] 提交

---
 src/main/java/com/highdatas/srs/service/impl/SchemeServiceImpl.java |   54 +++++++++++++++++++++++++++++++++++++++++++++++++++---
 1 files changed, 51 insertions(+), 3 deletions(-)

diff --git a/src/main/java/com/highdatas/srs/service/impl/SchemeServiceImpl.java b/src/main/java/com/highdatas/srs/service/impl/SchemeServiceImpl.java
index b556868..06a8211 100644
--- a/src/main/java/com/highdatas/srs/service/impl/SchemeServiceImpl.java
+++ b/src/main/java/com/highdatas/srs/service/impl/SchemeServiceImpl.java
@@ -1,10 +1,17 @@
 package com.highdatas.srs.service.impl;
 
-import com.highdatas.srs.entity.Scheme;
-import com.highdatas.srs.mapper.SchemeMapper;
-import com.highdatas.srs.service.ISchemeService;
+import com.baomidou.mybatisplus.mapper.EntityWrapper;
 import com.baomidou.mybatisplus.service.impl.ServiceImpl;
+import com.highdatas.srs.entity.Scheme;
+import com.highdatas.srs.entity.SchemeBill;
+import com.highdatas.srs.mapper.SchemeMapper;
+import com.highdatas.srs.service.ISchemeBillService;
+import com.highdatas.srs.service.ISchemeService;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+
+import java.math.BigDecimal;
+import java.util.List;
 
 /**
  * <p>
@@ -16,5 +23,46 @@
  */
 @Service
 public class SchemeServiceImpl extends ServiceImpl<SchemeMapper, Scheme> implements ISchemeService {
+    @Autowired
+    ISchemeBillService billService;
+    @Override
+    public Scheme refreshBudget(String schemeId) {
+        Scheme scheme = selectById(schemeId);
+        if (scheme == null) {
+            return null;
+        }
+        BigDecimal decimal = new BigDecimal(0);
+        List<SchemeBill> parent_id = billService.selectList(new EntityWrapper<SchemeBill>().eq("parent_id", schemeId));
+        if (parent_id != null) {
+            for (SchemeBill schemeBill : parent_id) {
+                BigDecimal money = schemeBill.getMoney();
+                if (money != null) {
+                    decimal = decimal.add(money);
+                }
+            }
+            scheme.setBudget(decimal).updateById();
+        }
+        return scheme;
+    }
 
+    @Override
+    public Scheme refreshCheckMoney(String schemeId) {
+        Scheme scheme = selectById(schemeId);
+        if (scheme == null) {
+            return null;
+        }
+        BigDecimal decimal = new BigDecimal(0);
+        List<SchemeBill> parent_id = billService.selectList(new EntityWrapper<SchemeBill>().eq("parent_id", schemeId));
+        if (parent_id != null) {
+            for (SchemeBill schemeBill : parent_id) {
+                BigDecimal money = schemeBill.getMoney();
+                Boolean checked = schemeBill.getChecked();
+                if (money != null && checked != null && checked) {
+                    decimal = decimal.add(money);
+                }
+            }
+            scheme.setPayMoney(decimal).updateById();
+        }
+        return scheme;
+    }
 }

--
Gitblit v1.8.0