|
@@ -14,6 +14,8 @@ import vip.xiaonuo.money.param.MoneyIncomeParam;
|
|
import vip.xiaonuo.money.service.IMoneyIncomeService;
|
|
import vip.xiaonuo.money.service.IMoneyIncomeService;
|
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
|
import org.springframework.stereotype.Service;
|
|
import org.springframework.stereotype.Service;
|
|
|
|
+import vip.xiaonuo.sys.modular.org.entity.SysOrg;
|
|
|
|
+import vip.xiaonuo.sys.modular.org.mapper.SysOrgMapper;
|
|
|
|
|
|
import java.math.BigDecimal;
|
|
import java.math.BigDecimal;
|
|
import java.math.RoundingMode;
|
|
import java.math.RoundingMode;
|
|
@@ -38,6 +40,8 @@ public class MoneyIncomeServiceImpl extends ServiceImpl<MoneyIncomeMapper, Money
|
|
private MoneyIncomeMapper incomeMapper;
|
|
private MoneyIncomeMapper incomeMapper;
|
|
@Autowired
|
|
@Autowired
|
|
private MoneyBusinessMapper businessMapper;
|
|
private MoneyBusinessMapper businessMapper;
|
|
|
|
+ @Autowired
|
|
|
|
+ private SysOrgMapper orgMapper;
|
|
|
|
|
|
|
|
|
|
@Override
|
|
@Override
|
|
@@ -200,23 +204,25 @@ public class MoneyIncomeServiceImpl extends ServiceImpl<MoneyIncomeMapper, Money
|
|
|
|
|
|
@Override
|
|
@Override
|
|
public CommonResult add(MoneyIncomeAddParam addParam) {
|
|
public CommonResult add(MoneyIncomeAddParam addParam) {
|
|
- String orgId = addParam.getOrgId();
|
|
|
|
String time = addParam.getTime();
|
|
String time = addParam.getTime();
|
|
- QueryWrapper<MoneyIncome> queryWrapper = new QueryWrapper<>();
|
|
|
|
- queryWrapper.eq("year", time).eq("org_id", orgId);
|
|
|
|
- MoneyIncome moneyIncome = incomeMapper.selectOne(queryWrapper);
|
|
|
|
- if (ObjectUtil.isNotEmpty(moneyIncome)) {
|
|
|
|
- moneyIncome.setCompanyName(addParam.getCompany());
|
|
|
|
- moneyIncome.setYear(addParam.getTime());
|
|
|
|
- moneyIncome.setYearTarget(addParam.getYearTarget());
|
|
|
|
- incomeMapper.updateById(moneyIncome);
|
|
|
|
- } else {
|
|
|
|
- MoneyIncome income = new MoneyIncome();
|
|
|
|
- income.setCompanyName(addParam.getCompany());
|
|
|
|
- income.setYear(addParam.getTime());
|
|
|
|
- income.setYearTarget(addParam.getYearTarget());
|
|
|
|
- income.setOrgId(orgId);
|
|
|
|
- incomeMapper.insert(income);
|
|
|
|
|
|
+ List<SysOrg> sysOrgs = orgMapper.selectList(new QueryWrapper<SysOrg>().eq("CATEGORY", "COMPANY").eq("PARENT_ID", "1543842934270394368"));
|
|
|
|
+ for (SysOrg sysOrg : sysOrgs) {
|
|
|
|
+ QueryWrapper<MoneyIncome> queryWrapper = new QueryWrapper<>();
|
|
|
|
+ queryWrapper.eq("year", time).eq("org_id", sysOrg.getId());
|
|
|
|
+ MoneyIncome moneyIncome = incomeMapper.selectOne(queryWrapper);
|
|
|
|
+ if (ObjectUtil.isNotEmpty(moneyIncome)) {
|
|
|
|
+ moneyIncome.setCompanyName(sysOrg.getName());
|
|
|
|
+ moneyIncome.setYear(addParam.getTime());
|
|
|
|
+ moneyIncome.setYearTarget(addParam.getYearTarget());
|
|
|
|
+ incomeMapper.updateById(moneyIncome);
|
|
|
|
+ } else {
|
|
|
|
+ MoneyIncome income = new MoneyIncome();
|
|
|
|
+ income.setCompanyName(sysOrg.getName());
|
|
|
|
+ income.setYear(addParam.getTime());
|
|
|
|
+ income.setYearTarget(addParam.getYearTarget());
|
|
|
|
+ income.setOrgId(sysOrg.getId());
|
|
|
|
+ incomeMapper.insert(income);
|
|
|
|
+ }
|
|
}
|
|
}
|
|
return CommonResult.ok();
|
|
return CommonResult.ok();
|
|
}
|
|
}
|