SpringBoot @Cacheable自定義KeyGenerator方式

Posted on by

@Cacheable自定義KeyGenerator

1. 概述

SpringBoot 使用 @Cacheable 可以方便的管理緩存數據，在不指定 key 屬性的情況下，默認使用 SimpleKeyGenerator 生成 key。除此之外，我們也可以自定義實現 KeyGenerator 接口，生成自己的 key 名稱策略。

2. MySimpleKey 類

MySimpleKey類的作用是存放參數數據，必須實現equals、hashCode。如果需要自定義key格式，同樣需要實現toString接口，下面的例子是把參數用逗號分隔。

public class MySimpleKey implements Serializable {
    public static final MySimpleKey EMPTY = new MySimpleKey(new Object[0]);
    private final Object[] params;
    private transient int hashCode;
    public MySimpleKey(Object... elements) {
        Assert.notNull(elements, "Elements must not be null");
        this.params = (Object[])elements.clone();
        this.hashCode = Arrays.deepHashCode(this.params);
    }
    public boolean equals(@Nullable Object other) {
        return this == other || other instanceof MySimpleKey && Arrays.deepEquals(this.params, ((MySimpleKey)other).params);
    }
    public final int hashCode() {
        return this.hashCode;
    }
    public String toString() {
        return StringUtils.arrayToCommaDelimitedString(this.params);
    }
}

3. MyKeyGenerator 類

MyKeyGenerator 實現 KeyGenerator 的接口，裡面隻有一個 generate 方法

public class MyKeyGenerator implements KeyGenerator {
    @Override
    public Object generate(Object o, Method method, Object... objects) {
        if (objects.length == 0) {
            return MySimpleKey.EMPTY;
        } else {
            if (objects.length == 1) {
                Object param = objects[0];
                if (param != null && !param.getClass().isArray()) {
                    return param;
                }
            }
            return new MySimpleKey(objects);
        }
    }
}

定義MyKeyGenerator Bean：

@Component
public class MyRedisConf {
    @Bean
    public MyKeyGenerator myKeyGenerator(){
        return new MyKeyGenerator();
    }
}

4. 配置keyGenerator

在 @Cacheable 配置 keyGenerator 屬性，值就是前面配置的Bean名稱

    @Override
    @Cacheable(value = {"REDIS:GETSTRING3"}, keyGenerator = "myKeyGenerator")
    public String getString3(String tag, String name) {
        return tag + " " + name;
    }

測試結果如下，tag、name 參數確實以逗號分隔

127.0.0.1:6379[5]> KEYS *

1) “REDIS:GETSTRING3::hello,zhangsan”

Spring-Cache key設置

第一種方式：手動設置

為瞭便於key的不重復，我們可以手動設置key有類名、方法名、參數等組合

屬性名稱

描述

示例

methodName

當前方法名

#root.methodName

method

當前方法

#root.method.name

target

當前被調用的對象

#root.target

targetClass

當前被調用的對象的class

#root.targetClass

args

當前方法參數組成的數組

#root.args[0]

caches

當前被調用的方法使用的Cache

#root.caches[0].name

 
key = "#root.targetClass.simpleName+':'+#root.methodName+':'+#param"

如下圖所示

第二種方式：自定義keyGenerator

1、自定義CacheKeyGenerator 實現KeyGenerator

public class CacheKeyGenerator implements KeyGenerator {
    /**
     * (非 Javadoc)
     * <p>
     * Title: generate
     * </p>
     * 
     * @param target
     * @param method
     * @param params
     * @return
     * @see org.springframework.cache.interceptor.KeyGenerator#generate(java.lang.Object,
     *      java.lang.reflect.Method, java.lang.Object[])
     */
    @Override
    public Object generate(Object target, Method method, Object... params) {
        StringBuilder key = new StringBuilder();
        key.append(target.getClass().getSimpleName()).append(":").append(method.getName()).append(":");
        if (params.length == 0) {
            return key.toString();
        }
        for (int i = 0; i < params.length; i++) {
            Object param = params[i];
            if (param == null || param instanceof LogableParam) {
                del(key);
            } else if (ClassUtils.isPrimitiveArray(param.getClass())) {
                int length = Array.getLength(param);
                for (int j = 0; j < length; j++) {
                    key.append(Array.get(param, j));
                    key.append(',');
                }
            } else if (ClassUtils.isPrimitiveOrWrapper(param.getClass()) || param instanceof String) {
                key.append(param);
            } else {
                key.append(param.toString());
            }
            key.append('-');
        }
        del(key);
        return key.toString();
    }
    private StringBuilder del(StringBuilder stringBuilder) {
        if (stringBuilder.toString().endsWith("-")) {
            stringBuilder.deleteCharAt(stringBuilder.length() - 1);
        }
        return stringBuilder;
    }
}

2、配置xml

<cache:annotation-driven cache-manager="cacheManager" key-generator="cacheKeyGenerator"
        proxy-target-class="true" />
<bean id="cacheKeyGenerator" class="com.tensoon.util.CacheKeyGenerator"></bean>

3、配置註解

如下圖所示

以上為個人經驗，希望能給大傢一個參考，也希望大傢多多支持WalkonNet。

推薦閱讀：