利用AtomicInteger实现高性能的ID生成器

废话不多说,直接上代码,目前该生成器用于生成流量平台的订单号:

package com.base.common.util.idworker;

import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.concurrent.atomic.AtomicInteger;

public class IdWorker {

    private static long lastTimes = 0;
    private static AtomicInteger currentSequence = new AtomicInteger(100);

    /**
     * 
     * @param machineId
     *            机器编码:0000-9999,分布式部署的时候指定不同的机器编号
     * @return
     */
    public static String nextId(String machineId) {
        StringBuffer sb = new StringBuffer(machineId);
        long currentTimes = System.currentTimeMillis();
        if (lastTimes == currentTimes) {
            int result = currentSequence.incrementAndGet();
            if (result > 999) {
                currentTimes = genTimes(lastTimes);
            }
            sb.append(getTimeString(currentTimes)).append(result);
        } else {
            lastTimes = currentTimes;
            currentSequence = new AtomicInteger(100);
            sb.append(getTimeString(currentTimes)).append(currentSequence);
        }

        return sb.toString();
    }

    private static long genTimes(long lastTimes) {
        try {
            Thread.sleep(1);
        } catch (InterruptedException e) {
            // do nothing
        }
        long times = System.currentTimeMillis();
        if (times == lastTimes) {
            return genTimes(lastTimes);
        }
        return times;
    }

    private static String getTimeString(long currentTimes) {
        Date date = new Date(currentTimes);
        SimpleDateFormat sdf = new SimpleDateFormat("yyMMddHHmmssSSS");
        return sdf.format(date);
    }

    public static void main(String[] args) throws Exception {
        Thread t1 = new Thread(new Runnable() {

            @Override
            public void run() {
                for(;;) {
                    System.out.println("t1"+IdWorker.nextId("1"));
                }
            }
        });
        Thread t2 = new Thread(new Runnable() {

            @Override
            public void run() {
                for(;;) {
                    System.out.println("t2"+IdWorker.nextId("1"));
                }
            }
        });
        t1.start();
        t2.start();
        t1.join();
        t2.join();
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值