kcim-cost/src/main/java/com/imed/costaccount/common/util/UserContext.java

package com.imed.costaccount.common.util;

import com.imed.costaccount.model.User;
import org.apache.shiro.SecurityUtils;

/**
 * @author 李加喜
 * @Package com.imed.costaccount.common.util
 * @date 2021-08-03 8:46
 */
public class UserContext {
    private static final ThreadLocal<User> current = new ThreadLocal();

    public UserContext() {
    }

    public static User getCurrentUser() {
        return (User) SecurityUtils.getSubject().getPrincipal();
    }

    public static void setCurrentUser(User user) {
        current.set(user);
    }

    public static void removeUser() {
        current.remove();
    }

    public static Long getHospId(){
        User principal = (User) SecurityUtils.getSubject().getPrincipal();
        return principal.getHospId();
    }
}