spring boot cache redis 簡單理解自定義 KeyGenerator

簡單理解自定義 KeyGenerator

一般情況我們在 spring boot 中會使用 redis 作為緩存 但我們是需要自定義 cache key 的生成方式
1.為什么不使用 spring 默認的 生成策略？
先看看源碼：

public class DefaultKeyGenerator implements KeyGenerator {
        public static final int NO_PARAM_KEY = 0;
        public static final int NULL_PARAM_KEY = 53;

        public DefaultKeyGenerator() {
        }

        public Object generate(Object target, Method method, Object... params) {
                if (params.length == 0) {
//沒有參數直接返回 0
                        return 0;
                } else {
                        if (params.length == 1) {
                                Object param = params[0];
                                if (param == null) {
//有一個參數 并且為 null 返回 53
                                        return 53;
                                }
//參數不是數組 直接返回參數
                                if (!param.getClass().isArray()) {
                                        return param;
                                }
                        }
//其它情況 返回 深度計算的 hashcode  簡單理解可以認為就是一個 hashcode
                        return Arrays.deepHashCode(params);
                }
        }
}

DefaultKeyGenerator 最終返回的 緩存鍵值當參數列表的值相同時是一樣的 這樣就會造成獲取到錯誤的緩存數據

spring 還有一個 生成策略 同樣也是有這樣的 問題

public class SimpleKeyGenerator implements KeyGenerator {

     @Override
     public Object generate(Object target, Method method, Object... params) {
            return generateKey(params);
     }

     /**
        * Generate a key based on the specified parameters.
        */
     public static Object generateKey(Object... params) {
            if (params.length == 0) {
                 return SimpleKey.EMPTY;
            }
            if (params.length == 1) {
                 Object param = params[0];
                 if (param != null && !param.getClass().isArray()) {
                        return param;
                 }
            }
            return new SimpleKey(params);
     }

}

public class SimpleKey implements Serializable {

     public static final SimpleKey EMPTY = new SimpleKey();

     private final Object[] params;
     private final int hashCode;

     /**
        * Create a new {@link SimpleKey} instance.
        * @param elements the elements of the key
        */
     public SimpleKey(Object... elements) {
            Assert.notNull(elements, "Elements must not be null");
            this.params = new Object[elements.length];
            System.arraycopy(elements, 0, this.params, 0, elements.length);
            this.hashCode = Arrays.deepHashCode(this.params);
     }

     @Override
     public boolean equals(Object obj) {
            return (this == obj || (obj instanceof SimpleKey
                        && Arrays.deepEquals(this.params, ((SimpleKey) obj).params)));
     }

     @Override
     public final int hashCode() {
            return this.hashCode;
     }

     @Override
     public String toString() {
            return getClass().getSimpleName() + " [" + StringUtils.arrayToCommaDelimitedString(this.params) + "]";
     }

}

最后 cache key 使用的是這個

return getClass().getSimpleName() + " [" + StringUtils.arrayToCommaDelimitedString(this.params) + "]";

這也會在一定的情況下出現獲取到錯誤的緩存數據

所以我們需要自己自定義 cache key 的生成方式
我一般是這樣用的 類名+方法名+參數列表的類型+參數值 再做 哈希散列 作為key

@Bean
public KeyGenerator myKeyGenerator(){
     return new KeyGenerator() {
            @Override
            public Object generate(Object target, Method method, Object... params) {
                 StringBuilder sb = new StringBuilder();
                 sb.append(target.getClass().getName());
                 sb.append(method.getName());
                 sb.append("&");
                 for (Object obj : params) {
                        if (obj != null){
                             sb.append(obj.getClass().getName());
                             sb.append("&");
                             sb.append(JSON.toJSONString(obj));
                             sb.append("&");
                        }
                 }
                 log.info("redis cache key str: "+sb.toString());

                 log.info("redis cache key sha256Hex: "+DigestUtils.sha256Hex(sb.toString()));
                 return DigestUtils.sha256Hex(sb.toString());
            }
     };
}