更新脚本

src/main/java/com/xkrs/controller/DispatchFirePointController.java

@@ -6,14 +6,16 @@ import com.xkrs.dao.FirePointPreciseDao;
 import com.xkrs.dao.StreetDao;
 import com.xkrs.dao.SysUserDao;
 import com.xkrs.helper.FirePointSubscribeManager;
-import com.xkrs.model.entity.FirePointOrdinaryEntity;
+import com.xkrs.model.bean.VipTimeRangeBean;
+import com.xkrs.model.entity.SysUserEntity;
 import com.xkrs.model.qo.AllFirePointQo;
 import com.xkrs.service.DispatchFirePointService;
 import com.xkrs.service.impl.FirePointServiceImpl;
+import com.xkrs.utilsnew.JsonUtils;
+import com.xkrs.utilsold.DateTimeUtil;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.context.i18n.LocaleContextHolder;
-import org.springframework.data.domain.Sort;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestBody;
@@ -23,7 +25,6 @@ import javax.annotation.Resource;
 import java.util.ArrayList;
 import java.util.List;
 import java.util.Locale;
-import java.util.Map;
 
 import static com.xkrs.common.encapsulation.OutputEncapsulation.outputEncapsulationObject;
 
@@ -48,66 +49,42 @@ public class DispatchFirePointController {
     @GetMapping("/debug")
     public String debug() {
 
-//        List<SysUserEntity> sysUserEntityList = sysUserDao.findAll();
+        System.out.println("开始更新用户VIP时间范围");
-//        for (SysUserEntity sysUserEntity : sysUserEntityList) {
-//            String addTime = sysUserEntity.getAddTime();
-//            String vipTimeRangeJson = sysUserEntity.getVipTimeRangeJson();
-//            if (!DateTimeUtil.isValidDateTime(addTime)) {
-//                return sysUserEntity.getUserName();
-//            }
-//            if (!DateTimeUtil.isValidDateTime(vipTimeRangeJson)) {
-//                return sysUserEntity.getUserName();
-//            }
-//            VipTimeRangeBean.VipTimeRangeItemBean vipTimeRangeItemBean = new VipTimeRangeBean.VipTimeRangeItemBean();
-//            vipTimeRangeItemBean.setStartTime(addTime);
-//            vipTimeRangeItemBean.setEndTime(vipTimeRangeJson);
-//            List<VipTimeRangeBean.VipTimeRangeItemBean> list = new ArrayList<>();
-//            list.add(vipTimeRangeItemBean);
-//            VipTimeRangeBean vipTimeRangeBean = new VipTimeRangeBean();
-//            vipTimeRangeBean.setList(list);
-//            String serialize = JsonUtils.<VipTimeRangeBean>serialize(vipTimeRangeBean);
-//            System.out.println("serialize = " + serialize);
-//            sysUserEntity.setVipTimeRangeJson(serialize);
-//            sysUserDao.save(sysUserEntity);
-//        }
 
-        List<Map<String, String>> maps = streetDao.selectProCityPair();
+        List<SysUserEntity> sysUserEntityList = sysUserDao.findAll();
-        List<FirePointOrdinaryEntity> firePointOrdinaryEntityList = firePointOrdinaryDao.findAll(Sort.by(Sort.Direction.DESC, "satelliteTime"));
+        for (SysUserEntity sysUserEntity : sysUserEntityList) {
-        int size = firePointOrdinaryEntityList.size();
+            String addTime = sysUserEntity.getAddTime();
-        List<String> errorList = new ArrayList<>();
+            String vipTimeRangeJson = sysUserEntity.getVipTimeRangeJson();
-        for (int i = 0; i < size; i++) {
+            if (!DateTimeUtil.isValidDateTime(addTime)) {
-            FirePointOrdinaryEntity firePointOrdinaryEntity = firePointOrdinaryEntityList.get(i);
+                return sysUserEntity.getUserName();
-            String countyCode = firePointOrdinaryEntity.getCountyCode();
-            String proCode = countyCode.substring(0, 2) + "0000";
-            firePointOrdinaryEntity.setProCode(proCode);
-            firePointOrdinaryEntity.setProName(obtainProName(maps, proCode));
-            String cityCode = countyCode.substring(0, 4) + "00";
-            firePointOrdinaryEntity.setCityCode(cityCode);
-            firePointOrdinaryEntity.setCityName(obtainCityName(maps, cityCode));
-            firePointOrdinaryDao.save(firePointOrdinaryEntity);
-            if (i % 100 == 0) {
-                System.out.println(i + "/" + size);
             }
+            if (!DateTimeUtil.isValidDateTime(vipTimeRangeJson)) {
+                return sysUserEntity.getUserName();
             }
-        return outputEncapsulationObject(PromptMessageEnum.SUCCESS, errorList, locale);
+            VipTimeRangeBean.VipTimeRangeItemBean vipTimeRangeItemBean = new VipTimeRangeBean.VipTimeRangeItemBean();
+            vipTimeRangeItemBean.setStartTime(addTime);
+            vipTimeRangeItemBean.setEndTime(vipTimeRangeJson);
+            List<VipTimeRangeBean.VipTimeRangeItemBean> list = new ArrayList<>();
+            list.add(vipTimeRangeItemBean);
+            VipTimeRangeBean vipTimeRangeBean = new VipTimeRangeBean();
+            vipTimeRangeBean.setList(list);
+            String serialize = JsonUtils.<VipTimeRangeBean>serialize(vipTimeRangeBean);
+            System.out.println("serialize = " + serialize);
+            sysUserEntity.setVipTimeRangeJson(serialize);
+            sysUserDao.save(sysUserEntity);
         }
 
-    private String obtainProName(List<Map<String, String>> maps, String proCode) {
+        System.out.println("开始更新省编码市编码");
-        for (Map<String, String> map : maps) {
+        firePointOrdinaryDao.updateProCityCode();
-            if (map.get("pro_code").equals(proCode)) {
+        System.out.println("开始更新省名市名");
-                return map.get("pro_name");
+        firePointOrdinaryDao.updateProCityName();
-            }
-        }
-        return null;
-    }
 
-    private String obtainCityName(List<Map<String, String>> maps, String cityCode) {
+//        for (int i = 0; i < 30; i++) {
-        for (Map<String, String> map : maps) {
+//            firePointOrdinaryDao.updateProCityCodeById((long) i);
-            if (map.get("city_code").equals(cityCode)) {
+//            firePointOrdinaryDao.updateProCityNameById((long) i);
-                return map.get("city_name");
+//        }
-            }
+
-        }
+        return outputEncapsulationObject(PromptMessageEnum.SUCCESS, "OKKKK", locale);
-        return null;
     }
 
     @GetMapping("/autoSync")

src/main/java/com/xkrs/dao/FirePointOrdinaryDao.java

@@ -3,8 +3,27 @@ package com.xkrs.dao;
 import com.xkrs.model.entity.FirePointOrdinaryEntity;
 import org.springframework.data.jpa.repository.JpaRepository;
 import org.springframework.data.jpa.repository.JpaSpecificationExecutor;
+import org.springframework.data.jpa.repository.Modifying;
+import org.springframework.data.jpa.repository.Query;
 import org.springframework.stereotype.Component;
 
 @Component
 public interface FirePointOrdinaryDao extends JpaRepository<FirePointOrdinaryEntity, Long>, JpaSpecificationExecutor<FirePointOrdinaryEntity> {
+
+    @Query(value = "UPDATE fire_point_ordinary SET pro_code = concat(left(county_code, 2), '0000'), city_code = concat(left(county_code, 4), '00')", nativeQuery = true)
+    @Modifying(clearAutomatically = true)
+    void updateProCityCode();
+
+    @Query(value = "UPDATE fire_point_ordinary AS f SET pro_name = (SELECT s.pro_name AS proname FROM street AS s WHERE s.pro_code = f.pro_code LIMIT 1), city_name = (SELECT s.city_name AS cityname FROM street AS s WHERE s.city_code = f.city_code LIMIT 1)", nativeQuery = true)
+    @Modifying(clearAutomatically = true)
+    void updateProCityName();
+
+    @Query(value = "UPDATE fire_point_ordinary SET pro_code = concat(left(county_code, 2), '0000'), city_code = concat(left(county_code, 4), '00') WHERE id = ?1", nativeQuery = true)
+    @Modifying(clearAutomatically = true)
+    void updateProCityCodeById(Long id);
+
+    @Query(value = "UPDATE fire_point_ordinary AS f SET pro_name = (SELECT s.pro_name AS proname FROM street AS s WHERE s.pro_code = f.pro_code LIMIT 1), city_name = (SELECT s.city_name AS cityname FROM street AS s WHERE s.city_code = f.city_code LIMIT 1) WHERE id = ?1", nativeQuery = true)
+    @Modifying(clearAutomatically = true)
+    void updateProCityNameById(Long id);
+
 }