diff --git a/src/main/java/com/dc/dc_project/service/impl/StandardCategoryServiceImpl.java b/src/main/java/com/dc/dc_project/service/impl/StandardCategoryServiceImpl.java index 5d3892f..f18c244 100644 --- a/src/main/java/com/dc/dc_project/service/impl/StandardCategoryServiceImpl.java +++ b/src/main/java/com/dc/dc_project/service/impl/StandardCategoryServiceImpl.java @@ -14,6 +14,7 @@ import com.dc.dc_project.utils.TreeUtil; import org.springframework.stereotype.Service; import lombok.extern.slf4j.Slf4j; +import java.util.Comparator; import java.util.List; import java.util.stream.Collectors; @@ -117,7 +118,10 @@ public class StandardCategoryServiceImpl extends ServiceImpl list = this.list(queryWrapper); - List tree = TreeUtil.buildTree(list.stream().map(StandardCategoryVo::poToVo).collect(Collectors.toList())); + List tree = TreeUtil.buildTreeWithSort( + list.stream().map(StandardCategoryVo::poToVo).collect(Collectors.toList()), + Comparator.comparing(StandardCategoryVo::getSortOrder) + ); return ResponseResult.success(tree); } }