update basic

whycxzp

2021-01-14 9a053c132a1c9b7675a1694dc4fc10a69eeda3fd

 src/main/java/com/whyc/service/UserBridgeService.java

@@ -2,14 +2,14 @@

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.whyc.mapper.PermissionMapper;
import com.whyc.mapper.RoleMapper;
import com.whyc.mapper.UserMapper;
import com.whyc.mapper.*;
import com.whyc.pojo.Permission;
import com.whyc.pojo.User;
import org.apache.shiro.authz.AuthorizationInfo;
import org.apache.shiro.authz.SimpleAuthorizationInfo;
import org.springframework.cache.annotation.CacheConfig;
import org.springframework.cache.annotation.Cacheable;
import org.springframework.context.annotation.Lazy;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
@@ -20,20 +20,22 @@
 * 处理 shiro认证,授权,和数据库交互
 */
@Service
//Unified Naming
@CacheConfig(cacheNames ={"userBridge"})
public class UserBridgeService {

    @Resource
    private UserMapper userMapper;

    @Resource
    private RoleMapper roleMapper;
    private UserRoleMapper userRoleMapper;

    @Resource
    private PermissionMapper permissionMapper;
    private RolePermissionMapper rolePermissionMapper;

    public User findPasswordAndSlatByUserName(String userName) {
        QueryWrapper<User> queryWrapper = Wrappers.query();
        queryWrapper.select("id","password","salt").eq("name",userName);
        queryWrapper.select("id","name","password","salt").eq("name",userName);
        try{
            return userMapper.selectOne(queryWrapper);
        }catch (Exception e){
@@ -42,27 +44,28 @@
        }
    }

    @Cacheable(cacheNames = "authorizationCache",key = "#root.method")
    @Cacheable(key = "#root.methodName+#p0.id")
    public AuthorizationInfo getAuthorizationInfo(User user) {
        System.out.println("=========执行了UserBridgeService.getAuthorization方法==========");
        SimpleAuthorizationInfo authorizationInfo = new SimpleAuthorizationInfo();
        //添加Roles和Permissions
        List<String> roles = this.findRolesByUserId(user.getId());
        List<String> perms = this.findPermissionsByUserId(user.getId());
        List<String> roles = findRolesByUserId(user.getId());
        List<String> perms = findPermissionsByUserId(user.getId());

        authorizationInfo.addRoles(roles);
        authorizationInfo.addStringPermissions(perms);
        return authorizationInfo;
    }

    //@Cacheable(key = "#root.methodName+#userId")
    private List<String> findPermissionsByUserId(int userId) {
        QueryWrapper<Permission> query = Wrappers.query();
        List<String> perms = new LinkedList<>();
        //perms.add("water:all");
        List<String> perms = rolePermissionMapper.findPermissionsByUserId(userId);
        return perms;
    }

    //@Cacheable(key="#root.methodName+#userId")
    private List<String> findRolesByUserId(int userId) {
        List<String> roles = new LinkedList<>();
        List<String> roles =userRoleMapper.findRolesByUserId(userId);
        //roles.add("dev");
        return roles;
    }