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