分布式唯一id 分布式唯一订单号生成

思路是项目启动的时候分配给每台机器的IP一个机器码，这样可以避免分布式集群环境并发订单号重复

基本都可以使用了，拿去即用，也可以根据自己的规则生成机器码，后面代码代码基本可以不用动

 

package org.com.rsmall.admin.init;

import org.apache.commons.lang3.StringUtils;
import org.com.rsmall.core.config.redis.RedisConstants;
import org.com.rsmall.core.config.redis.RedisUtils;
import org.com.utils.IPUtils;
import org.com.utils.IdUtils;
import org.com.utils.log.Log4jKit;
import org.springframework.boot.CommandLineRunner;
import org.springframework.stereotype.Component;

import java.util.Set;

/**
 * @ClassName MachineIdRunner
 * @Author yupanpan
 * @Date 2021/1/12 15:38
 */
@Component
public class MachineIdRunner implements CommandLineRunner {

    @Override
    public void run(String... args) throws Exception {
        createMachineId();
    }

    private void createMachineId() {
        try {
            String ip = IPUtils.getIpV4();
            if(StringUtils.isNotBlank(ip)){
                Log4jKit.info("本机IP: {}",ip);
                Object o = RedisUtils.get(RedisConstants.getMachineIdKey(ip));
                if(o!=null){
                    Integer machineId=Integer.valueOf(o.toString());
                    IdUtils.machineId=machineId;
                }else {
                    Integer machineId=0;
                    Set<String> keys = RedisUtils.keys(RedisConstants.PREFIX_MACHINE_ID+"*");
                    for (String key : keys) {
                        Object o1 = RedisUtils.get(key);
                        Integer mId = Integer.valueOf(o1.toString());
                        if(mId>=machineId){
                            machineId=mId;
                        }
                        if(machineId==0){
                            machineId=1;
                        }
                    }
                    if(machineId>=31){
                        machineId=0;
                    }
                    if(machineId!=0){
                        ++machineId;
                    }
                    IdUtils.machineId=machineId;
                    RedisUtils.set(RedisConstants.getMachineIdKey(ip),machineId);
                }
                Log4jKit.info("ip: {},初始化机器码: {}",ip,IdUtils.machineId);
            }else {
                IdUtils.machineId=0;
            }
        }catch (Exception e){
            Log4jKit.error("初始化机器码失败",e);
            IdUtils.machineId=0;
        }
    }

}

 

package org.com.rsmall.core.config.redis;

/**
 * @ClassName RedisConstants
 * @Author yupanpan
 * @Date 2021/1/13 17:40
 */
public class RedisConstants {

    //机器码key前缀
    public static final String PREFIX_MACHINE_ID="MACHINE_ID";

    private static final String PREFIX_SELECT_PRODUCT="SELECT_PRODUCT";
    private static final String PREFIX_SUBMIT="SUBMIT";
    private static final String PREFIX_AGAIN_SUBMIT="AGAIN_SUBMIT";

    public static String getSelectProductKey(String value){
        return PREFIX_SELECT_PRODUCT+":"+value;
    }
    public static String getSubmitKey(String value){
        return PREFIX_SUBMIT+":"+value;
    }
    public static String getAgainSubmitKey(String value){
        return PREFIX_AGAIN_SUBMIT+":"+value;
    }
    public static String getMachineIdKey(String value){
        return PREFIX_MACHINE_ID+":"+value;
    }

}

 

package org.com.utils;

import org.com.utils.log.Log4jKit;

import java.net.Inet4Address;
import java.net.InetAddress;
import java.net.NetworkInterface;
import java.net.SocketException;
import java.util.Enumeration;

/**
 * @ClassName IPUtils
 * @Author yupanpan
 * @Date 2021/1/12 16:00
 */
public class IPUtils {

    /**
     * 获取本机ip IPV6地址 fe80:0:0:0:216:3eff:fe34:bc09%eth0
     * @return
     */
    public static InetAddress getInetAddressIPV6() {
        try {
            for (Enumeration<NetworkInterface> interfaces = NetworkInterface.getNetworkInterfaces(); interfaces.hasMoreElements();) {
                NetworkInterface networkInterface = interfaces.nextElement();
                if (networkInterface.isLoopback() || networkInterface.isVirtual() || !networkInterface.isUp()) {
                    continue;
                }
                Enumeration<InetAddress> addresses = networkInterface.getInetAddresses();
                if (addresses.hasMoreElements()) {
                    return addresses.nextElement();
                }
            }
        } catch (SocketException e) {
            Log4jKit.error("Error when getting host ip address: <{}>.", e.getMessage());
        }
        return null;
    }

    /**
     * 获取本机ip IPV4地址 101.201.77.63
     * @return
     */
    private static InetAddress getInetAddressIPV4() {
        try {
            Enumeration<NetworkInterface> allNetInterfaces = NetworkInterface.getNetworkInterfaces();
            while (allNetInterfaces.hasMoreElements()) {
                NetworkInterface netInterface = allNetInterfaces.nextElement();
                Enumeration<InetAddress> addresses = netInterface.getInetAddresses();
                while (addresses.hasMoreElements()) {
                    InetAddress address = addresses.nextElement();
                    if (address != null
                            && address instanceof Inet4Address
                            && !address.isLoopbackAddress() //loopback地址即本机地址，IPv4的loopback范围是127.0.0.0 ~ 127.255.255.255
                            && address.getHostAddress().indexOf(":") == -1) {
                        return address;
                    }
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }

    public static String getIpV4(){
        InetAddress address = getInetAddressIPV4();
        return address!=null?address.getHostAddress():"127.0.0.0";
    }

    public static String getIpV6(){
        InetAddress address = getInetAddressIPV6();
        return address.getHostAddress();
    }

}

 

 

package org.com.utils;

import org.apache.ibatis.logging.LogException;

import java.net.Inet4Address;
import java.net.InetAddress;
import java.net.NetworkInterface;
import java.util.Date;
import java.util.Enumeration;
import java.util.Random;
import java.util.UUID;

public class IdUtils {

    /**
     * (目前最多支持32台机器。如需更多，分配动态数据中心，可支持32*32台机器)
     */
    //机器码范围只能是0-31  最多支持32台机器
    public static Integer machineId;
    //数据中心范围只能是0-31  最多支持32台机器，写死为0
    public static Integer datacenterId=0;

    private static String createMachineNumberUUId() {
        int machineId = 1;//最大支持1-9个集群机器部署
        int hashCodeV = UUID.randomUUID().toString().hashCode();
        if (hashCodeV < 0) {//有可能是负数
            hashCodeV = -hashCodeV;
        }
        // 0 代表前面补充0
        // 4 代表长度为4
        // d 代表参数为正数型
        return machineId + String.format("%015d", hashCodeV).replace("0", "");
    }

    public static Long getId() {
        return IdGenerator.getId();
    }

    public static Long getSerialNo() {
//        Long machineNumberUUId = Long.valueOf(createMachineNumberUUId());
//        Long machineId = machineNumberUUId % 32;
        SnowFlake snowFlake = new SnowFlake(datacenterId, machineId);
        return snowFlake.nextId();
    }

    /**
     * 生成quantity个digits位数随机码
     *
     * @param quantity 多少个
     * @param digits   多少位数
     */
    public static String[] randomCode(int quantity, int digits) {
        if (digits <= 0 || quantity <= 0) {
			throw new LogException("quantity or digits cannot be less than or equal to 0");
        }
        int bound = 10;
        for (int i = 1; i < digits; i++) {
            bound = bound * 10;
        }
        if (quantity > bound) {
			throw new LogException("random over limit");
        }
        String format = "%0digitsd".replace("digits", digits + "");
        Random myRandom = new Random();
        String[] result = new String[quantity];
        int[] luckNums = new int[quantity];
        //生成quantity个不同的随机数
        for (int i = 0; i < quantity; i++) {
            int randomNum = myRandom.nextInt(bound);
            while (existed(randomNum, luckNums, i)) {
                randomNum = myRandom.nextInt(bound);
            }
            luckNums[i] = randomNum;
            String randomCode = String.format(format, randomNum);
            result[i] = randomCode;
        }
        return result;
    }

    /**
     * 检验新生成的随机数是否已经存在
     *
     * @param num      新产生的随机数
     * @param luckNums 当前随机数数组
     * @param index    当前随机数的索引
     * @return boolean 是否存在，存在则返回true，不存在则返回false
     */
    public static boolean existed(int num, int[] luckNums, int index) {
        boolean contains = (num + "").contains("4");//去除带4随机码
        if (contains) {
            return true;
        }
        for (int i = 0; i < index; i++) {
            if (num == luckNums[i]) {
                return true;
            }
            //这个地方还可以增加一些处理，用来筛选不想要的随机码
        }
        return false;
    }

    /**
    public static void main(String[] args) throws Exception {
        getQuantityOrderNo(10000);
    }
     */

    //生成20位订单编号  snowFlake算法+2位随机码
    public static synchronized String getOrderSerialNo() {
        String orderSerialNo = getSerialNo() + randomCode(1, 2)[0];
        return orderSerialNo;
    }

    //生成20位订单编号  年月日时分+2位机器码(确定订单号是在哪台机器(IP)生成)+当前毫秒后5位+1位随机码
    //（年月日时分+当前毫秒后5位确定唯一,即使同分钟,毫秒一样，秒一定是不一样的。如秒和毫秒一样,分钟一定是不一样的）
    public static synchronized String getOrderNo() throws Exception {
        //处理同一毫秒下可能产生重复订单号情况(微秒、纳秒、皮秒、飞秒等)
        Thread.sleep(1);
        Date date = new Date();
        String time = date.getTime() + "";
        String digits5 = time.substring(time.length() - 5);
        String yyyyMMddHHmm = DateUtils.minuteFormat(new Date());
        String orderSerialNo = yyyyMMddHHmm + String.format("%02d", machineId) + digits5 + randomCode(1, 1)[0];
        return orderSerialNo;
    }

    //生成quantity个20位订单编号
    public static synchronized String[] getQuantityOrderSerialNo(int quantity) {
        String[] orderSerialNos = new String[quantity];
        String[] randomCode = randomCode(quantity, 2);
        for (int i = 0; i < randomCode.length; i++) {
            orderSerialNos[i] = getOrderSerialNo() + randomCode[i];
        }
        return orderSerialNos;
    }

    //生成quantity个20位订单编号
    public static synchronized String[] getQuantityOrderNo(int quantity) throws Exception {
        String[] orderSerialNos = new String[quantity];
        for (int i = 0; i < quantity; i++) {
            String orderNo = getOrderNo();
            orderSerialNos[i] = orderNo;
        }
        return orderSerialNos;
    }

    /**
     public static void main(String[] args) {
     Long machineNumberUUId=Long.valueOf(createMachineNumberUUId());
     Long machineId=machineNumberUUId%32;
     SnowFlake snowFlake=new SnowFlake(0L,machineId);
     return snowFlake.nextId();
     }
     */
}

 

 

package org.com.utils;

/**
 * @ClassName SnowFlake
 * @Author yupanpan
 * @Date 2020/11/10 17:24
 */
public class SnowFlake {

    /**
     * 起始的时间戳
     */
    private final static long START_STMP = 1480166465631L;

    /**
     * 每一部分占用的位数
     */
    private final static long SEQUENCE_BIT = 12; //序列号占用的位数
    private final static long MACHINE_BIT = 5;   //机器标识占用的位数
    private final static long DATACENTER_BIT = 5;//数据中心占用的位数

    /**
     * 每一部分的最大值
     */
    private final static long MAX_DATACENTER_NUM = -1L ^ (-1L << DATACENTER_BIT);
    private final static long MAX_MACHINE_NUM = -1L ^ (-1L << MACHINE_BIT);
    private final static long MAX_SEQUENCE = -1L ^ (-1L << SEQUENCE_BIT);

    /**
     * 每一部分向左的位移
     */
    private final static long MACHINE_LEFT = SEQUENCE_BIT;
    private final static long DATACENTER_LEFT = SEQUENCE_BIT + MACHINE_BIT;
    private final static long TIMESTMP_LEFT = DATACENTER_LEFT + DATACENTER_BIT;

    private long datacenterId;  //数据中心
    private long machineId;     //机器标识
    private long sequence = 0L; //序列号
    private long lastStmp = -1L;//上一次时间戳

    public SnowFlake(long datacenterId, long machineId) {
        if (datacenterId > MAX_DATACENTER_NUM || datacenterId < 0) {
            throw new IllegalArgumentException("datacenterId can't be greater than MAX_DATACENTER_NUM or less than 0");
        }
        if (machineId > MAX_MACHINE_NUM || machineId < 0) {
            throw new IllegalArgumentException("machineId can't be greater than MAX_MACHINE_NUM or less than 0");
        }
        this.datacenterId = datacenterId;
        this.machineId = machineId;
    }

    /**
     * 产生下一个ID
     *
     * @return
     */
    public synchronized long nextId() {
        long currStmp = getNewstmp();
        if (currStmp < lastStmp) {
            throw new RuntimeException("Clock moved backwards.  Refusing to generate id");
        }

        if (currStmp == lastStmp) {
            //相同毫秒内，序列号自增
            sequence = (sequence + 1) & MAX_SEQUENCE;
            //同一毫秒的序列数已经达到最大
            if (sequence == 0L) {
                currStmp = getNextMill();
            }
        } else {
            //不同毫秒内，序列号置为0
            sequence = 0L;
        }

        lastStmp = currStmp;

        return (currStmp - START_STMP) << TIMESTMP_LEFT //时间戳部分
                | datacenterId << DATACENTER_LEFT       //数据中心部分
                | machineId << MACHINE_LEFT             //机器标识部分
                | sequence;                             //序列号部分
    }

    private long getNextMill() {
        long mill = getNewstmp();
        while (mill <= lastStmp) {
            mill = getNewstmp();
        }
        return mill;
    }

    private long getNewstmp() {
        return System.currentTimeMillis();
    }

    /**
    public static void main(String[] args) {
        SnowFlake snowFlake = new SnowFlake(0, 0);

        for (int i = 0; i < (1 << 12); i++) {
            System.out.println(snowFlake.nextId());
        }

    }
     */

}