簡(jiǎn)直騷操作，ThreadLocal還能當(dāng)緩存用

時(shí)間：2020-08-13 11:07:45

關(guān)鍵字： Local Thread

手機(jī)看文章

掃描二維碼
隨時(shí)隨地手機(jī)看文章

[導(dǎo)讀]背景說(shuō)明有朋友問(wèn)我一個(gè)關(guān)于接口優(yōu)化的問(wèn)題，他的優(yōu)化點(diǎn)很清晰，由于接口中調(diào)用了內(nèi)部很多的 service 去組成了一個(gè)完成的業(yè)務(wù)功能。每個(gè) service 中的邏輯都是獨(dú)立的，這樣就導(dǎo)致了很多查詢(xún)是重復(fù)的，看下圖你就明白了。上層查詢(xún)傳遞下去對(duì)于這種場(chǎng)景最好

背景說(shuō)明

有朋友問(wèn)我一個(gè)關(guān)于接口優(yōu)化的問(wèn)題，他的優(yōu)化點(diǎn)很清晰，由于接口中調(diào)用了內(nèi)部很多的 service 去組成了一個(gè)完成的業(yè)務(wù)功能。每個(gè) service 中的邏輯都是獨(dú)立的，這樣就導(dǎo)致了很多查詢(xún)是重復(fù)的，看下圖你就明白了。

上層查詢(xún)傳遞下去

對(duì)于這種場(chǎng)景最好的就是在上層將需要的數(shù)據(jù)查詢(xún)出來(lái)，然后傳遞到下層去消費(fèi)。這樣就不用重復(fù)查詢(xún)了。

如果開(kāi)始寫(xiě)代碼的時(shí)候是這樣做的沒(méi)問(wèn)題，但很多時(shí)候，之前寫(xiě)的時(shí)候都是獨(dú)立的，或者復(fù)用的老邏輯，里面就是有獨(dú)立的查詢(xún)。

如果要做優(yōu)化就只能將老的方法重載一個(gè)，將需要的信息直接傳遞過(guò)去。

public void xxx(int goodsId) {
    Goods goods = goodsService.get(goodsId);
    .....
}
public void xxx(Goods goods) {
    .....
}

加緩存

如果你的業(yè)務(wù)場(chǎng)景允許數(shù)據(jù)有一定延遲，那么重復(fù)調(diào)用你可以直接通過(guò)加緩存來(lái)解決。這樣的好處在于不會(huì)重復(fù)查詢(xún)數(shù)據(jù)庫(kù)，而是直接從緩存中取數(shù)據(jù)。

更大的好處在于對(duì)于優(yōu)化類(lèi)的影響最小，原有的代碼邏輯都不用改變，只需要在查詢(xún)的方法上加注解進(jìn)行緩存即可。

public void xxx(int goodsId) {
    Goods goods = goodsService.get(goodsId);
    .....
}
public void xxx(Goods goods) {
    Goods goods = goodsService.get(goodsId);
    .....
}
class GoodsService {
    @Cached(expire = 10, timeUnit = TimeUnit.SECONDS)
    public Goods get(int goodsId) {
        return dao.findById(goodsId);
    }
}

如果你的業(yè)務(wù)場(chǎng)景不允許有緩存的話，上面這個(gè)方法就不能用了。那么是不是還得改代碼，將需要的信息一層層往下傳遞呢？

自定義線程內(nèi)的緩存

我們總結(jié)下目前的問(wèn)題：

同一次請(qǐng)求內(nèi)，多次相同的查詢(xún)獲取 RPC 等的調(diào)用。
數(shù)據(jù)實(shí)時(shí)性要求高，不適合加緩存，主要是加緩存也不好設(shè)置過(guò)期時(shí)間，除非采用數(shù)據(jù)變更主動(dòng)更新緩存的方式。
只需要在這一次請(qǐng)求里緩存即可，不影響其他地方。
不想改動(dòng)已有代碼。

總結(jié)后發(fā)現(xiàn)這個(gè)場(chǎng)景適合用 Thread Local 來(lái)傳遞數(shù)據(jù)，對(duì)已有代碼改動(dòng)量最小，而且也只對(duì)當(dāng)前線程生效，不會(huì)影響其他線程。

public void xxx(int goodsId) {
    Goods goods = ThreadLocal.get();
    if (goods == null) {
        goods = goodsService.get(goodsId);
    }
    .....
}

上面代碼就是使用了 ThreadLocal 來(lái)獲取數(shù)據(jù)，如果有的話就直接使用，不用去重新查詢(xún)，沒(méi)有的話就去查詢(xún)，不影響老邏輯。

雖然能實(shí)現(xiàn)效果，但是不太好，不夠優(yōu)雅。也不夠通用，如果一次請(qǐng)求內(nèi)要緩存多種類(lèi)型的數(shù)據(jù)怎么處理? ThreadLocal 就不能存儲(chǔ)固定的類(lèi)型。還有就是老的邏輯還是得改，加了個(gè)判斷。

下面介紹一種比較優(yōu)雅的方式：

自定義緩存注解，加在查詢(xún)的方法上。
定義切面切到加了緩存注解的方法上，第一次獲取返回值存入 ThreadLocal。第二次直接從 ThreadLocal 中取值返回。
ThreadLocal 中存儲(chǔ) Map，Key 為某方法的某一標(biāo)識(shí)，這樣可以緩存多種類(lèi)型的結(jié)果。
在 Filter 中將 ThreadLocal 進(jìn)行 remove 操作，因?yàn)榫€程是復(fù)用的，使用完需要清空。

注意：ThreadLocal 不能跨線程，如果有跨線程需求，請(qǐng)使用阿里的 ttl 來(lái)裝飾。

注解定義

@Target({ ElementType.METHOD })
@Retention(RetentionPolicy.RUNTIME)
public @interface ThreadLocalCache {
    /**
     * 緩存key，支持SPEL表達(dá)式
     * @return
     */
    String key() default "";
}

存儲(chǔ)定義

/**
 * 線程內(nèi)緩存管理
 *
 * @作者 尹吉?dú)g
 * @時(shí)間 2020-07-12 10:47
 */
public class ThreadLocalCacheManager {
    private static ThreadLocal<Map> threadLocalCache = new ThreadLocal<>();
    public static void setCache(Map value) {
        threadLocalCache.set(value);
    }
    public static Map getCache() {
        return threadLocalCache.get();
    }
    public static void removeCache() {
        threadLocalCache.remove();
    }
    public static void removeCache(String key) {
        Map cache = threadLocalCache.get();
        if (cache != null) {
            cache.remove(key);
        }
    }
}

切面定義

/**
 * 線程內(nèi)緩存
 *
 * @作者 尹吉?dú)g
 * @時(shí)間 2020-07-12 10:48
 */
@Aspect
public class ThreadLocalCacheAspect {
    @Around(value = "@annotation(localCache)")
    public Object aroundAdvice(ProceedingJoinPoint joinpoint, ThreadLocalCache localCache) throws Throwable {
        Object[] args = joinpoint.getArgs();
        Method method = ((MethodSignature) joinpoint.getSignature()).getMethod();
        String className = joinpoint.getTarget().getClass().getName();
        String methodName = method.getName();
        String key = parseKey(localCache.key(), method, args, getDefaultKey(className, methodName, args));
        Map cache = ThreadLocalCacheManager.getCache();
        if (cache == null) {
            cache = new HashMap();
        }
        Map finalCache = cache;
        Map<String, Object> data = new HashMap<>();
        data.put("methodName", className + "." + methodName);
        Object cacheResult =  CatTransactionManager.newTransaction(() -> {
            if (finalCache.containsKey(key)) {
                return finalCache.get(key);
            }
            return null;
        }, "ThreadLocalCache", "CacheGet", data);
        if (cacheResult != null) {
            return cacheResult;
        }
        return CatTransactionManager.newTransaction(() -> {
            Object result = null;
            try {
                result = joinpoint.proceed();
            } catch (Throwable throwable) {
                throw new RuntimeException(throwable);
            }
            finalCache.put(key, result);
            ThreadLocalCacheManager.setCache(finalCache);
            return result;
        }, "ThreadLocalCache", "CachePut", data);
    }
    private String getDefaultKey(String className, String methodName, Object[] args) {
        String defaultKey = className + "." + methodName;
        if (args != null) {
            defaultKey = defaultKey + "." + JsonUtils.toJson(args);
        }
        return defaultKey;
    }
    private String parseKey(String key, Method method, Object[] args, String defaultKey){
        if (!StringUtils.hasText(key)) {
            return defaultKey;
        }
        LocalVariableTableParameterNameDiscoverer nameDiscoverer = new LocalVariableTableParameterNameDiscoverer();
        String[] paraNameArr = nameDiscoverer.getParameterNames(method);
        ExpressionParser parser = new SpelExpressionParser();
        StandardEvaluationContext context = new StandardEvaluationContext();
        for(int i = 0;i < paraNameArr.length; i++){
            context.setVariable(paraNameArr[i], args[i]);
        }
        try {
            return parser.parseExpression(key).getValue(context, String.class);
        } catch (SpelEvaluationException e) {
            // 解析不出SPEL默認(rèn)為類(lèi)名+方法名+參數(shù)
            return defaultKey;
        }
    }
}

過(guò)濾器定義

/**
 * 線程緩存過(guò)濾器
 *
 * @作者 尹吉?dú)g
 * @個(gè)人微信 jihuan900
 * @微信公眾號(hào) 猿天地
 * @GitHub https://github.com/yinjihuan
 * @作者介紹 http://cxytiandi.com/about
 * @時(shí)間 2020-07-12 19:46
 */
@Slf4j
public class ThreadLocalCacheFilter implements Filter {
    @Override
    public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain filterChain) throws IOException, ServletException {
        filterChain.doFilter(servletRequest, servletResponse);
        // 執(zhí)行完后清除緩存
        ThreadLocalCacheManager.removeCache();
    }
}

自動(dòng)配置類(lèi)

@Configuration
public class ThreadLocalCacheAutoConfiguration {
    @Bean
    public FilterRegistrationBean idempotentParamtFilter() {
        FilterRegistrationBean registration = new FilterRegistrationBean();
        ThreadLocalCacheFilter filter = new ThreadLocalCacheFilter();
        registration.setFilter(filter);
        registration.addUrlPatterns("/*");
        registration.setName("thread-local-cache-filter");
        registration.setOrder(1);
        return registration;
    }
    @Bean
    public ThreadLocalCacheAspect threadLocalCacheAspect() {
        return new ThreadLocalCacheAspect();
    }
}

使用案例

@Service
public class TestService {
    /**
     * ThreadLocalCache 會(huì)緩存，只對(duì)當(dāng)前線程有效
     * @return
     */
    @ThreadLocalCache
    public String getName() {
        System.out.println("開(kāi)始查詢(xún)了");
        return "yinjihaun";
    }
    /**
     * 支持SPEL表達(dá)式
     * @param id
     * @return
     */
    @ThreadLocalCache(key = "#id")
    public String getName(String id) {
        System.out.println("開(kāi)始查詢(xún)了");
        return "yinjihaun" + id;
    }
}

功能代碼：https://github.com/yinjihuan/kitty^[1]

案例代碼：https://github.com/yinjihuan/kitty-samples^[2]

關(guān)于作者 ：尹吉?dú)g，簡(jiǎn)單的技術(shù)愛(ài)好者，《Spring Cloud 微服務(wù)-全棧技術(shù)與案例解析》, 《Spring Cloud 微服務(wù) 入門(mén) 實(shí)戰(zhàn)與進(jìn)階》作者。

特別推薦一個(gè)分享架構(gòu)+算法的優(yōu)質(zhì)內(nèi)容，還沒(méi)關(guān)注的小伙伴，可以長(zhǎng)按關(guān)注一下：

簡(jiǎn)直騷操作，ThreadLocal還能當(dāng)緩存用

長(zhǎng)按訂閱更多精彩▼
如有收獲，點(diǎn)個(gè)在看，誠(chéng)摯感謝

免責(zé)聲明：本文內(nèi)容由21ic獲得授權(quán)后發(fā)布，版權(quán)歸原作者所有，本平臺(tái)僅提供信息存儲(chǔ)服務(wù)。文章僅代表作者個(gè)人觀點(diǎn)，不代表本平臺(tái)立場(chǎng)，如有問(wèn)題，請(qǐng)聯(lián)系我們，謝謝！