diff --git a/src/com/engine/salary/entity/salaryacct/dto/SalaryAcctImportFieldDTO.java b/src/com/engine/salary/entity/salaryacct/dto/SalaryAcctImportFieldDTO.java index 37b71dd32..b51422d68 100644 --- a/src/com/engine/salary/entity/salaryacct/dto/SalaryAcctImportFieldDTO.java +++ b/src/com/engine/salary/entity/salaryacct/dto/SalaryAcctImportFieldDTO.java @@ -24,6 +24,9 @@ public class SalaryAcctImportFieldDTO { //公式项 private Collection formulaItems; + //sql项 + private Collection sqlItems; + //输入项 private Collection inputItems; diff --git a/src/com/engine/salary/service/impl/SalaryAcctExcelServiceImpl.java b/src/com/engine/salary/service/impl/SalaryAcctExcelServiceImpl.java index e800229a9..a381436c0 100644 --- a/src/com/engine/salary/service/impl/SalaryAcctExcelServiceImpl.java +++ b/src/com/engine/salary/service/impl/SalaryAcctExcelServiceImpl.java @@ -311,6 +311,14 @@ public class SalaryAcctExcelServiceImpl extends Service implements SalaryAcctExc .salaryItemName(salaryItemPO.getName()) .build()) .collect(Collectors.toList()); + // 薪资项目中的SQl项 + List sqlItems = salaryItems.stream() + .filter(salaryItemPO -> Objects.equals(salaryItemPO.getValueType(), SalaryValueTypeEnum.SQL.getValue())) + .map(salaryItemPO -> SalaryAcctImportFieldDTO.ImportFieldDTO.builder() + .salaryItemId(salaryItemPO.getId()) + .salaryItemName(salaryItemPO.getName()) + .build()) + .collect(Collectors.toList()); // 薪资项目中的输入项 List inputItems = salaryItems.stream() .filter(salaryItemPO -> Objects.equals(salaryItemPO.getValueType(), SalaryValueTypeEnum.INPUT.getValue())) @@ -320,7 +328,7 @@ public class SalaryAcctExcelServiceImpl extends Service implements SalaryAcctExc .build()) .collect(Collectors.toList()); // 转换成dto - return SalaryAcctImportFieldDTO.builder().formulaItems(formulaItems).inputItems(inputItems).build(); + return SalaryAcctImportFieldDTO.builder().formulaItems(formulaItems).sqlItems(sqlItems).inputItems(inputItems).build(); } @Override