ContextPermissionManager.java 1.68 KB
package com.server.shiro.context;

import com.server.shiro.constants.Constants;
import com.server.shiro.persistent.bean.SysUser;
import com.server.shiro.persistent.repository.UserRepository;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.context.WebApplicationContext;

import java.util.List;

/**
 * Created by yinbinhome@163.com on 2018/2/23.
 * description:
 */

//@Component("permissionManager")
//@Singleton
public class ContextPermissionManager implements PermissionManage {
    private static final Logger logger = LoggerFactory.getLogger(ContextPermissionManager.class);


    @Value("${spring.application.name}")
    private String APPLICATION_NAME;

    @Autowired
    private UserRepository userRepository;

    @Autowired
    private WebApplicationContext webApplicationContext;

    @Override
    public void initPermission() {
        List<SysUser> user = (List<SysUser>) userRepository.findAll();
        logger.info("从数据库拉取<用户-权限>对象,存储到上下文里");
        webApplicationContext.getServletContext().setAttribute(APPLICATION_NAME+"_"+Constants.USER_PERMISSION,user);
    }

    @Override
    public void reloadPermission() {
        initPermission();
        logger.info("更新上下文里的<用户-权限>");
    }

    @Override
    public Object pullPermission() {
        Object users_permission=webApplicationContext.getServletContext().getAttribute(APPLICATION_NAME+"_"+Constants.USER_PERMISSION);
        logger.info("从上下文里取<用户-权限>对象");
        return users_permission;
    }

}