AbstractCacheComponent.java 4.45 KB
package com.yoho.search.cache.beans;

import com.alibaba.fastjson.JSON;
import com.yoho.core.redis.cluster.operations.serializer.RedisKeyBuilder;
import com.yoho.search.cache.impls.EhCache;
import com.yoho.search.cache.impls.SearchRedis;
import com.yoho.search.cache.model.CacheObject;
import com.yoho.search.service.recall.models.common.ParamQueryFilter;
import org.springframework.beans.factory.annotation.Autowired;

import java.lang.reflect.ParameterizedType;
import java.lang.reflect.Type;
import java.util.Map;

public abstract class AbstractCacheComponent<T> {

    @Autowired
    private EhCache ehCache;
    @Autowired
    private SearchRedis searchRedis;

    public T queryWithCache(ParamQueryFilter paramQueryFilter, Map<String, String> paramMap) throws Exception {
        //1、生成RedisKeyBuilder
        RedisKeyBuilder redisKeyBuilder = this.genRedisKeyBuilder(paramQueryFilter,paramMap);
        if (redisKeyBuilder == null) {
            return null;
        }
        //2、取本地缓存,缓存命中,则直接返回
        T result = this.getValueFromEhcache(redisKeyBuilder);
        if (result != null) {
            return result;
        }
        //3、取redis缓存,缓存命中,则回写ehcahce
        result = this.getValueFromRedis(redisKeyBuilder);
        if (result != null) {
            this.addValueToEhcache(redisKeyBuilder, result);
            return result;
        }
        return this.doInnerQuery(redisKeyBuilder, paramQueryFilter, paramMap);
    }

    private T doInnerQuery(RedisKeyBuilder redisKeyBuilder, ParamQueryFilter paramQueryFilter, Map<String, String> paramMap) throws Exception {
        T result = this.doRealQuery(paramQueryFilter, paramMap);
        if (result == null) {
            return result;
        }
        this.addValueToEhcache(redisKeyBuilder, result);
        this.addValueToRedis(redisKeyBuilder, result);
        return result;
    }

    protected abstract T doRealQuery(ParamQueryFilter paramQueryFilter, Map<String, String> paramMap) throws Exception;

    private T getValueFromEhcache(RedisKeyBuilder redisKeyBuilder) throws Exception {
        if (!useEhcache()) {
            return null;
        }
        CacheObject cacheObject = ehCache.get(redisKeyBuilder);
        if (cacheObject != null) {
            Object object = cacheObject.toObject();
            return (T) object;
        }
        return null;
    }

    private void addValueToEhcache(RedisKeyBuilder redisKeyBuilder, T result) {
        if(!useEhcache()){
            return;
        }
        ehCache.addOrUpdate(redisKeyBuilder, new CacheObject(result), this.cacheTimeInMinute());
    }

    private void addValueToRedis(RedisKeyBuilder redisKeyBuilder, T result) {
        if(!useRedis()){
            return;
        }
        String jsonString = JSON.toJSONString(result);
        CacheObject toCacheResult = new CacheObject(jsonString);
        searchRedis.addOrUpdate(redisKeyBuilder, toCacheResult, this.cacheTimeInMinute());
    }

    private T getValueFromRedis(RedisKeyBuilder redisKeyBuilder) {
        if (!useRedis()) {
            return null;
        }
        CacheObject cacheObject = searchRedis.get(redisKeyBuilder);
        if (cacheObject == null || cacheObject.getValue()==null) {
            return null;
        }
        if(! (cacheObject.getValue() instanceof  String)){
            return null;
        }
        String redisValue = (String) cacheObject.toObject();
        Type superClass = getClass().getGenericSuperclass();
        Type type = ((ParameterizedType) superClass).getActualTypeArguments()[0];
        return JSON.parseObject(redisValue, type);
    }

    protected abstract int cacheTimeInMinute();

    protected abstract String cacheSceneKey();

    protected RedisKeyBuilder genRedisKeyBuilder(ParamQueryFilter paramQueryFilter,Map<String,String> paramMap){
        RedisKeyBuilder redisKeyBuilder  = RedisKeyBuilder.newInstance();
        redisKeyBuilder.appendFixed("YOHOSEARCH").appendFixed(":");//前缀
        redisKeyBuilder.appendFixed(cacheSceneKey()).appendFixed(":");//场景
        redisKeyBuilder.appendFixed(cacheTimeInMinute()).appendFixed(":");//缓存时间
        redisKeyBuilder.appendVar(paramQueryFilter.getParamMd5Key());//报文
        return redisKeyBuilder;
    }

    protected boolean useEhcache() {
        return false;
    }

    protected boolean useRedis() {
        return true;
    }

}