package xxx.xxx.xxx;

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

/**
 * Map 快取實現
 */
public class MapCache {

    /**
     * 預設儲存1024個快取
     */
    private static final int DEFAULT_CACHES = 1024;

    private static final MapCache INS = new MapCache();

    public static MapCache single() {
        return INS;
    }

    /**
     * 快取容器
     */
    private Map<String, CacheObject> cachePool;

    public MapCache() {
        this(DEFAULT_CACHES);
    }

    public MapCache(int cacheCount) {
        cachePool = new ConcurrentHashMap<>(cacheCount);
    }

    /**
     * 讀取一個快取
     *
     * @param key 快取key
     * @param <T>
     * @return
     */
    public <T> T get(String key) {
        CacheObject cacheObject = cachePool.get(key);
        if (null != cacheObject) {
            long cur = System.currentTimeMillis() / 1000;
            if (cacheObject.getExpired() <= 0 || cacheObject.getExpired() > cur) {
                Object result = cacheObject.getValue();
                return (T) result;
            }
        }
        return null;
    }

    /**
     * 讀取一個hash型別快取
     *
     * @param key   快取key
     * @param field 快取field
     * @param <T>
     * @return
     */
    public <T> T hget(String key, String field) {
        key = key + ":" + field;
        return this.get(key);
    }

    /**
     * 設定一個快取
     *
     * @param key   快取key
     * @param value 快取value
     */
    public void set(String key, Object value) {
        this.set(key, value, -1);
    }

    /**
     * 設定一個快取並帶過期時間
     *
     * @param key     快取key
     * @param value   快取value
     * @param expired 過期時間，單位為秒
     */
    public void set(String key, Object value, long expired) {
        expired = expired > 0 ? System.currentTimeMillis() / 1000 + expired : expired;
        CacheObject cacheObject = new CacheObject(key, value, expired);
        cachePool.put(key, cacheObject);
    }

    /**
     * 設定一個hash快取
     *
     * @param key   快取key
     * @param field 快取field
     * @param value 快取value
     */
    public void hset(String key, String field, Object value) {
        this.hset(key, field, value, -1);
    }

    /**
     * 設定一個hash快取並帶過期時間
     *
     * @param key     快取key
     * @param field   快取field
     * @param value   快取value
     * @param expired 過期時間，單位為秒
     */
    public void hset(String key, String field, Object value, long expired) {
        key = key + ":" + field;
        expired = expired > 0 ? System.currentTimeMillis() / 1000 + expired : expired;
        CacheObject cacheObject = new CacheObject(key, value, expired);
        cachePool.put(key, cacheObject);
    }

    /**
     * 根據key刪除快取
     *
     * @param key 快取key
     */
    public void del(String key) {
        cachePool.remove(key);
    }

    /**
     * 根據key和field刪除快取
     *
     * @param key   快取key
     * @param field 快取field
     */
    public void hdel(String key, String field) {
        key = key + ":" + field;
        this.del(key);
    }

    /**
     * 清空快取
     */
    public void clean() {
        cachePool.clear();
    }

    static class CacheObject {
        private String key;
        private Object value;
        private long expired;

        public CacheObject(String key, Object value, long expired) {
            this.key = key;
            this.value = value;
            this.expired = expired;
        }

        public String getKey() {
            return key;
        }

        public Object getValue() {
            return value;
        }

        public long getExpired() {
            return expired;
        }
    }
}