收入数据与成本分摊参数模板的导出

src/main/java/com/imed/costaccount/web/ExcelController.java

@@ -11,13 +11,8 @@ import com.imed.costaccount.common.exception.CostException;
 import com.imed.costaccount.common.util.Result;
 import com.imed.costaccount.common.util.UserContext;
 import com.imed.costaccount.constants.NumberConstant;
-import com.imed.costaccount.model.Accounting;
-import com.imed.costaccount.model.AccountingProduct;
-import com.imed.costaccount.model.Product;
-import com.imed.costaccount.model.User;
-import com.imed.costaccount.service.AccountingProductService;
-import com.imed.costaccount.service.AccountingService;
-import com.imed.costaccount.service.UserService;
+import com.imed.costaccount.model.*;
+import com.imed.costaccount.service.*;
 import com.imed.costaccount.service.impl.DepartmentServiceImpl;
 import com.imed.costaccount.service.impl.ProductServiceImpl;
 import io.swagger.annotations.Api;
@@ -25,6 +20,8 @@ import io.swagger.annotations.ApiOperation;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.poi.ss.usermodel.Sheet;
 import org.apache.shiro.SecurityUtils;
+import org.jetbrains.annotations.NotNull;
+import org.springframework.util.CollectionUtils;
 import org.springframework.web.bind.annotation.*;
 import org.springframework.web.multipart.MultipartFile;
 
@@ -55,12 +52,18 @@ public class ExcelController {
 
     private final AccountingProductService accountingProductService;
 
-    public ExcelController(UserService userService, DepartmentServiceImpl departmentService, ProductServiceImpl productService, AccountingService accountingService, AccountingProductService accountingProductService) {
+    private final ResponsibilityDepartmentService responsibilityDepartmentService;
+
+    private final CostShareParamService costShareParamService;
+
+    public ExcelController(UserService userService, DepartmentServiceImpl departmentService, ProductServiceImpl productService, AccountingService accountingService, AccountingProductService accountingProductService, ResponsibilityDepartmentService responsibilityDepartmentService, CostShareParamService costShareParamService) {
         this.userService = userService;
         this.departmentService = departmentService;
         this.productService = productService;
         this.accountingService = accountingService;
         this.accountingProductService = accountingProductService;
+        this.responsibilityDepartmentService = responsibilityDepartmentService;
+        this.costShareParamService = costShareParamService;
     }
 
     @ApiOperation("用户导出模板设置")
@@ -170,7 +173,7 @@ public class ExcelController {
     /**
      *  收入成本项目批量导入模板
      */
-    @ApiOperation("收入成本导出模板设置")
+    @ApiOperation("成本项目导出模板设置")
     @GetMapping("/getImportProductTemplate")
     public void getImportProductTemplate(HttpServletResponse response) throws IOException {
         User user = (User) SecurityUtils.getSubject().getPrincipal();
@@ -206,7 +209,7 @@ public class ExcelController {
      *  收入成本项目批量导入
      */
     @PostMapping("/importProduct")
-    @ApiOperation("批量导入收入成本信息")
+    @ApiOperation("批量导入成本项目信息")
     public Result importProduct(@RequestParam("file") MultipartFile file){
         InputStream in;
         try {
@@ -225,11 +228,10 @@ public class ExcelController {
     }
     /**
      * 收入成本数据导出模板
-     * @param accountType 1收入数据  2 成本数据(支出数据)
      */
     @ApiOperation("收入数据导出模板设置")
-    @GetMapping("/getImportProductAccountTemplate")
-    public void getImportProductAccountTemplate(HttpServletResponse response,Integer accountType) throws IOException {
+    @GetMapping("/getImportIncomeProductAccountTemplate")
+    public void getImportProductAccountTemplate(HttpServletResponse response) throws IOException {
         Long hospId = UserContext.getHospId();
         String uuid = UUID.randomUUID().toString();
         String url = System.getProperty("java.io.tmpdir") + File.separator + uuid + File.separator + uuid + ".xls";
@@ -242,37 +244,14 @@ public class ExcelController {
         writer.merge(0, 1, 0, 6, "为了保证成功导入，请勿修改模板格式", false);
         writer.passCurrentRow();
         writer.passCurrentRow();
-        String stateType= NumberConstant.ONE.equals(accountType)?"医院收入数据批量导入":"医院成本数据批量导入";
-        writer.merge(2, 2, 0, 6, stateType, false);
+        writer.merge(2, 2, 0, 6, "医院收入数据批量导入", false);
         writer.passCurrentRow();
         // 冻结前四行
         writer.setFreezePane(4);
         writer.writeRow(Arrays.asList("","","开单科室","开单科室代码", "执行科室","执行科室代码","金额"));
-        // 所有的成本项目
-        List<Product> productList = productService.list(new QueryWrapper<Product>().lambda().eq(Product::getHospId, hospId));
-        // 所有成本会计对照数据
-        List<AccountingProduct> accountingProductList = accountingProductService.list(new QueryWrapper<AccountingProduct>().lambda().eq(AccountingProduct::getHospId, hospId));
-        // 所有会计科目列表数据
-        List<Accounting> accountingList = accountingService.list(new QueryWrapper<Accounting>().lambda().eq(Accounting::getHospId, hospId));
-        List<Product> products = new ArrayList<>();
-        Map<Long, List<AccountingProduct>> accountProductMap = accountingProductList.stream().collect(Collectors.groupingBy(AccountingProduct::getProductId));
-        Map<Long, List<Accounting>> accountMap = accountingList.stream().collect(Collectors.groupingBy(Accounting::getId));
-        productList.forEach(i->{
-            Long productId = i.getId();
-            List<AccountingProduct> accountingProducts = accountProductMap.get(productId);
-            if (CollUtil.isNotEmpty(accountingProducts)){
-                Long accountingId = accountingProducts.get(0).getAccountingId();
-                List<Accounting> accountings = accountMap.get(accountingId);
-                if (CollUtil.isNotEmpty(accountings) && accountType.equals(accountings.get(0).getAccountingType())){
-                    products.add(i);
-                }
-            }
-        });
-        // 写入响应第一列 第二列的数据
-        for (int j = 0; j<products.size(); j++){
-            writer.writeCellValue(0,j+4,productList.get(j).getProductCode());
-            writer.writeCellValue(1,j+4,productList.get(j).getProductName());
-        }
+        int accountType=NumberConstant.ONE;
+        int column=NumberConstant.FOUR;
+        getProductByAccountType(hospId, writer,accountType,column);
         writer.setColumnWidth(0,10);
         writer.setColumnWidth(1,10);
         writer.setColumnWidth(2,20);
@@ -288,6 +267,9 @@ public class ExcelController {
         writer.close();
         IoUtil.close(out);
     }
+
+
+
     /**
      *  收入成本项目批量导入
      */
@@ -308,4 +290,148 @@ public class ExcelController {
             throw new CostException(500, "导入失败");
         }
     }
+    /**
+     * 成本数据导出模板
+     */
+    @GetMapping("/getImportCostProductAccountTemplate")
+    @ApiOperation("成本数据导出模板设置")
+    public void getImportCostProductAccountTemplate(HttpServletResponse response) throws IOException {
+        Long hospId = UserContext.getHospId();
+        String uuid = UUID.randomUUID().toString();
+        String url = System.getProperty("java.io.tmpdir") + File.separator + uuid + File.separator + uuid + ".xls";
+        FileUtil.del(FileUtil.file(url));
+
+        ExcelWriter writer = new ExcelWriter(url);
+        // 样式
+        Sheet sheet = writer.getSheet();
+        writer.merge(0, 1, 0, 6, "为了保证成功导入，请勿修改模板格式", false);
+        writer.passCurrentRow();
+        writer.passCurrentRow();
+        writer.merge(2, 2, 0, 6, "医院成本数据批量导入", false);
+        writer.passCurrentRow();
+        // 所有科室里面对应存在责任中心对应的科室
+        List<Department> departmentLinkedList = getDepartments(hospId);
+        // 设置科室的代码集合
+        List<String> departmentCodeList = departmentLinkedList.stream().map(Department::getDepartmentCode).collect(Collectors.toList());
+        List<String> departmentNameList = departmentLinkedList.stream().map(Department::getDepartmentName).collect(Collectors.toList());
+        departmentCodeList.add(0,null);
+        departmentCodeList.add(1,null);
+        writer.writeRow(departmentCodeList);
+        departmentNameList.add(0,"成本项目代码");
+        departmentNameList.add(1,"成本项目名称");
+        writer.writeRow(departmentNameList);
+        // 设置科室名称的集合
+        writer.setFreezePane(5);
+        int accountType=NumberConstant.TWO;
+        int column=NumberConstant.FIVE;
+        getProductByAccountType(hospId, writer,accountType,column);
+        writer.setColumnWidth(0,15);
+        writer.setColumnWidth(1,15);
+        response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;charset=utf-8");
+        response.setHeader("Content-Disposition", "attachment;filename=" + uuid + ".xls");
+        ServletOutputStream out = null;
+        out = response.getOutputStream();
+        writer.flush(out, true);
+        writer.close();
+        IoUtil.close(out);
+    }
+    /**
+     * 获取执行类型的成本项目
+     * @param hospId 医院Id
+     * @param writer 操作流
+     * @param accountType 会计科目类型 1 收入  2支出(成本)
+     * @param column 控制第几列
+     */
+    private void getProductByAccountType(Long hospId, ExcelWriter writer,Integer accountType,Integer column) {
+        // 所有的成本项目
+        List<Product> productList = productService.list(new QueryWrapper<Product>().lambda().eq(Product::getHospId, hospId));
+        // 所有成本会计对照数据
+        List<AccountingProduct> accountingProductList = accountingProductService.list(new QueryWrapper<AccountingProduct>().lambda().eq(AccountingProduct::getHospId, hospId));
+        // 所有会计科目列表数据
+        List<Accounting> accountingList = accountingService.list(new QueryWrapper<Accounting>().lambda().eq(Accounting::getHospId, hospId));
+        List<Product> products = new ArrayList<>();
+        Map<Long, List<AccountingProduct>> accountProductMap = accountingProductList.stream().collect(Collectors.groupingBy(AccountingProduct::getProductId));
+        Map<Long, List<Accounting>> accountMap = accountingList.stream().collect(Collectors.groupingBy(Accounting::getId));
+        productList.forEach(i->{
+            Long productId = i.getId();
+            List<AccountingProduct> accountingProducts = accountProductMap.get(productId);
+            if (CollUtil.isNotEmpty(accountingProducts)){
+                Long accountingId = accountingProducts.get(0).getAccountingId();
+                List<Accounting> accountings = accountMap.get(accountingId);
+                if (CollUtil.isNotEmpty(accountings) && accountType.equals(accountings.get(0).getAccountingType())){
+                    products.add(i);
+                }
+            }
+        });
+        // 写入响应第一列 第二列的数据
+        for (int j = 0; j<products.size(); j++){
+            writer.writeCellValue(0,j+column,productList.get(j).getProductCode());
+            writer.writeCellValue(1,j+column,productList.get(j).getProductName());
+        }
+    }
+    /**
+     * 成本分摊参数导出模板
+     */
+    @GetMapping("/getShareParamTemplate")
+    @ApiOperation("成本分摊参数导出模板")
+    public void getShareParamTemplate(HttpServletResponse response) throws IOException{
+        Long hospId = UserContext.getHospId();
+        String uuid = UUID.randomUUID().toString();
+        String url = System.getProperty("java.io.tmpdir") + File.separator + uuid + File.separator + uuid + ".xls";
+        FileUtil.del(FileUtil.file(url));
+
+        ExcelWriter writer = new ExcelWriter(url);
+        // 样式
+        Sheet sheet = writer.getSheet();
+        writer.merge(0, 1, 0, 6, "为了保证成功导入，请勿修改模板格式", false);
+        writer.passCurrentRow();
+        writer.passCurrentRow();
+        writer.merge(2, 2, 0, 6, "医院成本分摊参数数据批量导入", false);
+        writer.passCurrentRow();
+        // 所有科室里面对应存在责任中心对应的科室进行显示
+        List<Department> departmentLinkedList = getDepartments(hospId);
+        // 设置科室的代码集合
+        List<String> departmentCodeList = departmentLinkedList.stream().map(Department::getDepartmentCode).collect(Collectors.toList());
+        List<String> departmentNameList = departmentLinkedList.stream().map(Department::getDepartmentName).collect(Collectors.toList());
+        departmentCodeList.add(0,null);
+        departmentCodeList.add(1,null);
+        writer.writeRow(departmentCodeList);
+        departmentNameList.add(0,"成本分摊参数代码");
+        departmentNameList.add(1,"成本分摊参数名称");
+        writer.writeRow(departmentNameList);
+        // 设置科室名称的集合
+        writer.setFreezePane(5);
+        // 为第一列和第二列设置成本分摊参数的列表数据
+        List<CostShareParam> costShareParamList = costShareParamService.list(new QueryWrapper<CostShareParam>().lambda().eq(CostShareParam::getHospId, hospId));
+        for (int j = 0; j<costShareParamList.size(); j++){
+            writer.writeCellValue(0,j+5,costShareParamList.get(j).getShareParamCode());
+            writer.writeCellValue(1,j+5,costShareParamList.get(j).getShareParamName());
+        }
+        writer.setColumnWidth(0,15);
+        writer.setColumnWidth(1,15);
+        response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;charset=utf-8");
+        response.setHeader("Content-Disposition", "attachment;filename=" + uuid + ".xls");
+        ServletOutputStream out = null;
+        out = response.getOutputStream();
+        writer.flush(out, true);
+        writer.close();
+        IoUtil.close(out);
+    }
+
+    @NotNull
+    private List<Department> getDepartments(Long hospId) {
+        List<Department> departmentLinkedList = new LinkedList<>();
+        List<Department> departmentList = departmentService.list(new QueryWrapper<Department>().lambda().eq(Department::getHospId, hospId).orderByDesc(Department::getCreateTime));
+        List<ResponsibilityDepartment> responsibilityDepartmentList = responsibilityDepartmentService.list(new QueryWrapper<ResponsibilityDepartment>().lambda()
+                .eq(ResponsibilityDepartment::getHospId, hospId));
+        Map<Long, List<ResponsibilityDepartment>> responsibilityDepartmentMap = responsibilityDepartmentList.stream().collect(Collectors.groupingBy(ResponsibilityDepartment::getDepartmentId));
+        departmentList.forEach(i -> {
+            Long id = i.getId();
+            if (!CollectionUtils.isEmpty(responsibilityDepartmentMap.get(id))) {
+                // TODO 暂时先不考虑关联的责任中心是否存在
+                departmentLinkedList.add(i);
+            }
+        });
+        return departmentLinkedList;
+    }
 }