Localstorage.java

package com.subsidy.util;

import org.springframework.web.bind.annotation.RestController;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

@RestController
public class Localstorage {
    private static final ThreadLocal<Object> local = ThreadLocal.withInitial(() -> null);
    private static ConcurrentHashMap<String, Object> map = new ConcurrentHashMap<>();
    public static void setUser(Object obj, String type) {
//        Map<String, Object> map = new HashMap<>();
        map.put("user", obj);
        map.put("type", type);
        Localstorage.local.set(map);
    }

    public static Map<String, Object> getMap() {
        return (Map<String, Object>) Localstorage.local.get();
    }

    public static Object getUser() {
//        Map<String, Object> map = (Map<String, Object>) Localstorage.local.get();
        return map.get("user");
    }

    public static void remove() {
        Localstorage.local.remove();
    }
}