tech.hdis.framework.security.manager.SessionManager Maven / Gradle / Ivy
package tech.hdis.framework.security.manager;
import tech.hdis.framework.security.session.entity.Session;
import tech.hdis.framework.security.session.interfaces.SessionService;
import org.springframework.stereotype.Component;
import javax.annotation.Resource;
import java.util.Arrays;
import java.util.HashSet;
import java.util.Set;
/**
* Session管理，用于开发者调用。
*
* @author 黄志文
*/
@Component
public class SessionManager {
private static SessionService sessionService;
@Resource
public void setSessionService(SessionService sessionService) {
SessionManager.sessionService = sessionService;
}
/**
* 新增或修改当前线程的session
*
* @param userId 用户ID
* @param role 角色
*/
public static void saveOrUpdateSession(String userId, String role) {
saveOrUpdateSession(userId, new HashSet<>(Arrays.asList(role)), null);
}
/**
* 新增或修改当前线程的session
*
* @param userId 用户ID
* @param roles 角色
* @param permissions 权限
*/
public static void saveOrUpdateSession(String userId, Set roles, Set permissions) {
Session session = getSession();
if (session == null) {
session = new Session();
}
session.setUserId(userId);
session.setRoles(roles);
session.setPermissions(permissions);
sessionService.saveOrUpdateSession(session);
}
/**
* 得到session
*
* @return Session
*/
public static Session getSession() {
return sessionService.getSession();
}
/**
* 得到用户ID
*
* @return 用户ID
*/
public static String getUserId() {
return sessionService.getSession().getUserId();
}
/**
* 登出，删除session
*/
public static void logout() {
sessionService.logout();
}
}