大家好，我是3y，今天繼續來聊我的開源項目austin啊，但實際內容更新不多。這文章主是想吹下水，主要聊聊我在更新項目中學到的小技巧。
今天所說的小技巧可能有很多人都會，但肯定也會有跟我一樣之前沒用過的。
消息推送平臺🔥推送下發【郵件】【短信】【微信服務號】【微信小程序】【企業微信】【釘釘】等消息類型。
Spring注入集合
之前我一直不知道，原來Spring是能注入集合的，直到一個pull request被提了過來。
https://gitee.com/zhongfucheng/austin/pulls/31
我之前寫了一個自定義註解，它的作用就是收集自定義註解所標識的Bean，然後最後把這些Bean放到Map裏
@Component
public class SmsScriptHolder {
private Map<String, SmsScript> handlers = new HashMap<>(8);
public void putHandler(String scriptName, SmsScript handler) {
handlers.put(scriptName, handler);
}
public SmsScript route(String scriptName) {
return handlers.get(scriptName);
}
}
/**
* 標識 短信渠道
*
* @author 3y
*/
@Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.TYPE})
@Component
public @interface SmsScriptHandler {
/**
* 這裏輸入腳本名
*
* @return
*/
String value();
}
/**
* sms發送腳本的抽象類
*
* @author 3y
*/
@Slf4j
public abstract class BaseSmsScript implements SmsScript {
@Autowired
private SmsScriptHolder smsScriptHolder;
@PostConstruct
public void registerProcessScript() {
if (ArrayUtils.isEmpty(this.getClass().getAnnotations())) {
log.error("BaseSmsScript can not find annotation!");
return;
}
Annotation handlerAnnotations = null;
for (Annotation annotation : this.getClass().getAnnotations()) {
if (annotation instanceof SmsScriptHandler) {
handlerAnnotations = annotation;
break;
}
}
if (handlerAnnotations == null) {
log.error("handler annotations not declared");
return;
}
//註冊handler
smsScriptHolder.putHandler(((SmsScriptHandler) handlerAnnotations).value(), this);
}
}
結果，pull request提的代碼過來特別簡單就替代了我的代碼了。只要在使用的時候，直接注入Map：
@Autowired
private Map<String, SmsScript> smsScripts;
這一行代碼就能夠實現，把SmsScript的實現類都注入到這個Map裏。同樣的，我們亦可以使用List<Interface> 把該接口下的實現類都注入到這個List裏。
這好奇讓我去看看Spring到底是怎麼實現的，但實際上並不難。入口在org.springframework.beans.factory.annotation.AutowiredAnnotationBeanPostProcessor.AutowiredFieldElement#inject
接着定位到：org.springframework.beans.factory.support.DefaultListableBeanFactory#resolveDependency
深入 org.springframework.beans.factory.support.DefaultListableBeanFactory#doResolveDependency
最後實現注入的位置： org.springframework.beans.factory.support.DefaultListableBeanFactory#resolveMultipleBeans 數組 相關實現
access_token存儲到Redis
在接入微信相關渠道時，我就說過austin藉助了wxjava這個開源組件庫（該組件庫對接微信相關api，使調用變得尤其簡單）。
比如，我們調用微信的api是需要access_token的參數的。如果是我們自己編寫代碼調用微信api，那我們需要先獲取access_token，然後把該access_token拼接在url上。此時，我們又需要考慮access_token會不會失效了，失效了我們要有重試的策略。
而wxjava把這些都封裝好了，屏蔽了內部實現細節。只要我們把微信渠道的賬號信息寫到WxMpConfigStorage裏，那該組件就會幫我們去拿到access_token，內部也會有相應的重試策略。
第一版我爲了圖方便，我是使用WxMpDefaultConfigImpl實現類把渠道相關信息存儲在本地內存裏（包括access_token），而在上週我把渠道相關信息轉都存儲至Redis。
主要是獲取access_token它的調用次數是有限的，如果項目集羣部署，而access_token又存儲在本地內存中，那就很大概率不到一天時間調用獲取access_token次數就滿了，要是拿不到access_token，那就沒辦法調用微信的接口了。
https://developers.weixin.qq.com/doc/offiaccount/Basic_Information/Get_access_token.html
對於wxjava這個組件庫，調用微信的api都是通過Wx(xx)Service來使用的，而我是想把Wx(xx)Service做成是單例的。那在實現access_token存儲到Redis的時候，我就很自然就要對舊代碼進行一波重構（因爲第一版寫出來的代碼，多多少少都有點不滿意）。
歷史背景：
1、WxServiceUtils的邏輯是項目啓動的時檢索數據庫裏所有的微信渠道賬號信息，將Wx(xx)Service寫入到Map裏。Wx(xx)Service要做成單例自然就會想到用Map存儲（因爲消息推送平臺很可能會對接很多個服務號或者小程序，這裏數據結構肯定優先是Map啦）
如果渠道的賬號通過後臺有存在變更行爲，那程序內部會執行refresh()刷新。但這個僅僅是在程序內能監聽到的變更，如果是直接通過SQL修改表的記錄，目前是沒有機制刷新Map的內容的。
2、AccountUtils的邏輯是程序運行時得到發送賬號的Id，通過Id去數據庫檢索賬號配置，實時返回賬號最新的內容。（除了微信渠道賬號，其他所有的渠道賬號都是在這裏獲取信息）
更新：把原有管理微信賬號信息的WxServiceUtils類給棄用了，將所有的發送渠道賬號信息都歸到AccountUtils進行管理。
Map.computeIfAbsent使用
在重構上面所講的邏輯時，我很快地寫出以下的代碼：
if (clazz.equals(WxMaService.class)) {
if (Objects.nonNull(miniProgramServiceMap.get(channelAccount))) {
return (T)miniProgramServiceMap.get(channelAccount);
}
WxMaService wxMaService = initMiniProgramService(JSON.parseObject(channelAccount.getAccountConfig(), WeChatMiniProgramAccount.class));
miniProgramServiceMap.put(channelAccount, wxMaService);
return (T) wxMaService;
} else if (clazz.equals(WxMpService.class)) {
if (Objects.nonNull(officialAccountServiceMap.get(channelAccount))) {
return (T)officialAccountServiceMap.get(channelAccount);
}
WxMpService wxMpService = initOfficialAccountService(JSON.parseObject(channelAccount.getAccountConfig(), WeChatOfficialAccount.class));
officialAccountServiceMap.put(channelAccount, wxMpService);
return (T) wxMpService;
}
等我寫完，然後簡單做了下自測，發現這代碼咋這麼醜啊，兩個if的邏輯實際上是一樣的。
我想，這一定會有什麼工具類能幫我去優化下這個代碼的，我正準備翻Hutool/Guava這種工具包時，我突然想起：JDK在1.8好像就提供了putIfXXX的方法啦，我還找個毛啊，直接看看JDK的方法能不能用先。
很快啊，我就找到了。
我首先看的是putIfAbsent，發現它實現很簡單，就是做了一層封裝。
default V putIfAbsent(K key, V value) {
V v = get(key);
if (v == null) {
v = put(key, value);
}
return v;
}
但卻很適合用來優化我上面的代碼。於是，很快啊，我就改成了這樣：
if (clazz.equals(WxMaService.class)) {
return (T) miniProgramServiceMap.putIfAbsent(channelAccount, initMiniProgramService(JSON.parseObject(channelAccount.getAccountConfig(), WeChatMiniProgramAccount.class)));
} else if (clazz.equals(WxMpService.class)) {
return (T) officialAccountServiceMap.putIfAbsent(channelAccount, initOfficialAccountService(JSON.parseObject(channelAccount.getAccountConfig(), WeChatOfficialAccount.class)));
}
這看着真簡潔啊，好像已經很完美了，本來有好幾行的代碼，優化了下變成了一行。
但我又思考了下，這個putIfAbsent的V我這邊傳入的是一個方法，每次這個方法都會執行的（不論我的Map裏有沒有這個K），這又感覺不太優雅了。
我又點進去computeIfAbsent看了下，嗯！這就是我想要的了：如果Map的V不存在時，纔去執行我生成V的邏輯
default V computeIfAbsent(K key,
Function<? super K, ? extends V> mappingFunction) {
Objects.requireNonNull(mappingFunction);
V v;
if ((v = get(key)) == null) {
V newValue;
if ((newValue = mappingFunction.apply(key)) != null) {
put(key, newValue);
return newValue;
}
}
return v;
}
（這個其實我在學lambda和stream流的時候曾經是體驗過的，我日常也會簡單寫點，只是不知道在JDK裏Map也有這樣的方法。）於是，最後的代碼就成了：
if (clazz.equals(WxMaService.class)) {
return (T) miniProgramServiceMap.computeIfAbsent(channelAccount, account -> initMiniProgramService(JSON.parseObject(account.getAccountConfig(), WeChatMiniProgramAccount.class)));
} else if (clazz.equals(WxMpService.class)) {
return (T) officialAccountServiceMap.computeIfAbsent(channelAccount, account -> initOfficialAccountService(JSON.parseObject(account.getAccountConfig(), WeChatOfficialAccount.class)));
}
又後來，等我發佈到Git倉庫後，有人提了pull request來修復ConcurrentHashMap的computeIfAbsent存在性能的問題。呀，不小心又學到了點東西。
https://bugs.openjdk.java.net/browse/JDK-8161372
微信掃碼登錄實現
我在生產環境下是沒有寫過「用戶登錄」的，導致有些業務功能我也不知道線上是怎麼實現的。而「用戶登錄註冊」這個功能之前會聽過和見識過一些技術棧「Shiro」、「JWT」、「Spring Security」、「CAS」、「OAuth2.0」等等。
但是，我的需求只是用來做簡單的校驗，不需要那麼複雜。如果就給我設計一張user表，對其簡單的增刪改查好像也滿足，但我又不想寫這樣的代碼，因爲我在大學的時候實現過類似的。
現在不都流行掃碼登錄嘛？我不是已經接入了微信服務號的模板消息了嗎，不正好有一個測試號給我去做嗎？於是就開幹了。
首先看看人家是怎麼寫的，於是被我找到了一篇博客：https://blog.51cto.com/cxhit/4924932
過程挺好懂的，就按着他給出的時序圖對着實現就完了。後端對我來說實現並不難，花的時間最長的還是在前端的交互上。畢竟我這當時選用的是低代碼平臺啊，不能隨便實現各種邏輯的啊。
在前端，就一個「輪詢」功能，要輪詢查看用戶是否已經訂閱登錄，就耗費了我很多時間在官方文檔上。後來，寫了不少的奇淫技巧，最後也就被我實現出來了。實現過程很糟糕，也不值一提，反正你們也不會從中學到什麼好東西，因爲我也沒有。
過程還是簡單複述下吧，後期可能也會有同學去實現這個功能。
1、首先我們要有一個接口，給到微信回調，所以我們一般會稱該接口爲回調接口。微信的一些重要的事件都會回調給我們，我們做響應的邏輯處理。就比如，用戶關注了服務號，這種消息微信就調用我們的接口。
2、在微信後臺配置我們的定義好的回調接口，給到微信進行回調。
（如果接口是通的，按正常的走，那就會配置成功）
3、編寫一個獲取微信帶參數的二維碼給到前端做展示。
4、前端拿到二維碼做展示，並且得到隨機生成的參數輪詢查看是否已登錄。
5、編寫檢查是否已登錄的接口給到前端進行判斷。（如果能從Redis裏拿到隨機參數，說明已經登錄了）
6、當用戶掃碼關注了服務號，則得到微信的回調。當用戶關注服務號時，會把隨機參數和openId傳給服務器，我則將信息存入Redis。
7、前端得知已登錄後，將用戶信息寫入localStorage
最後
每次代碼存在遇到“優雅”的寫法時，我都會懊惱自己怎麼不會，還吭哧吭哧地寫這破代碼這麼多年了。特別是Map.computeIfAbsent這個，我感覺沒理由我不知道呀。我從初學到現在工作主要用JDK 1.8，沒道理我現在才知道寫這個玩意。
有的時候都感覺我是不是已經是老古董了，新世界已經沒有承載我的船了。
不過寫開源項目有一大好處是，只要我的項目有人用，能大大提高我獲取“優雅”寫法的概率，這也是我一直推廣自己項目的一個原因之一。
如果想學Java項目的，強烈推薦我的開源項目消息推送平臺Austin（8K stars） ，可以用作畢業設計，可以用作校招，可以看看生產環境是怎麼推送消息的。開源項目消息推送平臺austin倉庫地址：
消息推送平臺🔥推送下發【郵件】【短信】【微信服務號】【微信小程序】【企業微信】【釘釘】等消息類型。