修改代码

pz-admin/src/main/java/com/pz/web/task/InsertTestOrderTask.java

@@ -23,6 +23,9 @@ import org.springframework.stereotype.Component;
 import javax.annotation.PostConstruct;
 import java.math.BigDecimal;
 import java.util.*;
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.Executors;
+import java.util.concurrent.TimeUnit;
 
 /**
  * 每天12点新增订单数据
@@ -63,38 +66,31 @@ public class InsertTestOrderTask {
     private final EmployeesDivideMapper employeesDivideMapper;
 
     public static List<Integer> businessIds = Arrays.asList(0, 1, 2, 3, 4, 5, 6);
-
-    @Scheduled(cron = "0 0 0 * * ?") // 每5秒执行一次
-    public void insertOrder() {
-        //每日单量
-        Integer num = RandomUtil.randomInt(30, 50);
-
-        //就诊人员
-        List<UserVsitor> userVsitors = userVsitorMapper.selectList(null);
-
+    @Scheduled(cron = "0 0 0 * * ?") // 凌晨12点执行
+    public void insertOrder() throws InterruptedException {
+        // 就诊人员
+        List<UserVsitor> userVsitors = userVsitorMapper.selectList(Wrappers.<UserVsitor>lambdaQuery().last("limit 20"));
         UserVsitor randomElement = getRandomElement(userVsitors);
-        for (int i = 0; i <= num; i++) {
-            //遍历固定机构
+        // 查询固定机构
         List<Company> companies = companyMapper.selectList(Wrappers.<Company>lambdaQuery().in(Company::getId, 5, 14, 15, 16, 17));
+        // 加入线程池并发执行
+        ExecutorService executorService = Executors.newFixedThreadPool(10);
         for (Company company : companies) {
-                //获取机构下的陪诊员
             Employees employees = employeesMapper.selectOne(Wrappers.<Employees>lambdaQuery().eq(Employees::getCompanyId, company.getId()).last("limit 1"));
-                //随机获取陪诊员所在城市的随机医院
+            executorService.submit(() -> {
                 List<Hospital> hospitals = hospitalMapper.selectList(Wrappers.<Hospital>lambdaQuery().eq(Hospital::getCityId, employees.getCityId()).last("limit 10"));
-                //随机获取城市中某个医院
                 Hospital hospital = getRandomElement(hospitals);
-                //遍历所有业务
                     for (Integer businessId : businessIds) {
-                    //主订单编号
+                        // 每日单量
+                        Integer num = RandomUtil.randomInt(30, 50);
+                        for (int i = 0; i <= num; i++) {
                         Long orderId = saveTotalOrder(businessId, employees.getId(), employees.getCityId().intValue(), company.getId());
                         switch (businessId) {
-                        //商城订单
                             case 0:
                                 StoreOrder storeOrder = new StoreOrder();
                                 storeOrder.setOrderId(orderId.intValue());
                                 Integer storeId = RandomUtil.randomInt(1, 5);
                                 storeOrder.setGoodsId(storeId);
-                            //随机地址id
                                 Integer addressId = RandomUtil.randomInt(1, 9);
                                 storeOrder.setAddressId(addressId);
                                 storeOrder.setStatus(2);
@@ -102,7 +98,6 @@ public class InsertTestOrderTask {
                                 storeOrder.setLogisticsCode("SF" + RandomUtil.randomNumbers(16));
                                 storeOrderMapper.insert(storeOrder);
                                 break;
-                        //预约陪诊
                             case 1:
                                 YypzOrder yypzOrder = new YypzOrder();
                                 yypzOrder.setOrderId(orderId.intValue());
@@ -113,7 +108,6 @@ public class InsertTestOrderTask {
                                 yypzOrder.setVisitor(randomElement.getId());
                                 yypzOrderMapper.insert(yypzOrder);
                                 break;
-                        //代办挂号
                             case 2:
                                 DbghOrder dbghOrder = new DbghOrder();
                                 dbghOrder.setOrderId(orderId.intValue());
@@ -123,7 +117,6 @@ public class InsertTestOrderTask {
                                 dbghOrder.setVisitor(randomElement.getId());
                                 dbghOrderMapper.insert(dbghOrder);
                                 break;
-                        //代办问诊
                             case 3:
                                 DbwzOrder dbwzOrder = new DbwzOrder();
                                 dbwzOrder.setOrderId(orderId.intValue());
@@ -139,7 +132,6 @@ public class InsertTestOrderTask {
                                 dbwzOrder.setVisitor(randomElement.getId());
                                 dbwzOrderMapper.insert(dbwzOrder);
                                 break;
-                        //住院陪护
                             case 4:
                                 ZyphOrder zyphOrder = new ZyphOrder();
                                 zyphOrder.setOrderId(orderId.intValue());
@@ -149,7 +141,6 @@ public class InsertTestOrderTask {
                                 zyphOrder.setOverTime(new Date());
                                 zyphOrderMapper.insert(zyphOrder);
                                 break;
-                        //代办买药
                             case 5:
                                 DbmyOrder dbmyOrder = new DbmyOrder();
                                 dbmyOrder.setOrderId(orderId.intValue());
@@ -160,7 +151,6 @@ public class InsertTestOrderTask {
                                 dbmyOrder.setPzShPhone(employees.getPhone());
                                 dbmyOrderMapper.insert(dbmyOrder);
                                 break;
-                        //诊前挂号
                             case 6:
                                 ZqghOrder zqghOrder = new ZqghOrder();
                                 zqghOrder.setOrderId(orderId.intValue());
@@ -171,10 +161,14 @@ public class InsertTestOrderTask {
                                 zqghOrderMapper.insert(zqghOrder);
                                 break;
                             default:
+                                break;
                         }
                     }
                 }
+            });
         }
+        executorService.shutdown();
+        executorService.awaitTermination(Long.MAX_VALUE, TimeUnit.NANOSECONDS);
     }
 
     public Long saveTotalOrder(Integer businessId, Integer emId, Integer cityId, Integer companyId) {