group.rober.base.adapter.SpringRedisCache Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of rober-base Show documentation

基础功能，如菜单，数据字典，参数对象，错误处理

The newest version!

package group.rober.base.adapter;


import org.springframework.core.convert.ConversionService;
import org.springframework.dao.DataAccessException;
import org.springframework.data.redis.cache.RedisCache;
import org.springframework.data.redis.cache.RedisCacheConfiguration;
import org.springframework.data.redis.cache.RedisCacheWriter;
import org.springframework.data.redis.connection.*;
import org.springframework.data.redis.core.RedisCallback;
import org.springframework.data.redis.core.RedisOperations;
import org.springframework.data.redis.serializer.StringRedisSerializer;
import org.springframework.util.Assert;

import java.nio.ByteBuffer;
import java.util.*;
import java.util.stream.Collectors;

public class SpringRedisCache extends RedisCache implements ShiroCacheParitalInterface {

    private final RedisCacheWriterExt redisCacheWriterExt;
    public SpringRedisCache(String name, RedisCacheWriterExt cacheWriter, RedisCacheConfiguration cacheConfig) {
        super(name, cacheWriter, cacheConfig);
        this.redisCacheWriterExt = cacheWriter;
    }

    @Override
    public RedisCacheWriterExt getNativeCache() {
        return this.redisCacheWriterExt;
    }


    @Override
    public int size() {
        byte[] pattern = getConversionService().convert(createCacheKey("*"), byte[].class);
        return this.redisCacheWriterExt.size(getName(), pattern);
    }

    @Override
    public Set keys() {
        byte[] pattern = getConversionService().convert(createCacheKey("*"), byte[].class);
        return this.redisCacheWriterExt.keys(getName(), pattern)
                .stream()
                .map(bytes -> deserializeCacheKey(bytes))
                .collect(Collectors.toSet());
    }

    @Override
    public Collection values() {
        byte[] pattern = getConversionService().convert(createCacheKey("*"), byte[].class);
        return this.redisCacheWriterExt.values(getName(), pattern)
                .stream()
                .map(bytes -> fromStoreValue(deserializeCacheValue(bytes)))
                .collect(Collectors.toList());
    }

    private String deserializeCacheKey(byte[] bytes) {
        return getCacheConfiguration().getKeySerializationPair().read(ByteBuffer.wrap(bytes));
    }

    private ConversionService getConversionService() {
        return getCacheConfiguration().getConversionService();
    }
}