修改后台模版管理和资源管理bug

src/main/java/com/lqkj/link/module/zone/repository/GeomInfoRepository.java

@@ -89,4 +89,15 @@ public interface GeomInfoRepository extends JpaRepository<GeomInfo, String> {
                 "    geom_name;"
     )
     List<GeomInfo> getNavigationInfo(Integer zoneId);
+
+
+    @Query(nativeQuery = true,
+        value = "select count(*) from geom_info where model_id in :infoIds and "
+    )
+    Integer findByModelIdIn(List<Integer> infoIds);
+
+    @Query(nativeQuery = true,
+        value = "select count(*) from geom_info where model_id in :infoIds and zone_id != :zoneId"
+    )
+    Integer findByModelIdInAndZoneId(List<Integer> infoIds, Integer zoneId);
 }

src/main/java/com/lqkj/link/module/zone/repository/ModelInfoRepository.java

@@ -47,4 +47,16 @@ public interface ModelInfoRepository extends JpaRepository<ModelInfo, Integer> {
         value = "update model_info set category_id = null where model_id in :deleteList"
     )
     void updateAllByIdInBatch(List<Integer> deleteList);
+
+
+    @Query(nativeQuery = true,
+        value = "select model_id from model_info where category_id = :categoryId"
+    )
+    List<Integer> findByCategoryId(Integer categoryId);
+
+
+    @Query(nativeQuery = true,
+        value = "select model_id from model_info where template_id = :zoneId"
+    )
+    List<Integer> findByTemplateId(Integer zoneId);
 }

src/main/java/com/lqkj/link/module/zone/service/ResourceService.java

@@ -4,6 +4,7 @@ import com.lqkj.link.module.authority.domain.UserInfo;
 import com.lqkj.link.module.authority.repository.UserInfoRepository;
 import com.lqkj.link.module.zone.domain.ModelCategory;
 import com.lqkj.link.module.zone.domain.ModelInfo;
+import com.lqkj.link.module.zone.repository.GeomInfoRepository;
 import com.lqkj.link.module.zone.repository.ModelCategoryRepository;
 import com.lqkj.link.module.zone.repository.ModelInfoRepository;
 import com.lqkj.link.util.AliOSSUtils;
@@ -11,6 +12,7 @@ import com.lqkj.link.util.FileUtils;
 import com.lqkj.link.util.Unzipper;
 import jakarta.servlet.http.HttpServletRequest;
 import org.apache.commons.compress.archivers.ArchiveException;
+import org.checkerframework.checker.units.qual.A;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.data.domain.Page;
 import org.springframework.data.domain.PageRequest;
@@ -32,6 +34,9 @@ public class ResourceService {
     @Autowired
     private AliOSSUtils aliOSSUtils;
 
+    @Autowired
+    private GeomInfoRepository geomInfoRepository;
+
     public ResourceService(ModelCategoryRepository categoryRepository, ModelInfoRepository infoRepository, UserInfoRepository userInfoRepository) {
         this.categoryRepository = categoryRepository;
         this.infoRepository = infoRepository;
@@ -62,6 +67,7 @@ public class ResourceService {
         category = categoryRepository.save(category);
         String filePath = category.getCompressFilePath();
         if (uploadCompress) {
+            checkIsReferenced(category.getCategoryId());
 //            infoRepository.deleteWithCategory(category.getCategoryId());
 //            Unzipper.unzip("." + filePath, "./upload/resource/" + category.getCategoryId() + "/", false);
             Unzipper.unZipFiles(new File("." + filePath), "./upload/resource/" + category.getCategoryId() + "/");
@@ -122,6 +128,14 @@ public class ResourceService {
         return null;
     }
 
+    private void checkIsReferenced(Integer categoryId) {
+        List<Integer> infoIds = infoRepository.findByCategoryId(categoryId);
+        Integer count = geomInfoRepository.findByModelIdIn(infoIds);
+        if (count > 0) {
+            throw new RuntimeException("资源被引用，无法修改和删除");
+        }
+    }
+
     public void deleteModel(Integer modelId) {
         ModelInfo modelInfo = infoRepository.findById(modelId).get();
         infoRepository.deleteById(modelId);
@@ -130,6 +144,7 @@ public class ResourceService {
 
     @Transactional
     public void deleteCategory(List<Integer> categoryId) {
+        categoryId.forEach(this::checkIsReferenced);
         deleteOssFiles(categoryId);
         categoryRepository.deleteAllByIdInBatch(categoryId);
         // 更新用户资源刷新状态

src/main/java/com/lqkj/link/module/zone/service/ZoneInfoService.java

@@ -91,6 +91,7 @@ public class ZoneInfoService {
         }
         ZoneInfo zoneInfo1 = zoneInfoRepository.save(zoneInfo);
         if (StringUtils.isNotBlank(zoneInfo.getTemplateFilePath()) && isUpdated) {
+            checkIsReferenced(zoneInfo1.getZoneId());
             // 清除元素与模型
             geomInfoRepository.deleteAllByZoneId(zoneInfo1.getZoneId());
             modelInfoRepository.deleteAllByTemplateIds(Collections.singletonList(zoneInfo1.getZoneId()));
@@ -260,6 +261,14 @@ public class ZoneInfoService {
         return result;
     }
 
+    private void checkIsReferenced(Integer zoneId) {
+        List<Integer> infoIds = modelInfoRepository.findByTemplateId(zoneId);
+        Integer count = geomInfoRepository.findByModelIdInAndZoneId(infoIds, zoneId);
+        if (count > 0) {
+            throw new RuntimeException("模板被引用，无法修改和删除");
+        }
+    }
+
     /**
      * 检查zip包内容
      * @param zoneInfo
@@ -287,8 +296,12 @@ public class ZoneInfoService {
 
     @Transactional
     public void deleteTemplate(List<Integer> zoneIds) {
-        zoneInfoRepository.deleteAllByIdInBatch(zoneIds);
-        modelInfoRepository.deleteAllByTemplateIds(zoneIds);
+        zoneIds.forEach(zoneId -> {
+            checkIsReferenced(zoneId);
+            zoneInfoRepository.deleteAllByIdInBatch(zoneIds);
+            modelInfoRepository.deleteAllByTemplateIds(zoneIds);
+
+        });
     }
 
     public List<String> autoCompleteMyWork(String userCode, String keyword) {