소스 검색

优化代码

Your Name 2 년 전
부모
커밋
5f54c5e9d4
1개의 변경된 파일8개의 추가작업 그리고 4개의 파일을 삭제
  1. 8 4
      service-store/src/main/java/com/java110/store/cmd/resourceStore/ListResourceStoreTypeTreeCmd.java

+ 8 - 4
service-store/src/main/java/com/java110/store/cmd/resourceStore/ListResourceStoreTypeTreeCmd.java

@@ -57,10 +57,14 @@ public class ListResourceStoreTypeTreeCmd extends Cmd {
             //todo 一级分类
             tResourceStoreTypeDto = getParentResoureceStoreTypes(storeTypes, tmpResourceStoreTypeDto);
 
-//            //todo 如果现在就是一级分类 跳过
-//            if (tmpResourceStoreTypeDto.getRstId().equals(tResourceStoreTypeDto.getRstId())) {
-//                continue;
-//            }
+            //todo 如果现在就是一级分类 跳过
+            if (tmpResourceStoreTypeDto.getRstId().equals(tResourceStoreTypeDto.getRstId())) {
+                subStoreTypes = new ArrayList<>();
+                ResourceStoreTypeDto selfResourceStoreTypeDto = BeanConvertUtil.covertBean(tmpResourceStoreTypeDto,ResourceStoreTypeDto.class);
+                subStoreTypes.add(selfResourceStoreTypeDto);
+                tResourceStoreTypeDto.setSubTypes(subStoreTypes);
+                continue;
+            }
 
             if (tResourceStoreTypeDto.getSubTypes() == null) {
                 tResourceStoreTypeDto.setSubTypes(new ArrayList<>());