com.github.yiuman.citrus.system.service.MenuService Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of citrus-system Show documentation
There is a newer version: 0.15.0
package com.github.yiuman.citrus.system.service;

import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.github.yiuman.citrus.support.crud.rest.BaseTreeController;
import com.github.yiuman.citrus.support.crud.rest.CrudRestful;
import com.github.yiuman.citrus.support.crud.rest.Operations;
import com.github.yiuman.citrus.support.crud.service.BaseSimpleTreeService;
import com.github.yiuman.citrus.system.dto.ResourceDto;
import com.github.yiuman.citrus.system.entity.Resource;
import org.springframework.aop.support.AopUtils;
import org.springframework.context.ApplicationContext;
import org.springframework.http.HttpMethod;
import org.springframework.stereotype.Service;
import org.springframework.web.bind.annotation.RequestMapping;

import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.Set;

/**
 * 菜单树逻辑层
 *
 * @author yiuman
 * @date 2020/5/22
 */
@Service
public class MenuService extends BaseSimpleTreeService {

    private final ApplicationContext applicationContext;

    private final ResourceService resourceService;

    public MenuService(ApplicationContext applicationContext, ResourceService resourceService) {
        this.applicationContext = applicationContext;
        this.resourceService = resourceService;
    }

    @Override
    public List list() {
        return list(null);
    }

    @SuppressWarnings({"rawtypes", "unchecked"})
    @Override
    public List list(Wrapper wrapper) {
        if (wrapper == null) {
            wrapper = Wrappers.query();
        }
        //菜单为0
        ((QueryWrapper) wrapper).eq("type", 0);
        return super.list(wrapper);
    }

    @Override
    public Resource getRoot() {
        Resource root = super.getRoot();
        root.setResourceName("系统菜单");
        return root;
    }

    @Override
    public boolean beforeSave(Resource entity) {
        entity.setType(0);
        return true;
    }

    @SuppressWarnings({"rawtypes", "unchecked"})
    @Override
    public void afterSave(Resource entity) {
        boolean isCrudRest = false;
        boolean isAssignableFromTreeRest = false;
        //若是CrudRestful则保存操作资源，则增删改查资源
        Map crudRestfuls = applicationContext.getBeansOfType(CrudRestful.class);
        Set> entries = crudRestfuls.entrySet();
        for (Map.Entry entry : entries) {
            Class restBeanClass = (Class) AopUtils.getTargetClass(entry.getValue());
            RequestMapping annotation = restBeanClass.getAnnotation(RequestMapping.class);
            if (annotation != null && Arrays.asList(annotation.value()).contains(entity.getPath())) {
                isCrudRest = true;
                isAssignableFromTreeRest = restBeanClass.isAssignableFrom(BaseTreeController.class);
                break;
            }
        }

        if (isCrudRest) {
            resourceService.remove(Wrappers.query().eq("parent_id", entity.getId()));
            createCrudDefaultResource(entity);

            if (isAssignableFromTreeRest) {
                createTreeDefaultResource(entity);
            }
        }

    }

    /**
     * 根据菜单主键获取操作资源
     *
     * @param key 菜单主键
     * @return 操作资源列表
     */
    public List getOperationByKey(Long key) {
        return super.list(Wrappers.query().eq(getParentField(), key).eq("type", 2));
    }

    /**
     * 保存菜单默认的资源
     *
     * @param menu 当前菜单
     */
    private void createCrudDefaultResource(Resource menu) {
        final Integer resourceType = 2;
        resourceService.batchSave(Arrays.asList(
                new ResourceDto("列表", resourceType, menu.getId(), menu.getPath(), HttpMethod.GET.name()),
                new ResourceDto("保存", resourceType, menu.getId(), menu.getPath(), HttpMethod.POST.name()),
                new ResourceDto("查看", resourceType, menu.getId(), menu.getPath() + Operations.GET, HttpMethod.GET.name()),
                new ResourceDto("删除", resourceType, menu.getId(), menu.getPath() + Operations.DELETE, HttpMethod.DELETE.name()),
                new ResourceDto("批量删除", resourceType, menu.getId(), menu.getPath() + Operations.BATCH_DELETE, HttpMethod.POST.name()),
                new ResourceDto("导入", resourceType, menu.getId(), menu.getPath() + Operations.IMPORT, HttpMethod.POST.name()),
                new ResourceDto("导出", resourceType, menu.getId(), menu.getPath() + Operations.EXPORT, HttpMethod.GET.name())
        ));
    }

    /**
     * 保存树形菜单资源的默认资源
     *
     * @param menu 当前树形菜单资源
     */
    private void createTreeDefaultResource(Resource menu) {
        final Integer resourceType = 2;
        resourceService.batchSave(Arrays.asList(
                new ResourceDto("加载树", resourceType, menu.getId(), menu.getPath() + Operations.Tree.TREE, HttpMethod.GET.name()),
                new ResourceDto("加载子节点", resourceType, menu.getId(), menu.getPath() + Operations.Tree.GET_BY_PARENT, HttpMethod.GET.name()),
                new ResourceDto("移动", resourceType, menu.getId(), menu.getPath() + Operations.Tree.MOVE, HttpMethod.POST.name()),
                new ResourceDto("初始化", resourceType, menu.getId(), menu.getPath() + Operations.Tree.INIT, HttpMethod.POST.name())
        ));
    }
}