From ab0b2ea827a5116c1bcc8d57b082e14aff34e1d2 Mon Sep 17 00:00:00 2001
From: zhaowenxuan <chacca165@163.com>
Date: 星期一, 27 五月 2024 17:49:48 +0800
Subject: [PATCH] 修改线程池创建方式,增加时间非空判断

---
 ltkj-admin/src/main/java/com/ltkj/web/controller/system/TjOrderController.java |   98 ++++++++++++++++++++++++++++--------------------
 1 files changed, 57 insertions(+), 41 deletions(-)

diff --git a/ltkj-admin/src/main/java/com/ltkj/web/controller/system/TjOrderController.java b/ltkj-admin/src/main/java/com/ltkj/web/controller/system/TjOrderController.java
index d24c32c..afe75ae 100644
--- a/ltkj-admin/src/main/java/com/ltkj/web/controller/system/TjOrderController.java
+++ b/ltkj-admin/src/main/java/com/ltkj/web/controller/system/TjOrderController.java
@@ -23,6 +23,7 @@
 import cn.hutool.extra.pinyin.PinyinUtil;
 import com.alibaba.fastjson2.JSON;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.fasterxml.jackson.databind.BeanProperty;
@@ -35,6 +36,7 @@
 import com.ltkj.common.utils.PDFDocumentUtil;
 import com.ltkj.common.utils.SecurityUtils;
 import com.ltkj.framework.config.MatchUtils;
+import com.ltkj.framework.config.ThreadPoolConfig;
 import com.ltkj.framework.config.UserHoder;
 import com.ltkj.framework.datasource.DynamicDataSourceContextHolder;
 import com.ltkj.hosp.domain.*;
@@ -65,6 +67,7 @@
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.http.HttpRequest;
+import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
 import org.springframework.security.access.prepost.PreAuthorize;
 import org.springframework.transaction.annotation.Transactional;
 import org.springframework.util.DigestUtils;
@@ -164,6 +167,8 @@
     private LtkjMiddleHeadService headService;
     @Resource
     private TjvLtkjvtjpatService tjvLtkjvtjpatService;
+    @Autowired
+    private ThreadPoolConfig threadPoolConfig;
 
 
     /**
@@ -1037,7 +1042,8 @@
             wq.in(TjCustomer::getCusIdcard, cusIds);
             List<Long> list = tjCustomerService.list(wq).stream().map(TjCustomer::getCusId).collect(Collectors.toList());
             if (list.size() > 0) {
-                ExecutorService threadPool = Executors.newFixedThreadPool(list.size());
+//                ExecutorService threadPool = Executors.newFixedThreadPool(list.size());
+                ThreadPoolTaskExecutor executor = threadPoolConfig.threadPoolTaskExecutor();
                 for (Long aLong : list) {
 //                    threadPool.execute(new Runnable() {
 //                        @Override
@@ -1050,7 +1056,7 @@
 //                            }
 //                        }
 //                    });
-                    Future<AjaxResult> future = threadPool.submit(new Callable<AjaxResult>() {
+                    Future<AjaxResult> future = executor.submit(new Callable<AjaxResult>() {
                         @Override
                         public AjaxResult call() throws Exception {
                             return getAjaxResult(aLong, sysUser, daoDto.getTjCategory());
@@ -1064,10 +1070,9 @@
                         e.printStackTrace();
                     }
                 }
-                threadPool.shutdown();
+                threadPools.shutdown();
             }
         }
-        System.out.println("tjNumbers = " + tjNumbers);
         // TODO 鍥犱负鏈変綋妫�鍙蜂細鏌ュ嚭绌烘暟鎹� 鍏堜娇鐢ㄦ鏁版嵁
         tjNumbers = new ArrayList<>();
         tjNumbers.add("02095240229171627");
@@ -1086,11 +1091,14 @@
                 try {
                     List<Map<String, String>> djdInfos = tjOrderService.selectDjdInfo(tjNumbers.get(i));
                     Map<String, Object> userInfo = tjOrderService.selectDjdUserInfo(tjNumbers.get(i));
-                    System.out.println(userInfo);
                     LocalDateTime tjsj = (LocalDateTime) userInfo.get("tjsj");
                     DateTimeFormatter dateFormat = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm");
+                    String time = null;
+                    if (tjsj != null) {
+                        time = dateFormat.format(tjsj);
+                    }
                     PDFDocumentUtil.makeUserInfoTable(document, (String) userInfo.get("tjname"), (String) userInfo.get("sex")
-                            , (String) userInfo.get("tjh"), (String) userInfo.get("lxdh"), (String) userInfo.get("tjname"), dateFormat.format(tjsj));
+                            , (String) userInfo.get("tjh"), (String) userInfo.get("lxdh"), (String) userInfo.get("tjname"), time);
                     PDFDocumentUtil.makeTjInfo(document, djdInfos);
                     // 澶氭潯鍒欐崲椤�
                     if (i + 1 < tjNumbers.size()) {
@@ -1101,13 +1109,9 @@
                 }
             }
             document.close();
-            // TODO 淇敼杩斿洖鏂囦欢base64
             ByteArrayOutputStream outputStream = (ByteArrayOutputStream) map.get("stream");
-            // 璁剧疆HTTP鍝嶅簲澶�
-            response.setContentType("application/pdf");
-            response.setHeader("Content-Disposition", "inline; filename=\"downloaded.pdf\"");
-            response.getOutputStream().write(outputStream.toByteArray());
-            response.getOutputStream().flush();
+            String encodeToString = java.util.Base64.getEncoder().encodeToString(outputStream.toByteArray());
+            return AjaxResult.success().put("file", encodeToString);
         } catch (IOException | DocumentException e) {
             e.printStackTrace();
         }
@@ -2372,34 +2376,46 @@
         return AjaxResult.error("璇烽�夋嫨瑕佹挙閿�鐨勪汉鍛�!");
     }
 
-//    @PostMapping("makeTjPDF1")
-//    @ApiOperation(value = "鏍规嵁浣撴鍙风敓鎴怭DF")
-//    public AjaxResult makeTjPDF(@RequestBody List<String> tjNumbers, HttpServletResponse response) {
-//        System.out.println("tjNumbers = " + tjNumbers);
-//        Map<String, Object> map = null;
-//        try {
-//            map = PDFDocumentUtil.getDocument();
-//            Document document = (Document) map.get("document");
-//            for (int i = 0; i < tjNumbers.size(); i++) {
-//                try {
-//                    List<Map<String, String>> djdInfos = tjOrderService.selectDjdInfo(tjNumbers.get(i));
-//                    Map<String, Object> userInfo = tjOrderService.selectDjdUserInfo(tjNumbers.get(i));
-//                    PDFDocumentUtil.makeUserInfoTable(document, userInfo.get("tjname"), userInfo.get("sex"), userInfo.get("tjh"), userInfo.get("lxdh"), userInfo.get("tjname"), userInfo.get("tjsj"));
-//                    PDFDocumentUtil.makeTjInfo(document, djdInfos);
-//                    // 澶氭潯鍒欐崲椤�
-//                    if (i + 1 < tjNumbers.size()) {
-//                        document.newPage();
-//                    }
-//                } catch (DocumentException | IOException e) {
-//                    e.printStackTrace();
-//                }
-//            }
-//            document.close();
-//        } catch (IOException | DocumentException e) {
-//            e.printStackTrace();
-//        }
-//        return AjaxResult.success();
-//    }
-
+    @PostMapping("makeTjPDF")
+    @ApiOperation(value = "鎵归噺鐢熸垚PDF")
+    public AjaxResult makeTjPDF(@RequestBody List<String> reservations) {
+        QueryWrapper<TjOrder> queryWrapper = new QueryWrapper<>();
+        queryWrapper.in("reservation_id",reservations);
+        List<TjOrder> tjNumbers = tjOrderService.list(queryWrapper);
+        Map<String, Object> map = null;
+        try {
+            map = PDFDocumentUtil.getDocument();
+            Document document = (Document) map.get("document");
+            for (int i = 0; i < tjNumbers.size(); i++) {
+                String tjNumber = tjNumbers.get(i).getTjNumber();
+                try {
+                    List<Map<String, String>> djdInfos = tjOrderService.selectDjdInfo(tjNumber);
+                    Map<String, Object> userInfo = tjOrderService.selectDjdUserInfo(tjNumber);
+                    LocalDateTime tjsj = (LocalDateTime) userInfo.get("tjsj");
+                    DateTimeFormatter dateFormat = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm");
+                    String time = null;
+                    if (tjsj != null) {
+                        time = dateFormat.format(tjsj);
+                    }
+                    PDFDocumentUtil.makeUserInfoTable(document, (String) userInfo.get("tjname"), (String) userInfo.get("sex")
+                            , (String) userInfo.get("tjh"), (String) userInfo.get("lxdh"), (String) userInfo.get("tjname"), time);
+                    PDFDocumentUtil.makeTjInfo(document, djdInfos);
+                    // 澶氭潯鍒欐崲椤�
+                    if (i + 1 < tjNumbers.size()) {
+                        document.newPage();
+                    }
+                } catch (DocumentException | IOException e) {
+                    e.printStackTrace();
+                }
+            }
+            document.close();
+            ByteArrayOutputStream outputStream = (ByteArrayOutputStream) map.get("stream");
+            String encodeToString = java.util.Base64.getEncoder().encodeToString(outputStream.toByteArray());
+            return AjaxResult.success().put("file", encodeToString);
+        } catch (IOException | DocumentException e) {
+            e.printStackTrace();
+            return AjaxResult.error(e.getMessage());
+        }
+    }
 }
 

--
Gitblit v1.8.0