package com.tianbo.warehouse.bean;

import org.springframework.cache.CacheManager;
import org.springframework.cache.annotation.CachingConfigurerSupport;
import org.springframework.cache.annotation.EnableCaching;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.data.redis.cache.RedisCacheConfiguration;
import org.springframework.data.redis.cache.RedisCacheManager;
import org.springframework.data.redis.cache.RedisCacheWriter;
import org.springframework.data.redis.connection.RedisConnectionFactory;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.serializer.Jackson2JsonRedisSerializer;
import org.springframework.data.redis.serializer.JdkSerializationRedisSerializer;
import org.springframework.data.redis.serializer.RedisSerializationContext;
import org.springframework.data.redis.serializer.StringRedisSerializer;

import java.time.Duration;
import java.util.HashMap;
import java.util.Map;

@Configuration
@EnableCaching
public class RedisConfig extends CachingConfigurerSupport {

    /**
     * 缓存管理器
     * @param redisConnectionFactory
     * @return
     */
    @Bean
    public CacheManager cacheManager(
            @SuppressWarnings("rawtypes") RedisConnectionFactory redisConnectionFactory) {
        return new RedisCacheManager(

                RedisCacheWriter.nonLockingRedisCacheWriter(redisConnectionFactory),

                this.getRedisCacheConfigurationWithTtl(246060), // 默认缓存时间策略

                this.getRedisCacheConfigurationMap()); // 指定缓存时间策略

    }
    /**
     * redis模板操作类
     * @param factory
     * @return
     */
    @Bean
    public RedisTemplate redisTemplate(RedisConnectionFactory factory) {
        final StringRedisTemplate template = new StringRedisTemplate();
        template.setConnectionFactory(factory);
        template.setKeySerializer(new StringRedisSerializer());
        template.setValueSerializer(new JdkSerializationRedisSerializer());
        return template;
    }

    /**
     * @Description: 设置缓存时间, 设置key与value的序列化.
     * @param seconds 缓存时间, 单位: 秒.
     * @return
     */
    private RedisCacheConfiguration getRedisCacheConfigurationWithTtl(Integer seconds) {

        return RedisCacheConfiguration.defaultCacheConfig()

                .serializeKeysWith(RedisSerializationContext.SerializationPair

                        .fromSerializer(new Jackson2JsonRedisSerializer<>(String.class)))

                .serializeValuesWith(RedisSerializationContext.SerializationPair

                        .fromSerializer(new Jackson2JsonRedisSerializer<>(Object.class)))

                .entryTtl(Duration.ofSeconds(seconds));

    }


    private Map<String, RedisCacheConfiguration> getRedisCacheConfigurationMap() {

        Map<String, RedisCacheConfiguration> redisCacheConfigurationMap = new HashMap<>();

        redisCacheConfigurationMap.put("short_cache", this.getRedisCacheConfigurationWithTtl(30));

        redisCacheConfigurationMap.put("long_cache", this.getRedisCacheConfigurationWithTtl(60));

        return redisCacheConfigurationMap;

    }
}