车辆排班

xx-vehicle/xx-vehicle-server/src/main/java/com/xxfc/platform/vehicle/biz/VehicleBiz.java

@@ -22,6 +22,7 @@ import com.xxfc.platform.vehicle.common.RestResponse;
 import com.xxfc.platform.vehicle.constant.*;
 import com.xxfc.platform.vehicle.constant.ResCode.ResCode;
 import com.xxfc.platform.vehicle.entity.*;
+import com.xxfc.platform.vehicle.jobhandler.VehicleJobHandler;
 import com.xxfc.platform.vehicle.mapper.BookRecordAccItemMapper;
 import com.xxfc.platform.vehicle.mapper.VehicleBookInfoMapper;
 import com.xxfc.platform.vehicle.mapper.VehicleMapper;
@@ -89,6 +90,9 @@ public class VehicleBiz extends BaseBiz<VehicleMapper, Vehicle> implements UserR
     @Autowired
     private VehicleBookInfoBiz vehicleBookInfoBiz;
 
+    @Autowired
+    private VehicleJobHandler vehicleJobHandler;
+
 
     @Override
     public UserFeign getUserFeign() {
@@ -237,6 +241,8 @@ public class VehicleBiz extends BaseBiz<VehicleMapper, Vehicle> implements UserR
 //                }
             }
             mapper.insertSelective(vehicle);
+            vehicleJobHandler.InsertBookInfo(vehicle.getId());
+
         }
         return RestResponse.suc();
     }

xx-vehicle/xx-vehicle-server/src/main/java/com/xxfc/platform/vehicle/jobhandler/VehicleJobHandler.java

@@ -17,6 +17,7 @@ import org.springframework.stereotype.Component;
 import java.time.LocalDate;
 import java.util.List;
 import java.util.stream.Collectors;
+import java.util.stream.Stream;
 
 /**
  * @author libin
@@ -66,4 +67,21 @@ public class VehicleJobHandler extends IJobHandler {
             return FAIL;
         }
     }
+
+    public void  InsertBookInfo(String vehicleId){
+        XxlJobLogger.log("----添加新建的车辆【{}】的bookInfo",vehicleId);
+        List<VehicleBookInfo> bookInfos = Stream.of(0, 1, 2).map(x -> {
+            LocalDate date = LocalDate.now().plusMonths(x);
+            int year = date.getYear();
+            int month = date.getMonthValue();
+            String yearAndMonth = String.format("%d-%02d", year, month);
+
+            VehicleBookInfo vehicleBookInfo = new VehicleBookInfo();
+            vehicleBookInfo.setVehicle(vehicleId);
+            vehicleBookInfo.setYearMonth(yearAndMonth);
+            return vehicleBookInfo;
+        }).collect(Collectors.toList());
+        XxlJobLogger.log("----插入车型对象:【{}】",bookInfos);
+        vehicleBookInfoBiz.InsertBatch(bookInfos);
+    }
 }