Browse Source

refactor: unify parent ID property access method in ${ClassName}ServiceImpl for consistency and clarity

PIG AI 1 week ago
parent
commit
90199c3c55
1 changed files with 5 additions and 10 deletions
  1. 5 10
      tree/树形ServiceImpl.java

+ 5 - 10
tree/树形ServiceImpl.java

@@ -66,19 +66,14 @@ public class ${ClassName}ServiceImpl extends ServiceImpl<${ClassName}Mapper, ${C
         return entity -> {
             TreeNode<${pk.attrType}> node = new TreeNode<>();
             node.setId(entity.$str.getProperty($pk.attrName)());
-#foreach($field in $fieldList)
-#if($field.attrName == 'name' || $field.fieldComment.contains('名称'))
-            node.setName(entity.$str.getProperty($field.attrName)());
-#end
-#end
-            node.setParentId(entity.getParentId() != null ? entity.getParentId() : 0L);
+            node.setName(entity.$str.getProperty($nameField)());
+            node.$str.setProperty($parentIdField)(entity.$str.getProperty($parentIdField)() != null ? entity.$str.getProperty($parentIdField)() : 0L);
 
             // 扩展属性
             Map<String, Object> extra = new HashMap<>();
 #foreach($field in $fieldList)
             extra.put(${ClassName}Entity.Fields.${field.attrName}, entity.$str.getProperty($field.attrName)());
 #end
-            
             node.setExtra(extra);
             return node;
         };
@@ -92,7 +87,7 @@ public class ${ClassName}ServiceImpl extends ServiceImpl<${ClassName}Mapper, ${C
     public List<${ClassName}Entity> getParentNodes() {
         LambdaQueryWrapper<${ClassName}Entity> wrapper = Wrappers.lambdaQuery();
         // 查询所有父级节点(parentId为null或0的节点)
-        wrapper.and(w -> w.isNull(${ClassName}Entity::getParentId).or().eq(${ClassName}Entity::getParentId, 0));
+        wrapper.and(w -> w.isNull(${ClassName}Entity::$str.getProperty($parentIdField)).or().eq(${ClassName}Entity::$str.getProperty($parentIdField), 0));
         wrapper.orderByAsc(${ClassName}Entity::$str.getProperty($pk.attrName)); // 按主键排序
         return list(wrapper);
     }
@@ -105,7 +100,7 @@ public class ${ClassName}ServiceImpl extends ServiceImpl<${ClassName}Mapper, ${C
     @Override
     public List<${ClassName}Entity> getChildrenByParentId(${pk.attrType} parentId) {
         LambdaQueryWrapper<${ClassName}Entity> wrapper = Wrappers.lambdaQuery();
-        wrapper.eq(${ClassName}Entity::getParentId, parentId);
+        wrapper.eq(${ClassName}Entity::$str.getProperty($parentIdField), parentId);
         wrapper.orderByAsc(${ClassName}Entity::$str.getProperty($pk.attrName)); // 按主键排序
         return list(wrapper);
     }
@@ -146,7 +141,7 @@ public class ${ClassName}ServiceImpl extends ServiceImpl<${ClassName}Mapper, ${C
         List<${pk.attrType}> childIds = new ArrayList<>();
         
         LambdaQueryWrapper<${ClassName}Entity> wrapper = Wrappers.lambdaQuery();
-        wrapper.eq(${ClassName}Entity::getParentId, parentId);
+        wrapper.eq(${ClassName}Entity::$str.getProperty($parentIdField), parentId);
         wrapper.select(${ClassName}Entity::$str.getProperty($pk.attrName));
         
         List<${ClassName}Entity> children = list(wrapper);