diff --git a/flyfish-data/src/main/java/com/flyfish/framework/domain/base/TreeQo.java b/flyfish-data/src/main/java/com/flyfish/framework/domain/base/TreeQo.java index 92ab739..585a097 100644 --- a/flyfish-data/src/main/java/com/flyfish/framework/domain/base/TreeQo.java +++ b/flyfish-data/src/main/java/com/flyfish/framework/domain/base/TreeQo.java @@ -4,6 +4,8 @@ import com.flyfish.framework.builder.CriteriaBuilder; import lombok.Getter; import lombok.Setter; +import java.util.List; + /** * 属性菜单的qo */ @@ -11,13 +13,15 @@ import lombok.Setter; @Setter public class TreeQo extends NameLikeQo { - private Boolean leaf; + private Integer depth; - private Integer level; + private String parentId; + + private List parentIds; @Override public CriteriaBuilder criteriaBuilder() { return super.criteriaBuilder() - .with("level", "leaf"); + .with("depth", "parentId").with("parentIds", "parentId", CriteriaBuilder.Builders.IN); } } diff --git a/flyfish-data/src/main/java/com/flyfish/framework/domain/po/Department.java b/flyfish-data/src/main/java/com/flyfish/framework/domain/po/Department.java index b6f9e7b..08290cc 100644 --- a/flyfish-data/src/main/java/com/flyfish/framework/domain/po/Department.java +++ b/flyfish-data/src/main/java/com/flyfish/framework/domain/po/Department.java @@ -1,6 +1,6 @@ package com.flyfish.framework.domain.po; -import com.flyfish.framework.domain.base.AuditDomain; +import com.flyfish.framework.domain.base.TreeDomain; import lombok.*; import org.springframework.data.mongodb.core.mapping.Document; @@ -15,7 +15,7 @@ import org.springframework.data.mongodb.core.mapping.Document; @NoArgsConstructor @AllArgsConstructor @Builder -public class Department extends AuditDomain { +public class Department extends TreeDomain { /** * 默认选中 @@ -32,23 +32,18 @@ public class Department extends AuditDomain { */ private String phone; - /** - * 层级 - */ - private Integer level; - /** * 地址 */ private String address; - /** - * 父节点,根节点0 - */ - private String parentId; - /** * 是否是部门(叶子节点) */ private Boolean leaf; + + /** + * 排序字段 + */ + private String pos; } diff --git a/flyfish-user/src/main/java/com/flyfish/framework/controller/DepartmentController.java b/flyfish-user/src/main/java/com/flyfish/framework/controller/DepartmentController.java index a302239..5d61adb 100644 --- a/flyfish-user/src/main/java/com/flyfish/framework/controller/DepartmentController.java +++ b/flyfish-user/src/main/java/com/flyfish/framework/controller/DepartmentController.java @@ -10,6 +10,6 @@ import org.springframework.web.bind.annotation.RestController; */ @RestController @RequestMapping("/departments") -public class DepartmentController extends BaseController { +public class DepartmentController extends TreeController { -} \ No newline at end of file +} diff --git a/flyfish-user/src/main/java/com/flyfish/framework/domain/DepartmentQo.java b/flyfish-user/src/main/java/com/flyfish/framework/domain/DepartmentQo.java index 99eff82..7988262 100644 --- a/flyfish-user/src/main/java/com/flyfish/framework/domain/DepartmentQo.java +++ b/flyfish-user/src/main/java/com/flyfish/framework/domain/DepartmentQo.java @@ -1,7 +1,6 @@ package com.flyfish.framework.domain; -import com.flyfish.framework.builder.CriteriaBuilder; -import com.flyfish.framework.domain.base.NameLikeQo; +import com.flyfish.framework.domain.base.TreeQo; import com.flyfish.framework.domain.po.Department; import lombok.Getter; import lombok.Setter; @@ -13,15 +12,6 @@ import lombok.Setter; */ @Getter @Setter -public class DepartmentQo extends NameLikeQo { +public class DepartmentQo extends TreeQo { - private Boolean leaf; - - private Integer level; - - @Override - public CriteriaBuilder criteriaBuilder() { - return super.criteriaBuilder() - .with("level", "leaf"); - } } diff --git a/flyfish-user/src/main/java/com/flyfish/framework/domain/PermissionQo.java b/flyfish-user/src/main/java/com/flyfish/framework/domain/PermissionQo.java index 88cf015..24f91e4 100644 --- a/flyfish-user/src/main/java/com/flyfish/framework/domain/PermissionQo.java +++ b/flyfish-user/src/main/java/com/flyfish/framework/domain/PermissionQo.java @@ -2,14 +2,11 @@ package com.flyfish.framework.domain; import com.flyfish.framework.builder.CriteriaBuilder; -import com.flyfish.framework.domain.base.NameLikeQo; import com.flyfish.framework.domain.base.TreeQo; import com.flyfish.framework.domain.po.Permission; import lombok.Getter; import lombok.Setter; -import java.util.List; - /** * 权限查询实体 * @@ -21,15 +18,10 @@ public class PermissionQo extends TreeQo { private Boolean admin; - private String parentId; - private String type; - private List parentIds; - @Override public CriteriaBuilder criteriaBuilder() { - return super.criteriaBuilder().with("admin", "parentId", "type") - .with("parentIds", "parentId", CriteriaBuilder.Builders.IN); + return super.criteriaBuilder().with("admin", "type"); } }