使用數(shù)據(jù)庫和Redis開發(fā)各系統(tǒng)獨立的自增ID生成器
背景
使用業(yè)務場景:
1.利用數(shù)據(jù)庫自增主鍵生成唯一ID,無法滿足各個系統(tǒng)獨自生成自增的唯一ID需求。
在分布式系統(tǒng),需要生成唯一ID的系統(tǒng)不止一個,這些ID的生成在各自業(yè)務內(nèi)是獨立的,同時包含業(yè)務規(guī)則,如果要利用數(shù)據(jù)庫自增主鍵生成唯一ID,需要每個業(yè)務部門創(chuàng)建一張表來為唯一ID服務。
2.數(shù)據(jù)庫自增主鍵無法滿足性能要求。
而因為Redis的高性能和單線程特性,使它在這方面有獨特的優(yōu)勢。
為了解決各個系統(tǒng)獨自生成自增的唯一ID的需求,而又實現(xiàn)簡單有效,做了如下設計。
各系統(tǒng)獨立的自增ID生成器
數(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)建時間',
updated_by VARCHAR(40) NOT NULL COMMENT '更新人',
updated_date DATETIME DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新時間',
PRIMARY KEY ( id )
)ENGINE=INNODB DEFAULT CHARSET=utf8;
主要代碼
- 定義服務接口
package com.davidwang456.test;
public interface IDGenerateService {
String getIDGenerateBySys(String sysId);
}
- 接口服務實現(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é)
這個設計的核心思想是數(shù)據(jù)庫控制各個子系統(tǒng)的起始值和步長,然后將值放到redis隊列中,從隊列中依此取值。當從redis中消耗掉后再次從數(shù)據(jù)庫中取一批值,依此循環(huán)。