【一起学设计模式】访问者模式实战：权限管理树删除节点操作

前言

申明：
本文章首发自本人博客：https://www.cnblogs.com/wang-meng 和公众号：壹枝花算不算浪漫 如若转载请标明来源

之前在我的博客(一枝花算不算浪漫)中已经更新过两篇设计模式相关的内容

• 【一起学设计模式】策略模式实战一：基于消息发送的策略模式实战
• 【一起学习设计模式】策略模式实战二：配合注解 干掉业务代码中冗余的if else...

上面内容都是基于真实业务场景精简后的设计(工作中真实场景使用到的)。

之前为了学习设计模式，看过网上很多相关博客讲解，大都是画下UML类图，举例几个毫不相干的demo，看了几遍仍然是云里雾里。

学习设计模式只有在真正的业务场景去使用才会更好的理解其精髓。这里举例自己工作中电商的业务场景，然后配合一些业务功能的实现，来学会设计模式，使自己的代码更优雅。

业务背景

权限功能模块-权限树-删除树上的某个权限

1. 要求判断该权限节点及其子节点是否有用户、角色关联，如若有关联则不允许删除
2. 要求删除该权限节点及其子节点所有数据

常规操作

先说下大多数人为了实现需求都会做的常规操作，这里举例说明，权限A，其子节点B、C

1. 查找权限A是否被用户、角色关联过
2. 查找全新啊A下面所有子节点B、C
3. 查找权限B、C是否被其他用户、角色关联过
4. 删除A、B、C

这里如果有个流程图效果会更佳，但是相信大家看到文字也能明白其中的运转流程。

这里只是简单地列了下操作的步骤，其实大家可能会做的更好，比如查询、删除 都可以批量去做处理，这里就不再讨论了。

通过上面的流程，我们知道一个方法就可以搞定这几个步骤，只是该方法包含了查询、删除等等逻辑操作，看起来并不精简。

访问者模式实现

1. 实现类图
   

2. 代码实现
   这里使用访问者模式 分开一个检查relatePriorityNode的visitor，还有一个removeNode的visitor，如果以后扩展其他操作方式直接增加新的visitor即可。

PriorityNode:

/**
 * 权限树节点
 * @author wangmeng
 *
 */
@Data
public class PriorityNode {

    /**
     * id
     */
    private Long id;
    /**
     * 权限编号
     */
    private String code;
    /**
     * 权限URL
     */
    private String url;
    /**
     * 权限备注
     */
    private String priorityComment;
    /**
     * 权限类型
     */
    private Integer priorityType;
    /**
     * 父权限id
     */
    private Long parentId;
    /**
     * 权限的创建时间
     */
    private Date gmtCreate;
    /**
     * 权限的修改时间
     */
    private Date gmtModified;
    /**
     * 子权限节点
     */
    private List children = new ArrayList();
    
    /**
     * 接收一个权限树访问者
     * @param visitor 权限树访问者
     */
    public void accept(PriorityNodeVisitor visitor) {
        visitor.visit(this);  
    }
}

PriorityNodeVisitor:

/**
 * 权限树节点的访问者接口
 *
 * @author wangmeng
 * @blog https://www.cnblogs.com/wang-meng/
 * @create 2019-12-01 10:12
 **/
public interface PriorityNodeVisitor {

    /**
     * 访问权限树节点
     *
     * @param node 权限树节点
     */
    void visit(PriorityNode node);
}

AbstractNodeVisitor:

/**
 * @author wangmeng
 * @blog https://www.cnblogs.com/wang-meng/
 * @create 2019-12-01 10:26
 **/
public abstract class AbstractNodeVisitor implements PriorityNodeVisitor{

    private PriorityService priorityService;

    public AbstractNodeVisitor(PriorityService priorityService) {
        this.priorityService = priorityService;
    }

    @Override
    public void visit(PriorityNode node) {
        List priorityDTOList = priorityService.listChildPriorities(node.getId());
        if (CollectionUtils.isNotEmpty(priorityDTOList)) {
            for (PriorityDTO priorityDTO : priorityDTOList) {
                PriorityNode priorityNode = new PriorityNode();
                BeanUtils.copyProperties(priorityDTO, priorityNode);
                // 使用递归处理
                priorityNode.accept(this);
            }
        }

        operateNode(node);
    }

    /**
     * 操作权限树
     * @param node 树节点
     */
    abstract void operateNode(PriorityNode node);
}

PriorityNodeRelateCheckVisitor:

/**
 * 权限树节点的关联检查访问者
 *
 * @author wangmeng
 * @blog https://www.cnblogs.com/wang-meng/
 * @create 2019-12-01 10:19
 **/
public class PriorityNodeRelateCheckVisitor extends AbstractNodeVisitor{

    /**
     * 关联检查结果
     */
    private Boolean relateCheckResult = false;
    /**
     * 权限管理模块的service组件
     */
    private PriorityService priorityService;
    /**
     * 角色和权限关系管理模块的DAO组件
     */
    private RolePriorityRelationshipService rolePriorityRelationshipService;
    /**
     * 账号和权限关系管理模块的Service组件
     */
    private AccountPriorityRelationshipService accountPriorityRelationshipService;

    /**
     * 构造函数
     */
    public PriorityNodeRelateCheckVisitor(PriorityService priorityService,
                                          RolePriorityRelationshipService rolePriorityRelationshipService,
                                          AccountPriorityRelationshipService accountPriorityRelationshipService) {
        super(priorityService);
        this.priorityService = priorityService;
        this.rolePriorityRelationshipService = rolePriorityRelationshipService;
        this.accountPriorityRelationshipService = accountPriorityRelationshipService;
    }

    @Override
    void operateNode(PriorityNode node) {
        Long nodeId = node.getId();
        // 检查权限是否被任何一个角色或者是账号关联了,如果被任何一个角色或者账号关联，则relateCheckResult=true
        int roleRelatedCount = rolePriorityRelationshipService
                .selectCount(new EntityWrapper().eq("priority_id", nodeId));
        if(roleRelatedCount > 0) {
            this.relateCheckResult = true;
        }
        int accountRelatedCount = accountPriorityRelationshipService
                .selectCount(new EntityWrapper().eq("priority_id", nodeId));
        if(accountRelatedCount > 0) {
            this.relateCheckResult = true;
        }

        this.relateCheckResult = false;
    }

    public Boolean getRelateCheckResult() {
        return relateCheckResult;
    }
}

PriorityNodeRemoveVisitor:

/**
 * 权限树节点的删除访问者
 *
 * @author wangmeng
 * @blog https://www.cnblogs.com/wang-meng/
 * @create 2019-12-01 10:13
 **/
public class PriorityNodeRemoveVisitor extends AbstractNodeVisitor{

    private PriorityService priorityService;

    /**
     * 构造函数
     * @param priorityService 权限service
     */
    public PriorityNodeRemoveVisitor(PriorityService priorityService) {
        super(priorityService);
        this.priorityService = priorityService;
    }

    @Override
    void operateNode(PriorityNode node) {
        // 删除权限
        priorityService.deleteById(node.getId());
    }
}

调用地方 PriorityServiceImpl:

@Override
    public Boolean removePriority(Long id) {
        try {
            // 根据id查询权限
            Priority priorityDO = baseMapper.selectById(id);
            PriorityNode priorityNode = priorityDO.clone(PriorityNode.class);

            // 检查这个权限以及其下任何一个子权限，是否被角色或者账号给关联着
            PriorityNodeRelateCheckVisitor relateCheckVisitor = new PriorityNodeRelateCheckVisitor(
                    this, rolePriorityRelationshipService, accountPriorityRelationshipService);
            relateCheckVisitor.visit(priorityNode);
            Boolean relateCheckResult = relateCheckVisitor.getRelateCheckResult();

            if(relateCheckResult) {
                return false;
            }

            // 递归删除当前权限以及其下所有的子权限
            PriorityNodeRemoveVisitor removeVisitor = new PriorityNodeRemoveVisitor(this);
            removeVisitor.visit(priorityNode);
        } catch (Exception e) {
             log.error("error", e);
            return false;
        }

        return true;
    }

申明：
本文章首发自本人博客：https://www.cnblogs.com/wang-meng 和公众号：壹枝花算不算浪漫 如若转载请标明来源

申明

本文章首发自本人博客：https://www.cnblogs.com/wang-meng 和公众号：壹枝花算不算浪漫，如若转载请标明来源!

感兴趣的小伙伴可关注个人公众号：壹枝花算不算浪漫