把簡單的東西“傻瓜化”是軟件開發(fā)追求的目標(biāo)之一雇盖。請看下圖:
左邊是在 application.yml 里配置了3個生成器,右邊可以直接注入到代碼中使用宫患,注意刊懈,不用寫任何代碼。這酸爽娃闲。
下面請看效果:
上面是3個生成器生成的第一個序號虚汛。哎吆,還不錯哦皇帮。
慢慢學(xué)會分析
序列號大家都非常熟悉卷哩,無非就是一個初始值、步長属拾,有時還有最大值将谊。這只是最基本的信息察蹲,還可以按需添加其他的搀崭。
很容易抽象出一個接口臊岸,如下代碼:
/**
* 序列號生成器
* @author lixinjie
* @since 2019-04-04
*/publicinterfaceSnGenerator{/**名稱逸寓,根據(jù)實際情況使用*/StringgetName();/**注冊到容器中的bean名稱*/StringgetBeanName();/**初始值*/longgetInitNum();/**步長*/longgetStep();/**獲取下一個序列號*/longnextNum();/**最大值*/longgetMaxNum();}
剩下的就是下面這三個問題了:
接口有了之后匈勋,自然就是基于Redis的實現(xiàn)了崎弃,這是遇到第一個問題纤虽。
還需要根據(jù)配置動態(tài)向容器中注冊bean定義狡刘,這是第二個問題襟诸。
自然需要從配置文件中讀出這些配置信息瓦堵,供上一步使用,這是第三個問題歌亲。
再來學(xué)會實現(xiàn)
接口實現(xiàn)時主要用到Redis的INCRBY命令菇用,這是個原子命令。即使你有多個節(jié)點陷揪,每個節(jié)點有多個線程同時來調(diào)用惋鸥,也是OK的。
而且key不存在時悍缠,首次調(diào)用時還會將key設(shè)置為0卦绣。這樣帶來的好處就是,程序不用考慮key是否存在扮休,直接調(diào)用自增就可以了迎卤。
唯一不爽的就是key只會被設(shè)置為0,如果初始值不是從0開始的玷坠,就真有些不爽蜗搔,最簡單的辦法就是在程序中把初始值作為偏移量疊加上去劲藐。
當(dāng)然還有另一個辦法,就是主動設(shè)置key的初始值樟凄。因為存在并發(fā)聘芜,自然要使用SETNX命令。這樣已經(jīng)完全OK缝龄,但還是會有人在心理上覺得不安全汰现。那就在應(yīng)用啟動階段執(zhí)行該命令,此時肯定不會有調(diào)用的叔壤。請看下面源碼:
/** * 基于Redis的實現(xiàn) *@authorlixinjie *@since2019-04-04 */publicclassRedisSnGeneratorimplementsSnGenerator{@AutowiredprivateStringRedisTemplate stringRedisTemplate;privateString name;privateString beanName;privatelonginitNum;privatelongstep;privatelongmaxNum;publicRedisSnGenerator(String name, String beanName,longinitNum,longstep,longmaxNum){this.name = name;this.beanName = beanName;this.initNum = initNum;this.step = step;this.maxNum = maxNum; }@PostConstructpublicvoidinit(){if(!stringRedisTemplate.hasKey(getName())) { stringRedisTemplate.opsForValue().setIfAbsent(getName(), String.valueOf(getInitNum())); } }@OverridepublicStringgetName(){returnname; }publicStringgetBeanName(){returnbeanName; }@OverridepubliclonggetInitNum(){returninitNum; }@OverridepubliclonggetStep(){returnstep; }@OverridepubliclongnextNum(){returnstringRedisTemplate.opsForValue().increment(getName(), getStep()); }@OverridepubliclonggetMaxNum(){returnmaxNum; }}
要動態(tài)注冊bean定義瞎饲,Spring框架提供了專用接口,BeanDefinitionRegistryPostProcessor:
publicinterfaceBeanDefinitionRegistryPostProcessorextendsBeanFactoryPostProcessor{/** *@paramregistry the bean definition registry used by the application context *@throwsorg.springframework.beans.BeansException in case of errors */voidpostProcessBeanDefinitionRegistry(BeanDefinitionRegistry registry)throwsBeansException;}
它是一個bean工廠后處理器炼绘,應(yīng)用啟動時框架會自動檢測并應(yīng)用該類型的實現(xiàn)類嗅战。
要使用配置文件中的內(nèi)容,Spring框架提供了專門的接口俺亮,EnvironmentAware:
publicinterfaceEnvironmentAwareextendsAware{/** * Set the {@codeEnvironment} that this component runs in. */voidsetEnvironment(Environment environment);}
它是一個Aware接口驮捍,應(yīng)用啟動時框架會自動檢測該類型的接口并把你需要的給你set進(jìn)去。
這塊源碼有點多脚曾,我們就只看兩部分吧东且。下面是從配置文件中讀出配置信息的:
private SnGeneInfo[] parseSnGeneInfos() { String prefix ="sngenerator"; String[] generators = environment.getProperty(prefix +".generators", String[].class); SnGeneInfo[] infos = new SnGeneInfo[generators.length];for(int i = 0; i < generators.length; i++) { infos[i] = buildSnGeneInfo(prefix, generators[i]); } return infos; } private SnGeneInfo buildSnGeneInfo(String prefix, String generator) { return new SnGeneInfo( prefix +":"+ generator, environment.getProperty(prefix +"."+ generator +".bean-name"), environment.getProperty(prefix +"."+ generator +".init-num", long.class), environment.getProperty(prefix +"."+ generator +".step", long.class), environment.getProperty(prefix +"."+ generator +".max-num", long.class) ); }
下面是注冊bean定義的:
@Override public void postProcessBeanDefinitionRegistry(BeanDefinitionRegistry registry) throwsBeansException { SnGeneInfo[] geneInfos = parseSnGeneInfos();System.out.println(logInfo(geneInfos));for (SnGeneInfo geneInfo : geneInfos) {BeanDefinitionBuilder bdb =BeanDefinitionBuilder.genericBeanDefinition(RedisSnGenerator.class);bdb.addConstructorArgValue(geneInfo.getKeyName());bdb.addConstructorArgValue(geneInfo.getBeanName());bdb.addConstructorArgValue(geneInfo.getInitNum());bdb.addConstructorArgValue(geneInfo.getStep());bdb.addConstructorArgValue(geneInfo.getMaxNum());registry.registerBeanDefinition(geneInfo.getBeanName(),bdb.getBeanDefinition());} }
需要很多序列的話?
如果需要非常多的序列生成器的話本讥,上面的方法不可取珊泳。可以采用“分組”序列生成器囤踩,每一組內(nèi)可以有足夠多的序列旨椒,且組與組之間互不影響晓褪。
下面請看使用方法堵漱,還是一樣的簡單:
如果想學(xué)習(xí)Java工程化、高性能及分布式涣仿、深入淺出勤庐。微服務(wù)、Spring好港,MyBatis愉镰,Netty源碼分析的朋友可以加我的Java高級交流:787707172,群里有阿里大牛直播講解技術(shù)钧汹,以及Java大型互聯(lián)網(wǎng)技術(shù)的視頻免費分享給大家丈探。
下面請看效果:
每一組內(nèi)部的key,就是上面的f1/f2/f3等拔莱,不用配置碗降,程序按需直接傳入即可隘竭。
此時獲取下一個序列號的方法需要帶一個參數(shù),就是用來傳這個key的讼渊。它是基于Redis的哈希(Hash)實現(xiàn)的动看。
/**
* 分組序列號生成器
* @author lixinjie
* @since 2019-04-04
*/publicinterfaceGroupSnGenerator{/**名稱,根據(jù)實際情況使用*/StringgetName();/**注冊到容器中的bean名稱*/StringgetBeanName();/**初始值*/longgetInitNum();/**步長*/longgetStep();/**獲取下一個序列號*/longnextNum(String identifier);/**最大值*/longgetMaxNum();}
此時也會遇到給hashkey設(shè)置初始值的問題爪幻,肯定和上面不一樣菱皆,請閱讀源碼吧。
/** * 基于Redis的實現(xiàn) *@authorlixinjie *@since2019-04-04 */publicclassRedisGroupSnGeneratorimplementsGroupSnGenerator{privateMap existHashKeys =newConcurrentHashMap<>();@AutowiredprivateStringRedisTemplate stringRedisTemplate;privateString name;privateString beanName;privatelonginitNum;privatelongstep;privatelongmaxNum;publicRedisGroupSnGenerator(String name, String beanName,longinitNum,longstep,longmaxNum){this.name = name;this.beanName = beanName;this.initNum = initNum;this.step = step;this.maxNum = maxNum; }@PostConstructpublicvoidinit(){if(!stringRedisTemplate.hasKey(getName())) { stringRedisTemplate.opsForHash().putIfAbsent(getName(),"flag", String.valueOf(getInitNum())); } }@OverridepublicStringgetName(){returnname; }publicStringgetBeanName(){returnbeanName; }@OverridepubliclonggetInitNum(){returninitNum; }@OverridepubliclonggetStep(){returnstep; }@OverridepubliclongnextNum(String identifier){if(!existHashKeys.containsKey(identifier)) { stringRedisTemplate.opsForHash().putIfAbsent(getName(), identifier, String.valueOf(getInitNum())); existHashKeys.putIfAbsent(identifier, Boolean.TRUE); }returnstringRedisTemplate.opsForHash().increment(getName(), identifier, getStep()); }@OverridepubliclonggetMaxNum(){returnmaxNum; }}
如果想學(xué)習(xí)Java工程化挨稿、高性能及分布式仇轻、深入淺出。微服務(wù)奶甘、Spring拯田,MyBatis,Netty源碼分析的朋友可以加我的Java高級交流:787707172甩十,群里有阿里大牛直播講解技術(shù)船庇,以及Java大型互聯(lián)網(wǎng)技術(shù)的視頻免費分享給大家。