使用數(shù)據(jù)庫和Redis開發(fā)各系統(tǒng)獨(dú)立的自增ID生成器
作者：架構(gòu)師筆記
在分布式系統(tǒng)，需要生成唯一ID的系統(tǒng)不止一個(gè)，這些ID的生成在各自業(yè)務(wù)內(nèi)是獨(dú)立的，同時(shí)包含業(yè)務(wù)規(guī)則，如果要利用數(shù)據(jù)庫自增主鍵生成唯一ID，需要每個(gè)業(yè)務(wù)部門創(chuàng)建一張表來為唯一ID服務(wù)。
背景
使用業(yè)務(wù)場(chǎng)景：
1.利用數(shù)據(jù)庫自增主鍵生成唯一ID，無法滿足各個(gè)系統(tǒng)獨(dú)自生成自增的唯一ID需求。
在分布式系統(tǒng)，需要生成唯一ID的系統(tǒng)不止一個(gè)，這些ID的生成在各自業(yè)務(wù)內(nèi)是獨(dú)立的，同時(shí)包含業(yè)務(wù)規(guī)則，如果要利用數(shù)據(jù)庫自增主鍵生成唯一ID，需要每個(gè)業(yè)務(wù)部門創(chuàng)建一張表來為唯一ID服務(wù)。
2.數(shù)據(jù)庫自增主鍵無法滿足性能要求。
而因?yàn)镽edis的高性能和單線程特性，使它在這方面有獨(dú)特的優(yōu)勢(shì)。
為了解決各個(gè)系統(tǒng)獨(dú)自生成自增的唯一ID的需求，而又實(shí)現(xiàn)簡(jiǎn)單有效，做了如下設(shè)計(jì)。
各系統(tǒng)獨(dú)立的自增ID生成器
數(shù)據(jù)庫設(shè)計(jì)(mysql)
DROP TABLE IF EXISTS sys_id_record;
CREATE TABLE sys_id_record(
id INT UNSIGNED AUTO_INCREMENT,
sys_id VARCHAR(40) NOT NULL COMMENT '子系統(tǒng)ID',
sys_name VARCHAR(100) NOT NULL COMMENT '子系統(tǒng)名稱',
id_from INT NOT NULL DEFAULT 1 COMMENT '子系統(tǒng)ID起始值',
lenth SMALLINT NOT NULL DEFAULT 10 COMMENT '子系統(tǒng)ID長度',
enabled VARCHAR(1) NOT NULL DEFAULT 'Y',
created_by VARCHAR(40) NOT NULL COMMENT '創(chuàng)建人',
created_date DATETIME COMMENT '創(chuàng)建時(shí)間',
updated_by VARCHAR(40) NOT NULL COMMENT '更新人',
updated_date DATETIME DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新時(shí)間',
PRIMARY KEY ( id )
)ENGINE=INNODB DEFAULT CHARSET=utf8;
主要代碼
- 定義服務(wù)接口
package com.davidwang456.test;
public interface IDGenerateService {
String getIDGenerateBySys(String sysId);
}
- 接口服務(wù)實(shí)現(xiàn)
package com.davidwang456.test;
import java.util.ArrayList;
import java.util.List;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;
@Service
public class IDGenerateServiceImpl implements IDGenerateService {
@Autowired
private SysIdRecordDao sysIdRecordDao;
@Autowired
private RedisTemplate redisTemplate;
@Override
public String getIDGenerateBySys(String sysId) {
init(sysId);
Object idSys=redisTemplate.opsForList().rightPop(sysId);
if(idSys==null) {
List<SysIdRecord> records=sysIdRecordDao.getEnabledSysIdRecord(sysId);
if(records!=null&&records.isEmpty()) {
SysIdRecord record=records.get(0);
List<Integer> buf_ids=new ArrayList<>();
for(int i=1;i<record.getLenth+1;i++) {
buf_ids.add(record.getIdFrom+i);
}
redisTemplate.opsForList().leftPushAll(sysId, buf_ids);
record.setIdFrom(record.getIdFrom()+record.getLenth());
sysIdRecordDao.save(record);
idSys=redisTemplate.opsForList().rightPop(sysId);
}
}
return sysId+":"+idSys;
}
private void init(String sysId) {
if(!redisTemplate.hasKey(sysId)) {
List<SysIdRecord> records=sysIdRecordDao.getEnabledSysIdRecord(sysId);
if(records==null||records.isEmpty()) {
SysIdRecord record=new SysIdRecord();
//TODO
sysIdRecordDao.save(record);
}
}
}
}
總結(jié)
這個(gè)設(shè)計(jì)的核心思想是數(shù)據(jù)庫控制各個(gè)子系統(tǒng)的起始值和步長，然后將值放到redis隊(duì)列中，從隊(duì)列中依此取值。當(dāng)從redis中消耗掉后再次從數(shù)據(jù)庫中取一批值，依此循環(huán)。
責(zé)任編輯：張燕妮
來源：
今日頭條
