forked from moral

blame | history | patch | commit | commitdiff | ignore whitespace
kaiyu
2021-08-31 e15ddfa800ddf87becbd8ea3e462ebd59fbd82b1 
 screen-manage/src/main/java/com/moral/api/service/impl/ManageRoleServiceImpl.java


@@ -4,14 +4,17 @@


import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;


import com.moral.api.entity.*;


import com.moral.api.mapper.*;


import com.moral.api.pojo.redisBean.AccountInfoDTO;


import com.moral.api.service.ManageRoleService;


import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;


import com.moral.api.util.LogUtils;


import com.moral.constant.Constants;


import com.moral.constant.ResponseCodeEnum;


import com.moral.util.TokenUtils;


import org.springframework.beans.factory.annotation.Autowired;


import org.springframework.stereotype.Service;


import org.springframework.transaction.annotation.Transactional;


import org.springframework.util.ObjectUtils;


import org.springframework.util.StringUtils;


import org.springframework.web.context.request.RequestContextHolder;


import org.springframework.web.context.request.ServletRequestAttributes;


@@ -20,6 +23,7 @@


import javax.servlet.http.HttpServletRequest;


import java.text.SimpleDateFormat;


import java.util.*;


import java.util.function.Predicate;




/**


 * <p>


@@ -77,6 +81,19 @@


            manageRoleMap.put("updateTime",updateTime);


            manageRoleMap.put("key",manageRole.getId());


            manageRolesList.add(manageRoleMap);


        }


        //除admin账号外  admin角色不可见


        AccountInfoDTO accountInfo = (AccountInfoDTO) TokenUtils.getUserInfo();


        List<ManageRole> roles = accountInfo.getRoles();


        if(!roles.get(0).getName().equals("admin")){


            manageRolesList.removeIf(new Predicate<Map<String, Object>>() {


                @Override


                public boolean test(Map<String, Object> map) {


                    if(map.get("name").equals("admin"))


                        return true;


                    return false;


                }


            });


        }


        resultMap.put("manageRoles",manageRolesList);


        int totalNumber = manageRoleMapper.selectCount(wrapper);


@@ -237,6 +254,21 @@


        return resultMap;


    }




    @Override


    public ManageRole getRoleByAccountId(Integer accountId) {


        QueryWrapper<ManageAccountRole> queryAccountRoleWrapper = new QueryWrapper<>();


        queryAccountRoleWrapper.eq("is_delete",Constants.NOT_DELETE);


        queryAccountRoleWrapper.eq("account_id",accountId);


        ManageAccountRole accountRole = manageAccountRoleMapper.selectOne(queryAccountRoleWrapper);


        if(ObjectUtils.isEmpty(accountRole))


            return null;


        Integer roleId = accountRole.getRoleId();


        QueryWrapper<ManageRole> queryRoleWrapper = new QueryWrapper();


        queryRoleWrapper.eq("is_delete", Constants.NOT_DELETE);


        queryRoleWrapper.eq("id", roleId);


        return  manageRoleMapper.selectOne(queryRoleWrapper);


    }




    /**


     * @Description: 补充父菜单


     * @Param: [list, menuId]list：menuId集合