PlayEdu/src/main/java/xyz/playedu/api/PlayEduBackendThreadLocal.java

package xyz.playedu.api;

import xyz.playedu.api.domain.AdminUser;

import java.util.LinkedHashMap;

public class PlayEduBackendThreadLocal {

    private static final java.lang.ThreadLocal<LinkedHashMap<String, Object>> THREAD_LOCAL = new java.lang.ThreadLocal<>();

    public PlayEduBackendThreadLocal() {
    }

    public static void put(String key, Object val) {
        LinkedHashMap<String, Object> hashMap = THREAD_LOCAL.get();
        if (hashMap == null) {
            hashMap = new LinkedHashMap<>();
        }
        hashMap.put(key, val);
        THREAD_LOCAL.set(hashMap);
    }

    public static Object get(String key) {
        return THREAD_LOCAL.get().getOrDefault(key, null);
    }

    public static Integer getAdminUserID() {
        return (Integer) get("admin_user_id");
    }

    public static void setAdminUserId(Integer userId) {
        put("admin_user_id", userId);
    }

    public static AdminUser getAdminUser() {
        return (AdminUser) get("admin_user");
    }

    public static void setAdminUser(AdminUser adminUser) {
        put("admin_user", adminUser);
    }

    public static void remove() {
        THREAD_LOCAL.remove();
    }

}