DEMOMySql/MicroServices/Business/Business.Core/Utilities/SnowFlake.cs

using Amazon.Runtime.Internal.Util;
using System.Net;
using System.Runtime.InteropServices;

namespace Business.Core.Utilities
{
    /// <summary>
    /// 雪花算法生成id
    /// </summary>
    public class SnowFlake
    {
        /// <summary>
        /// 机器id所占的位数
        /// </summary>
        private const int workerIdBits = 5;

        /// <summary>
        /// 数据表述id所占位数
        /// </summary>
        private const int datacenterIdBits = 5;

        /// <summary>
        /// 支持的最大机器id，结果是31
        /// </summary>
        private const long maxWorkerId = -1L ^ (-1L << workerIdBits);

        /// <summary>
        /// 支持的最大数据表示id，结果是31
        /// </summary>
        private const long maxDatacenterId = -1L ^ (-1L << datacenterIdBits);

        /// <summary>
        /// 序列在id中所占的位数
        /// </summary>
        private const int sequenceBits = 12;

        /// <summary>
        /// 数据标识id向左移17位
        /// </summary>
        private const int datacenterIdShift = sequenceBits + workerIdBits;

        /// <summary>
        /// 机器ID向左移12位
        /// </summary>
        private const int workerIdShift = sequenceBits;

        private const int timestampLeftShift = sequenceBits + workerIdBits + datacenterIdBits;

        /// <summary>
        /// 生成序列的掩码
        /// </summary>
        private const long sequenceMask = -1L ^ (-1L << sequenceBits);

        /// <summary>
        /// 数据中心ID(0-31)
        /// </summary>
        public long DatacenterId { get; private set; }

        /// <summary>
        /// 工作机器ID(0-31)
        /// </summary>
        public long WorkerId { get; private set; }

        /// <summary>
        /// 毫秒内序列(0-4095)
        /// </summary>
        public long Sequence { get; private set; }

        /// <summary>
        /// 上次生成ID的时间戳
        /// </summary>
        public long LastTimestamp { get; private set; }

        /// <summary>
        /// 开始时间戳。首次使用前设置，否则无效，默认2010-1-1
        /// </summary>
        public DateTime StartTimestamp { get; set; } = new DateTime(2010,1,1);

        static object syncRoot = new object();
        static readonly Lazy<SnowFlake> snowflake = new(() => new SnowFlake());

        /// <summary>
        /// 默认静态实例，WorkerId = 0,DatacenterId = 0
        /// </summary>
        public static SnowFlake Instance { get; } = snowflake.Value;

        /// <summary>
        /// 构造函数
        /// </summary>
        /// <exception cref="ArgumentOutOfRangeException"></exception>
        public SnowFlake()
        {
            //生成workerid和datacenterid
            SetWorkIdAndDatacenterId();
            if (WorkerId > maxWorkerId || WorkerId < 0)
            {
                throw new ArgumentOutOfRangeException(nameof(WorkerId),$"不能大于{maxWorkerId}或小于0");
            }
            if (DatacenterId > maxDatacenterId || DatacenterId < 0)
            {
                throw new ArgumentOutOfRangeException(nameof(DatacenterId), $"不能大于{maxDatacenterId}或小于0");
            }
            Sequence = 0L;
            LastTimestamp = -1L;
        }

        /// <summary>
        /// 生成WorkerId和DatacenterId
        /// </summary>
        /// <returns></returns>
        private void SetWorkIdAndDatacenterId()
        {
            string? ipAdress = "";
            try
            {
                string? hostName;
                if (RuntimeInformation.IsOSPlatform(OSPlatform.Linux))
                {
                    hostName = Environment.GetEnvironmentVariable("MY_NODE_NAME");
                    ipAdress = Environment.GetEnvironmentVariable("MY_HOST_IP");
                }
                else
                {
                    hostName = Dns.GetHostName();
                    IPHostEntry ipEntity = Dns.GetHostEntry(hostName);
                    for (int i = 0; i < ipEntity.AddressList.Length; i++)
                    {
                        if (ipEntity.AddressList[i].AddressFamily == System.Net.Sockets.AddressFamily.InterNetwork)
                        {
                            ipAdress = ipEntity.AddressList[i].ToString();
                        }
                    }
                }
                if (string.IsNullOrEmpty(ipAdress) || string.IsNullOrEmpty(hostName))
                {
                    WorkerId = 0;
                    DatacenterId = 0;
                }
                else
                {
                    WorkerId = StringToLong(ipAdress);
                    DatacenterId = StringToLong(hostName);
                }
            }
            catch (Exception)
            {
                WorkerId = 0;
                DatacenterId = 0;
            }
        }

        /// <summary>
        /// 生成workerid和datacenterid
        /// </summary>
        /// <param name=""></param>
        /// <returns></returns>
        private long StringToLong(string strInput)
        {
            int[] arrays = ToCodePoints(strInput);
            int sum = 0;
            for (int i = 0; i < arrays.Length; i++)
            {
                sum += arrays[i];
            }
            return (long)(sum % 32);
        }

        /// <summary>
        /// 将字符串转换成数组
        /// </summary>
        /// <param name="strInput"></param>
        /// <returns></returns>
        private int[] ToCodePoints(string strInput)
        {
            var codePoints = new List<int>(strInput.Length);
            for (int i = 0; i < strInput.Length; i++)
            {
                codePoints.Add(char.ConvertToUtf32(strInput, i));
                if (char.IsHighSurrogate(strInput[i]))
                {
                    i += 1;
                }
            }
            return codePoints.ToArray();
        }

        /// <summary>
        /// 获取下一个ID
        /// </summary>
        /// <returns></returns>
        public long NextId()
        {
            lock (syncRoot)
            {
                //获取当前时间戳
                long timestamp = GetCurrentTimestamp();
                if (timestamp > LastTimestamp)//时间戳改变，毫秒内序列重置
                {
                    Sequence = 0L;
                }
                else if (timestamp == LastTimestamp)//如果是同一时间生成的，则进行毫秒内序列
                {
                    Sequence = (Sequence + 1) & sequenceMask;
                    if (Sequence == 0)//毫秒内序列溢出
                    {
                        timestamp = GetNextTimestamp(LastTimestamp);//阻塞到下一个毫秒，获取新的时间戳
                    }
                }
                else//当前时间小于上一次ID生成的时间戳，证明系统时钟被回拨，此时需要做回拨处理
                {
                    Sequence = (Sequence + 1) & sequenceMask;
                    if (Sequence > 0)
                    {
                        timestamp = LastTimestamp;//停留再最后一次时间戳上，等待系统时间追上后即完全度过了时钟回拨问题
                    }
                    else {
                        timestamp = LastTimestamp + 1;//直接进位到下一个毫秒
                    }
                }
                LastTimestamp = timestamp;//上次生成ID的时间戳

                //移动并通过或运算拼到一起组成64位的ID
                var id = (timestamp << timestampLeftShift)
                        | (DatacenterId << datacenterIdShift)
                        | (WorkerId << workerIdShift)
                        | Sequence;
                return id;
            }
        }

        /// <summary>
        /// 阻塞到下一个毫秒，直到获得新的时间戳
        /// </summary>
        /// <param name="lastTimestamp"></param>
        /// <returns></returns>
        private long GetNextTimestamp(long lastTimestamp)
        {
            //获取当前时间戳
            long timestamp = GetCurrentTimestamp();
            while (timestamp <= lastTimestamp)
            { 
                timestamp = GetCurrentTimestamp();
            }
            return timestamp;
        }

        /// <summary>
        /// 获取当前时间戳
        /// </summary>
        /// <returns></returns>
        private long GetCurrentTimestamp()
        {
            return (long)(DateTime.Now - StartTimestamp).TotalMilliseconds;
        }
    }
}