com.github.netty.protocol.nrpc.service.RpcDBServiceImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spring-boot-protocol Show documentation
Spring Boot Protocol
The newest version!
package com.github.netty.protocol.nrpc.service;

import com.github.netty.core.util.ExpiryLRUMap;

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

/**
 * RpcDBServiceImpl
 *
 * @author wangzihao
 */
public class RpcDBServiceImpl implements RpcDBService {
    private static final String SHARING_GROUP = "/sharing";
    private final Map> memExpiryGroupMap = new ConcurrentHashMap<>(64);

    @Override
    public boolean exist2(String key, String group) {
        return getMemExpiryMap(group).containsKey(key);
    }

    @Override
    public void put(String key, byte[] data) {
        put4(key, data, -1, SHARING_GROUP);
    }

    @Override
    public void put3(String key, byte[] data, int expireSecond) {
        put4(key, data, expireSecond, SHARING_GROUP);
    }

    @Override
    public void put4(String key, byte[] data, int expireSecond, String group) {
        getMemExpiryMap(group).put(key, data, expireSecond);
    }

    @Override
    public int count(String group) {
        Map map = memExpiryGroupMap.get(group);
        if (map == null) {
            return 0;
        }
        return map.size();
    }

    @Override
    public boolean exist(String key) {
        return exist2(key, SHARING_GROUP);
    }

    @Override
    public byte[] get(String key) {
        return get2(key, SHARING_GROUP);
    }

    @Override
    public byte[] get2(String key, String group) {
        return getMemExpiryMap(group).get(key);
    }

    @Override
    public void changeKey(String oldKey, String newKey) {
        changeKey3(oldKey, newKey, SHARING_GROUP);
    }

    @Override
    public void changeKey3(String oldKey, String newKey, String group) {
        RpcDBExpiryLRUMap memExpiryMap = getMemExpiryMap(group);
        memExpiryMap.put(newKey, memExpiryMap.remove(oldKey));
    }

    @Override
    public void remove(String key) {
        remove2(key, SHARING_GROUP);
    }

    @Override
    public void remove2(String key, String group) {
        getMemExpiryMap(group).remove(key);
    }

    @Override
    public void removeBatch(List keys) {
        removeBatch2(keys, SHARING_GROUP);
    }

    @Override
    public void setMaxSize2(Integer maxSize, String group) {
        getMemExpiryMap(group).setMaxCacheSize(maxSize);
    }

    @Override
    public void setMaxSize(Integer maxSize) {
        setMaxSize2(maxSize, SHARING_GROUP);
    }

    @Override
    public void removeBatch2(List keys, String group) {
        if (keys == null || keys.isEmpty()) {
            return;
        }

        RpcDBExpiryLRUMap map = getMemExpiryMap(group);
        if (keys instanceof RandomAccess) {
            int size = keys.size();
            for (int i = 0; i < size; i++) {
                String key = keys.get(i);
                map.remove(key);
            }
        } else {
            for (String key : keys) {
                map.remove(key);
            }
        }
    }

    private RpcDBExpiryLRUMap getMemExpiryMap(String group) {
        RpcDBExpiryLRUMap memExpiryMap = memExpiryGroupMap.get(group);
        if (memExpiryMap == null) {
            synchronized (memExpiryGroupMap) {
                memExpiryMap = memExpiryGroupMap.get(group);
                if (memExpiryMap == null) {
                    memExpiryMap = new RpcDBExpiryLRUMap<>(-1);
                    memExpiryGroupMap.put(group, memExpiryMap);
                }
            }
        }
        return memExpiryMap;
    }

    private static class RpcDBExpiryLRUMap extends ExpiryLRUMap {
        RpcDBExpiryLRUMap(long defaultExpiryTime) {
            super(defaultExpiryTime);
        }
    }
}