号段模式 VS 批量生成ID
所谓 号段模式 跟 批量生成ID 有什么区别？ 乍一看，感觉基本上是一个意思。 其实还是不同的！
批量生成ID
一次按需批量生成多个ID（ 也就是一批），每次生成都需要访问数据库，将数据库中对应的批 消耗掉，并在内存中记录下来。
批量生成是 一次获取一批的数据， 比如一次性一百个ID。 这些id 是可以无序的， 这个时候就不存在所谓的最大值、最小值；当然也可以是顺序递增的，那么其实 只需要最小值、最大值 就可以了吧。但是，它还是 一次性获取100个， 这个算是它的缺点； 当然，用多少就是多少， 用不完的 如果可以 退还， 那么是非常理想的情况。
优点：避免了每次生成ID都要访问数据库并带来压力，提高性能
缺点：属于本地生成策略，存在单点故障，服务重启造成ID不连续
而号段模式
是 一次获取一个 id 段（所谓号， 就是id吧）。什么是 id 段？ 就是一个id 的某个取值的区间段，包括最小值、最大值， 直白的说就是 两个值：最小值、最大值。一般就获取一个 最大值和步长就好了（ 起始值 由两者做减法而得出）。
有了最小值、最大值，而且 要求id 完全严格的顺序递增， 那么客户端保存一个当前的id值， 使用的时候让它在 号段区间内递增，那么就可以 大幅减轻数据库压力。
此模式，数据库中一个表保存 最大值和步长 即可。或者保存一个最小值和步长吧！
它如何避免 id 的重复发放？ 也就是如何避免id冲突？
每个客户端每次都重新获取一个号段，就可以避免id冲突。就是说每次都在 最大值的基础上增加一个步长，然后返回新的最大值。
号段模式的一般流程
1 客户端（本地） 尝试获取 分布式id，本地存在号段（或称号段缓存）？ 执行2，否则3
2 当前id > 当前号段最大值？ 执行3，否则4
3 从数据库获取新的号段， 包括步长、最大值，并设置它为 本地 当前号段。
4 从本地 当前号段 获取新的id。
如下图：
可见， 访问数据库， 减少到只有一个地方， 也就是第三步，读写数据库的频率从1减小到了1/step。
号段模式的新的问题
一个问题是，如果 客户端突然异常，号段没有用完，下次客户端重启了，再次访问数据库，就会获取新的号段。导致id 就不连续了。 不连续 其实 一般场景也问题不大，虽然不连续，但是还是可以保证顺序递增的，只是 不再是 依次递增了罢。再说， 某些业务，可能还要求 不能单调递增，以 防止 被 猜测出 业务数据呢！
另外的 问题是， 在分布式环境， 可能有多个客户端，这样做 无法保证全局有序。 比如客户端C1、C2 先后非常短的时间去获取号段，C1 获取的是0-1000， C2 获取的是1000-2000， c1 使用到3， 而c2 可能是使用到 1002， 那么， 我们的业务上就出现这样的数据： 虽然相隔时间不多，但是 id 值 相差很大。 如果有更多的客户端，可能出现的id 差更大！
—— 怎么办？ 看场景， 其实 我们可以给客户端加一个标志，那么此种情况仍然可以保证 至少是 单个客户端 顺序递增的。而整体上， 可以保证 基本有序。 就是说如果不要求保证 整体、全局有序， 那么这个问题 也不是问题， 忽略即可。 其他情况呢？ 那就真不好解决，只能等待一个号段消费完成，再去消费下一个号段，显然非常不合理。
分布式id 一定是id 吗？ 也不一定， 只要是 作用跟id 差不多就可以了。
美团Leaf 号段模式
美团Leaf 在普通的号段基础上，又进一步优化， 提出了“双buffer优化” ， 但是感觉这个 所谓的“双buffer优化” ， 意义不大， 它仅仅是在 “取DB的时候网络发生抖动，或者DB发生慢查询就会导致整个系统的响应时间变慢”， 这种可能性不是没有， 但是过于吹毛求疵了吧！ 当然， 对应美团这样的公司，可能有这个需求，一般公司 肯定是不需要这样高要求的！
不管怎么样， 我们先看看 所谓的“双buffer优化” ：
它的理念是 “ DB取号段的过程能够做到无阻塞，不需要在DB取号段的时候阻塞请求线程，即当号段消费到某个点时就异步的把下一个号段加载到内存中。而不需要等到号段用尽的时候才去更新号段。这样做就可以很大程度上的降低系统的TP999指标 ”，
双buffer优化的详细实现如下图：
采用双buffer的方式，Leaf服务内部有两个号段缓存区segment。当前号段已下发10%时，如果下一个号段未更新，则另启一个更新线程去更新下一个号段。当前号段全部下发完后，如果下个号段准备好了则切换到下个号段为当前segment接着下发，循环往复。
- 每个biz-tag都有消费速度监控，通常推荐segment长度设置为服务高峰期发号QPS的600倍（10分钟），这样即使DB宕机，Leaf仍能持续发号10-20分钟不受影响。
- 每次请求来临时都会判断下个号段的状态，从而更新此号段，所以偶尔的网络抖动不会影响下个号段的更新。
——
"当前号段已下发10%时，如果下一个号段未更新，则另启一个更新线程去更新下一个号段。当前号段全部下发完后，如果下个号段准备好了则切换到下个号段为当前segment接着下发，循环往复"
—— 1 什么是“下一个号段未更新” ， 如果已经更新呢 ？
—— 2 如果下个号段没有准备好，咋办？
—— 说话不能只说半截， 考虑需要全面啊。 不能只有if ，没有else。。
当前号段已使用超过10%时，则启动新线程获取下一个可用号段，然后缓存到本地？ 这样做， 感觉也是可行的，不过会导致 号段差更大吧。
这个图，不太友好， 看了相当于没看。
具体还是得看源码 ↓↓
美团Leaf号段模式源码分析
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 | public class SegmentIDGenImpl implements IDGen { private static final Logger logger = LoggerFactory.getLogger(SegmentIDGenImpl.class); /** * IDCache未初始化成功时的异常码 */ private static final long EXCEPTION_ID_IDCACHE_INIT_FALSE = -1; /** * key不存在时的异常码 */ private static final long EXCEPTION_ID_KEY_NOT_EXISTS = -2; /** * SegmentBuffer中的两个Segment均未从DB中装载时的异常码 */ private static final long EXCEPTION_ID_TWO_SEGMENTS_ARE_NULL = -3; /** * 最大步长不超过100,0000 */ private static final int MAX_STEP = 1000000; /** * 一个Segment维持时间为15分钟 */ private static final long SEGMENT_DURATION = 15 * 60 * 1000L; // 最多5个线程，也就是最多5个任务同时执行（因为可能有多个tag，如果tag 只有1、2个，那么没必要5个线程）；idle时间是60s； // SynchronousQueue意思是，只能有一个线程执行一个tag的任务，立即执行，执行完立即获取；其他的都只能阻塞式等待 private ExecutorService service = new ThreadPoolExecutor(5, Integer.MAX_VALUE, 60L, TimeUnit.SECONDS, new SynchronousQueue<Runnable>(), new UpdateThreadFactory()); private volatile boolean initOK = false; // 注意它包含了所有的SegmentBuffer，k是业务类型，v是对应的 // 全局缓存 private Map<String, SegmentBuffer> cache = new ConcurrentHashMap<String, SegmentBuffer>(); private IDAllocDao dao; public static class UpdateThreadFactory implements ThreadFactory { private static int threadInitNumber = 0; private static synchronized int nextThreadNum() { return threadInitNumber++; } @Override public Thread newThread(Runnable r) { return new Thread(r, "Thread-Segment-Update-" + nextThreadNum()); } } @Override public boolean init() { logger.info("Init ..."); // 确保加载到kv后才初始化成功 updateCacheFromDb(); initOK = true; updateCacheFromDbAtEveryMinute(); return initOK; } private void updateCacheFromDbAtEveryMinute() {// 顾名思义， 每分钟执行一次。执行什么？ 执行 updateCacheFromDb方法 ScheduledExecutorService service = Executors.newSingleThreadScheduledExecutor(new ThreadFactory() { @Override public Thread newThread(Runnable r) { Thread t = new Thread(r); t.setName("check-idCache-thread"); t.setDaemon(true); return t; } }); service.scheduleWithFixedDelay(new Runnable() { @Override public void run() { updateCacheFromDb(); } }, 60, 60, TimeUnit.SECONDS); }// 通过数据库 来更新cache缓存， 也就是Map<String, SegmentBuffer> cache。注意它包含了所有的SegmentBuffer// 总结就是， 把db新增的tag， 初始化并添加到cache，把db删除的tag，从cache中删除；db中更新的呢？这里不管，后面由更新线程去维护到cache// init的时候执行一次，后面每分钟执行一次 private void updateCacheFromDb() { logger.info("update cache from db"); StopWatch sw = new Slf4JStopWatch(); try { List<String> dbTags = dao.getAllTags();// 可能有新加的tags， 这里仅仅加载 tag，不包括value if (dbTags == null || dbTags.isEmpty()) { return; } List<String> cacheTags = new ArrayList<String>(cache.keySet()); Set<String> insertTagsSet = new HashSet<>(dbTags); Set<String> removeTagsSet = new HashSet<>(cacheTags); //db中新加的tags灌进cache for(int i = 0; i < cacheTags.size(); i++){ String tmp = cacheTags.get(i); if(insertTagsSet.contains(tmp)){ insertTagsSet.remove(tmp); // 数据库中已经存在于cache中的，是旧的，先删除 } } for (String tag : insertTagsSet) {// 现在insertTagsSet的部分都是全新的tags SegmentBuffer buffer = new SegmentBuffer();// 默认 SegmentBuffer的init 是false buffer.setKey(tag); Segment segment = buffer.getCurrent();// currentPos 永远只会在 0/1之间切换 segment.setValue(new AtomicLong(0));// value表示 实际的分布式的唯一id值 segment.setMax(0); // max 是 当前号段的 最大值； step 是步长，它会根据消耗的速度自动调整！ segment.setStep(0);// 默认 value、max、 step 都是0 —— 也就是， 全部重置！ cache.put(tag, buffer);// 添加到全局的 cache logger.info("Add tag {} from db to IdCache, SegmentBuffer {}", tag, buffer); } //cache中已失效的tags从cache删除 for(int i = 0; i < dbTags.size(); i++){ String tmp = dbTags.get(i); if(removeTagsSet.contains(tmp)){ // 两个contains 都是在寻找重叠部分 removeTagsSet.remove(tmp);//cache中存在于数据库中的，先从缓存中删除 } } for (String tag : removeTagsSet) { // 现在removeTagsSet的部分都是是旧的、已经失效的tags cache.remove(tag); // 又从全局的 cache中删除。 logger.info("Remove tag {} from IdCache", tag); } } catch (Exception e) { logger.warn("update cache from db exception", e); } finally { sw.stop("updateCacheFromDb"); } } @Override public Result get(final String key) { if (!initOK) { // 全局cache 是否初始化完成？ 一般是不会说没有完成初始化 return new Result(EXCEPTION_ID_IDCACHE_INIT_FALSE, Status.EXCEPTION); } if (cache.containsKey(key)) { SegmentBuffer buffer = cache.get(key); if (!buffer.isInitOk()) {// 对应key的buffer 是否初始化完成？ 第一次调用get方法肯定是false synchronized (buffer) { // 没完成则加锁， 因为可能有多线程调用此get 方法 if (!buffer.isInitOk()) {// 再次判断， 防止 xxx问题 try { updateSegmentFromDb(key, buffer.getCurrent());// 通过数据库 初始化、更新buffer中 的当前号段 logger.info("Init buffer. Update leafkey {} {} from db", key, buffer.getCurrent()); buffer.setInitOk(true);// 必须到这里 才算完成buffer的 初始化； 也是全局唯一被调用的地方！ } catch (Exception e) { logger.warn("Init buffer {} exception", buffer.getCurrent(), e); } } } } // 到这里，肯定已经完成对应key的buffer的初始化， 所以直接从号段缓存中 获取id return getIdFromSegmentBuffer(cache.get(key));// 直接从号段缓存中 获取id } //可能数据库中不存在对应的key, 那就暂时只能异常返回 return new Result(EXCEPTION_ID_KEY_NOT_EXISTS, Status.EXCEPTION); }// 通过数据库 更新buffer中 的号段// 注意 initOK为true，号段 缓存可能还没有初始化； 也就是init() 方法实际并没有对缓存做 初始化。// 全局被调用的地方， 就两处 public void updateSegmentFromDb(String key, Segment segment) { StopWatch sw = new Slf4JStopWatch(); SegmentBuffer buffer = segment.getBuffer(); LeafAlloc leafAlloc; if (!buffer.isInitOk()) {// 再次判断，第一次调用get方法肯定init=false，于是进入这个if leafAlloc = dao.updateMaxIdAndGetLeafAlloc(key);// 直接查询数据库 buffer.setStep(leafAlloc.getStep()); buffer.setMinStep(leafAlloc.getStep());//leafAlloc中的step为DB中的step； 最小步长，表明真正的步长是可能大于minStep的 } else if (buffer.getUpdateTimestamp() == 0) {// 如果buffer已经init过，第二次会进这个if， 因为setUpdateTimestamp全局被调用两次，这个if和下一个else leafAlloc = dao.updateMaxIdAndGetLeafAlloc(key);// buffer.setUpdateTimestamp(System.currentTimeMillis());// 唯二的全局被调用；为什么有这个if， 为什么特地需要特地调用这个方法 buffer.setStep(leafAlloc.getStep());// 和上一个if 是不是重复了？ 代码是不是有些啰嗦？ buffer.setMinStep(leafAlloc.getStep());// minStep的作用仅仅是为了 动态调整， 实际使用的还是 step !! } else {// 如果buffer已经init过，第三次会进这个else； 这里其实是根据duration 动态调整步长！ long duration = System.currentTimeMillis() - buffer.getUpdateTimestamp(); int nextStep = buffer.getStep(); if (duration < SEGMENT_DURATION) {// 如果duration 太短了， 那么说明步长有点小，id 消耗快，那么进入调整。 if (nextStep * 2 > MAX_STEP) { // 如果*2 会超出最大步长，那就不调整了，否则就*2 //do nothing } else { nextStep = nextStep * 2; } } else if (duration < SEGMENT_DURATION * 2) { 如果 SEGMENT_DURATION < duration < SEGMENT_DURATION * 2, 那么不管 //do nothing with nextStep } else {// 否则就是 > SEGMENT_DURATION * 2; 那么 步长除以2，但是不能小于min nextStep = nextStep / 2 >= buffer.getMinStep() ? nextStep / 2 : nextStep; } logger.info("leafKey[{}], step[{}], duration[{}mins], nextStep[{}]", key, buffer.getStep(), String.format("%.2f",((double)duration / (1000 * 60))), nextStep); LeafAlloc temp = new LeafAlloc(); temp.setKey(key); temp.setStep(nextStep); leafAlloc = dao.updateMaxIdByCustomStepAndGetLeafAlloc(temp);// 动态调整记录到数据库中~ buffer.setUpdateTimestamp(System.currentTimeMillis());// 唯二的全局被调用 buffer.setStep(nextStep); buffer.setMinStep(leafAlloc.getStep());//leafAlloc的step为DB中的step } // must set value before set max long value = leafAlloc.getMaxId() - buffer.getStep();//leafAlloc.getMaxId()是后端新的最大值，buffer.getStep()是新的步长 segment.getValue().set(value);// value 和max 构成了 号段的起止值； value 其实命名并不好，容易让人误解！ segment.setMax(leafAlloc.getMaxId()); // value 、max、step 都设置好了，表示 号段初始化、更新完成 segment.setStep(buffer.getStep()); sw.stop("updateSegmentFromDb", key + " " + segment); }// 直接从号段缓存中 获取id// 获取的时候， 需要检查是否超过了 10%， 超过则 另外启动任务去异步 加载新的 号段！ public Result getIdFromSegmentBuffer(final SegmentBuffer buffer) { while (true) { buffer.rLock().lock(); try { final Segment segment = buffer.getCurrent(); // 每次get 都会进入这里，但是其实 isNextReady条件 只有设置一次 if (!buffer.isNextReady() && (segment.getIdle() < 0.9 * segment.getStep()) && buffer.getThreadRunning().compareAndSet(false, true)) { service.execute(new Runnable() {// 异步执行 @Override public void run() { // 进入了这个方法，buffer.getThreadRunning 已经被cas设置为了true； 保证了 这个方法不会有并发 Segment next = buffer.getSegments()[buffer.nextPos()]; boolean updateOk = false; try { updateSegmentFromDb(buffer.getKey(), next);// next 是指使用 下一个号段；这里也就是更新下一个号段 updateOk = true; logger.info("update segment {} from db {}", buffer.getKey(), next); } catch (Exception e) { logger.warn(buffer.getKey() + " updateSegmentFromDb exception", e); } finally { if (updateOk) { buffer.wLock().lock();// 这里使用了 写锁， 是因为有写 的并发竞争 buffer.setNextReady(true);// 表示下一个号段已经准备好！ buffer.getThreadRunning().set(false);// 操作完毕，又把buffer线程状态设置为false，因为task 已经执行完 buffer.wLock().unlock(); } else { buffer.getThreadRunning().set(false);// 不管怎么样，需要把buffer线程状态设置为false，因为task 已经执行完 } } } }); } long value = segment.getValue().getAndIncrement();// 这里可能有读的并发， if (value < segment.getMax()) { // 未达到号段的最大值，也就是最右端 return new Result(value, Status.SUCCESS);// 到这里， 就成功返回； 一般情况 就是进入这个if 然后返回！ } } finally { buffer.rLock().unlock(); // 不管怎么样，这里释放读锁 } //什么情况会需要wait？ 上面的方法没有进入上一个if已经达到号段的最大值！ 或者出现异常，当然一般不会有异常； //达到号段的最大值 意味着需要使用下一个号段， // waitAndSleep其实是等待 正在执行的线程把任务执行完成；具体是 判断并自旋10000，然后超过10000，那就每次sleep 10毫秒，然后退出.. waitAndSleep(buffer);// 总之，waitAndSleep保证没有正在执行的更新线程； 但也不是100% 保证！ buffer.wLock().lock();// 执行写锁， 排斥任何其他的线程！ try { final Segment segment = buffer.getCurrent();// 能继续使用同一个号段吗？ long value = segment.getValue().getAndIncrement(); if (value < segment.getMax()) {// 为什么需要再次判断？ max会在已经确定的情况下变化？ 也不会，大概是保险起见？ return new Result(value, Status.SUCCESS); } if (buffer.isNextReady()) {// 已经达到号段的最大值，此时前面必然已经完成了新号段的获取， 肯定进入此判断 buffer.switchPos(); // 全局唯一的 切换！ 虽然完成了切换， 但是不立即获取value，而是等待下一次的循环！ buffer.setNextReady(false); } else {// 这种情况， 不太可能发生吧！ logger.error("Both two segments in {} are not ready!", buffer); return new Result(EXCEPTION_ID_TWO_SEGMENTS_ARE_NULL, Status.EXCEPTION); } } finally { buffer.wLock().unlock(); } } } private void waitAndSleep(SegmentBuffer buffer) { int roll = 0; while (buffer.getThreadRunning().get()) { roll += 1; if(roll > 10000) { try { TimeUnit.MILLISECONDS.sleep(10); break;// 不管怎么样，需要退出， 不能在这个方法等太久。 就是说， 极端情况下， 此方法不能保证 等待正在执行的任务完成。 } catch (InterruptedException e) { logger.warn("Thread {} Interrupted",Thread.currentThread().getName()); break; } } } } public List<LeafAlloc> getAllLeafAllocs() { return dao.getAllLeafAllocs(); } public Map<String, SegmentBuffer> getCache() { return cache; } public IDAllocDao getDao() { return dao; } public void setDao(IDAllocDao dao) { this.dao = dao; }} |
代码质量还是不错的，简直是神乎其神的刀法！
美团leaf号段模式的部署和使用
对于数据库的leaf_alloc 表而言，我们只会更新其max_id 字段，不会更新step；
@Update("UPDATE leaf_alloc SET max_id = max_id + #{step} WHERE biz_tag = #{key}") @Update("UPDATE leaf_alloc SET max_id = max_id + step WHERE biz_tag = #{tag}")
step 是初始化完成的，默认应该是使用方。
官方提供了leaf-core、leaf-server。 其中 leaf-core是核心算法的实现， 但是它主要依赖于mybatis、数据库，但是它不能直接运行。 leaf-server 调用了它， 并负责完成leaf_alloc 表的初始化。对于号段模式，主要就是在SegmentService， 其核心作用就是 实例化了一个IDGen（具体是SegmentIDGenImpl），然后 使用它对外提供分布式id服务。
前面说错了，leaf_alloc 表的初始化并不是 leaf-server 程序完成的，而是需要手动新增自己需要的业务数据， 其实就是一条insert 语句，参见官方：
insert into leaf_alloc(biz_tag, max_id, step, description) values('leaf-segment-test', 1, 2000, 'Test leaf Segment Mode Get Id')
这个务必记住。
实际怎么做？
因为它不知道你的业务tag和step 是多少啊。 tag 当然是自定义， step 对于高并发的项目就1000，一般小项目就100够了吧！
所以呢，想简单使用的话，一般直接部署一个leaf-server就好了。
leaf-server 是必须的吗
leaf-server 可以独立运行和工作，并提供了rest http接口来获取id，好像也非常合理好用，我们直接使用它就可以了，那么它是不是必须的呢？肯定不是。原因是：
1 它本身有可能存在单点的问题了。
2 如果你不想通过http方式获取id（这样显然有性能损失）
3 它强制依赖了druid
我认为我们可以适当做一些改造、集成。 就是把leaf-server 当做一个普通jar 引入，或者只用其中需要的几个类，或者，完全不用它，就自己写； 因为它强制依赖了druid，如果我们已经使用了比如hikari 这样的数据连接池， 需要统一连接池， 不想引入其他的连接池， 那么我们完全可以 按照自己的方式实例化IDGen ！
参考
【推荐】国内首个AI IDE，深度理解中文开发场景，立即下载体验Trae
【推荐】编程新体验，更懂你的AI，立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包，你的智能百科全书，全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell：AI 加持，快人一步
· AI与.NET技术实操系列：向量存储与相似性搜索在 .NET 中的实现
· 基于Microsoft.Extensions.AI核心库实现RAG应用
· Linux系列：如何用heaptrack跟踪.NET程序的非托管内存泄露
· 开发者必知的日志记录最佳实践
· SQL Server 2025 AI相关能力初探
· 震惊！C++程序真的从main开始吗？99%的程序员都答错了
· 【硬核科普】Trae如何「偷看」你的代码？零基础破解AI编程运行原理
· 单元测试从入门到精通
· 上周热点回顾（3.3-3.9）
· winform 绘制太阳，地球，月球 运作规律