Browse Source

refactor: streamline tree node property access in ${ClassName}ServiceImpl for improved readability

PIG AI 1 week ago
parent
commit
bc6a7aa506
1 changed files with 2 additions and 8 deletions
  1. 2 8
      tree/树形ServiceImpl.java

+ 2 - 8
tree/树形ServiceImpl.java

@@ -65,24 +65,18 @@ public class ${ClassName}ServiceImpl extends ServiceImpl<${ClassName}Mapper, ${C
     private Function<${ClassName}Entity, TreeNode<${pk.attrType}>> getNodeFunction() {
         return entity -> {
             TreeNode<${pk.attrType}> node = new TreeNode<>();
-            node.setId(entity.get${str.capitalizeFirst($pk.attrName)}());
+            node.setId(entity.$str.getProperty($pk.attrName)());
 #foreach($field in $fieldList)
 #if($field.attrName == 'name' || $field.fieldComment.contains('名称'))
             node.setName(entity.get${str.capitalizeFirst($field.attrName)}());
 #end
 #end
-#if($pk.attrType == 'Long')
             node.setParentId(entity.getParentId() != null ? entity.getParentId() : 0L);
-#else
-            node.setParentId(entity.getParentId() != null ? entity.getParentId() : 0);
-#end
 
             // 扩展属性
             Map<String, Object> extra = new HashMap<>();
 #foreach($field in $fieldList)
-#if(!$field.primaryPk && $field.attrName != 'parentId')
-            extra.put(${ClassName}Entity.Fields.${field.attrName}, entity.get${str.capitalizeFirst($field.attrName)}());
-#end
+            extra.put(${ClassName}Entity.Fields.${field.attrName}, entity.$str.getProperty($field.attrName)());
 #end
             
             node.setExtra(extra);