From 94509204d25f7c0ad213ae2322be2bd5bfd17424 Mon Sep 17 00:00:00 2001
From: liding <756868258@qq.com>
Date: 星期一, 14 七月 2025 16:30:28 +0800
Subject: [PATCH] 1.初始化配煤计算器数据 2。配煤到待入库
---
ruoyi-admin/src/main/java/com/ruoyi/web/controller/init/MyStartupRunner.java | 175 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 173 insertions(+), 2 deletions(-)
diff --git a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/init/MyStartupRunner.java b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/init/MyStartupRunner.java
index 9129ba7..771817d 100644
--- a/ruoyi-admin/src/main/java/com/ruoyi/web/controller/init/MyStartupRunner.java
+++ b/ruoyi-admin/src/main/java/com/ruoyi/web/controller/init/MyStartupRunner.java
@@ -1,11 +1,182 @@
package com.ruoyi.web.controller.init;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.ruoyi.basic.dto.CoalFieldDto;
+import com.ruoyi.basic.dto.CoalPlanDto;
+import com.ruoyi.basic.entity.City;
+import com.ruoyi.basic.entity.District;
+import com.ruoyi.basic.entity.Province;
+import com.ruoyi.basic.mapper.*;
+import com.ruoyi.basic.service.CoalFieldService;
+import com.ruoyi.basic.service.CoalPlanService;
+import com.ruoyi.common.exception.base.BaseException;
+import com.ruoyi.web.controller.init.dto.AreaDTO;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.CommandLineRunner;
+import org.springframework.core.io.ClassPathResource;
import org.springframework.stereotype.Component;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Set;
+import java.util.function.Supplier;
+import java.util.stream.Collectors;
+import java.util.stream.Stream;
+
+@Slf4j
@Component
-public class MyStartupRunner implements CommandLineRunner{
+public class MyStartupRunner implements CommandLineRunner {
+ @Autowired
+ private ProvinceMapper provinceMapper;
+ @Autowired
+ private CityMapper cityMapper;
+ @Autowired
+ private DistrictMapper districtMapper;
+ @Autowired
+ private CoalFieldService coalFieldService;
+ @Autowired
+ private CoalPlanService coalPlanService;
+
+ /**
+ * 椤圭洰鍚姩鏃堕渶瑕佹墽琛岀殑鏂规硶
+ *
+ * @param args
+ * @throws Exception
+ */
@Override
public void run(String... args) throws Exception {
- System.out.println("锛侊紒锛侊紒锛侊紒锛侊紒锛侊紒锛侊紒锛侊紒锛侊紒锛侊紒锛侊紒锛�");
+ // 鍒濆鍖栫渷甯傚尯
+ initRegion();
+ // 鍒濆鏂板閰嶇叅璁＄畻鍣ㄤ腑鐨勭叅绉嶅瓧娈靛拰鏂规
+ initCoalFields();
+ }
+
+ private void initRegion() {
+ // 鍒ゆ柇鏄槸鍚︽湁鐪佸競鍖轰俊鎭�
+ List<Province> olds = provinceMapper.selectList(null);
+ if (olds.isEmpty()) {
+ // 璇诲彇鍦板尯json鏂囦欢
+ List<AreaDTO> areaDTOS = null;
+ try {
+ // 浠� resources/doc/ 鍔犺浇 JSON 鏂囦欢
+ ClassPathResource resource = new ClassPathResource("doc/area.json");
+
+ ObjectMapper objectMapper = new ObjectMapper();
+ areaDTOS = objectMapper.readValue(
+ resource.getInputStream(),
+ objectMapper.getTypeFactory().constructCollectionType(List.class, AreaDTO.class)
+ );
+ } catch (Exception e) {
+ log.error("璇诲彇鍦板尯json鏂囦欢澶辫触:" + e);
+ }
+
+ List<Province> provinces = new ArrayList<>();
+ List<City> cities = new ArrayList<>();
+ List<District> districts = new ArrayList<>();
+
+ if (!areaDTOS.isEmpty()) {
+ log.info("鍦板尯淇℃伅鍒濆鍖栧紑濮�");
+
+ // 鐪�
+ areaDTOS.forEach(areaDTO -> {
+ Province province = new Province();
+ province.setId(Long.parseLong(areaDTO.getNo()));
+ province.setName(areaDTO.getName());
+ provinces.add(province);
+
+ if (!areaDTO.getChildren().isEmpty()) {
+ // 甯�
+ areaDTO.getChildren().forEach(cityDTO -> {
+ City city = new City();
+ city.setId(Long.parseLong(cityDTO.getNo()));
+ city.setName(cityDTO.getName());
+ city.setProvinceId(Long.valueOf(areaDTO.getNo()));
+ cities.add(city);
+
+ if (!cityDTO.getChildren().isEmpty()) {
+ // 鍖�
+ cityDTO.getChildren().forEach(districtDTO -> {
+ District district = new District();
+ district.setId(Long.parseLong(districtDTO.getNo()));
+ district.setName(districtDTO.getName());
+ district.setCityId(Long.valueOf(cityDTO.getNo()));
+ districts.add(district);
+ });
+ }
+ });
+ }
+ });
+
+
+ provinceMapper.insert(provinces);
+ cityMapper.insert(cities);
+ districtMapper.insert(districts);
+ log.info("鍦板尯淇℃伅鍒濆鍖栫粨鏉�");
+ }
+ }
+
+
+ }
+
+ private void initCoalFields() {
+ try {
+ // 1. 鍒濆鍖栧瓧娈甸�昏緫
+ Supplier<CoalFieldDto> dtoSupplier = CoalFieldDto::new; // 鍋囪瀛樺湪鏃犲弬鏋勯�犲嚱鏁�
+
+ List<CoalFieldDto> fields = Stream.of("鍙戠儹閲�", "纭垎", "鐏板垎", "姘村垎")
+ .map(fieldName -> {
+ CoalFieldDto dto = dtoSupplier.get();
+ dto.setFieldName(fieldName); // 鍋囪瀛樺湪 setFieldName 鏂规硶
+ dto.setFieldDescription("閰嶇叅璁＄畻鍣ㄥ獟璐ㄥ瓧娈�");
+ return dto;
+ })
+ .toList();
+
+ // 2. 淇濆瓨瀛楁锛堝鏋滀笉瀛樺湪锛�
+ Set<String> existingNames = coalFieldService.getFieldNamesByNames(
+ fields.stream().map(CoalFieldDto::getFieldName).collect(Collectors.toSet())
+ );
+
+ fields.forEach(field -> {
+ if (!existingNames.contains(field.getFieldName())) {
+ coalFieldService.addOrEditCoalField(field);
+ }
+ });
+
+ // 3. 鏌ヨ鎵�鏈夊瓧娈礗D
+ List<CoalFieldDto> allFields = coalFieldService.getFieldsByNames(
+ fields.stream().map(CoalFieldDto::getFieldName).collect(Collectors.toSet())
+ );
+
+ // 4. 鍒涘缓鏂规
+
+ String planName = "閰嶇叅璁＄畻鍣ㄦ柟妗�";
+ boolean planExists = coalPlanService.checkPlanExists(allFields.stream()
+ .map(f -> f.getId().toString())
+ .collect(Collectors.joining(",")));
+
+ if (!planExists) {
+ // 鍒涘缓鏂版柟妗�
+ CoalPlanDto planDto = new CoalPlanDto();
+ planDto.setPlan(planName);
+ planDto.setCoalFields(allFields.stream()
+ .map(CoalFieldDto::getFields)
+ .collect(Collectors.joining(",")));
+ planDto.setFieldIds(allFields.stream()
+ .map(f -> f.getId().toString())
+ .collect(Collectors.joining(",")));
+ planDto.setSchemeDesc("绯荤粺鐢熸垚閰嶇叅璁＄畻鍣ㄦ墍闇�瀛楁鏂规");
+
+ coalPlanService.addOrEditCoalPlan(planDto);
+ log.info("鎴愬姛鍒涘缓閰嶇叅璁＄畻鍣ㄦ柟妗�");
+ } else {
+ log.info("閰嶇叅璁＄畻鍣ㄦ柟妗堝凡瀛樺湪锛岃烦杩囧垱寤�");
+ }
+ } catch (Exception e) {
+ // 璁板綍绯荤粺寮傚父鏃ュ織锛屽苟杞崲涓轰笟鍔″紓甯告姏鍑猴紙瑙﹀彂鍥炴粴锛�
+ log.error("閰嶇叅璁＄畻鍣ㄥ垵濮嬪寲澶辫触", e);
+ throw new BaseException("鍒濆鍖栭厤鐓ゆ柟妗堝け璐ワ紝璇风◢鍚庨噸璇�");
+ }
}
}
--
Gitblit v1.9.3