Snowflak算法 (C#版本)

最近看了twitter的分佈式自增ID算法Snowflak,本身根據開源的代碼寫了一個C#的版本,以記錄學習。 twitter的開源項目地址爲:https://github.com/twitter/snowflake 用Scala實現git


class IdWorker
    {
        //機器標識位數
        private const int WORKER_ID_BITS = 4;
        //機器標識位的最大值
        private const long MAX_WORKER_ID = -1L ^ -1L << WORKER_ID_BITS;
        //毫秒內自增位
        private const int SEQUENCE_BITS = 10;
        //自增位最大值
        private const long SEQUENCE_Mask = -1L ^ -1L << SEQUENCE_BITS;
        private const long twepoch = 1398049504651L;
        //時間毫秒值向左偏移位
        private const int timestampLeftShift = SEQUENCE_BITS + WORKER_ID_BITS;
        //機器標識位向左偏移位
        private const int WORKER_ID_SHIFT = SEQUENCE_BITS;
        private static readonly DateTime START_TIME = new System.DateTime(1970, 1, 1, 0, 0, 0, 0);
        private static readonly object LOCK = new object();
        private long sequence = 0L;
        private long lastTimestamp = -1L;
        private long workerId;
        public IdWorker(long workerId)
        {
            if (workerId > MAX_WORKER_ID || workerId < 0)
            {
                throw new ArgumentException(string.Format("worker id can't be greater than {0} or less than 0", MAX_WORKER_ID));
            }
            this.workerId = workerId;
        }
        /// <summary>
        /// 獲取下一個id值
        /// </summary>
        /// <returns></returns>
        public long NextId()
        {
            lock (LOCK)
            {
                long timestamp = TimeGen();
                //當前時間小於上一次時間,錯誤
                if (timestamp < this.lastTimestamp)
                {
                    throw new Exception(string.Format("Clock moved backwards.  Refusing to generate id for %d milliseconds", lastTimestamp - timestamp));
                }
                //當前毫秒內
                if (this.lastTimestamp == timestamp)
                {
                    //+1 求餘
                    this.sequence = (this.sequence + 1) & SEQUENCE_Mask;
                    //當前毫秒內計數滿了,等待下一秒
                    if (this.sequence == 0)
                    {
                        timestamp = tilNextMillis(lastTimestamp);
                    }
                }
                else      //不是當前毫秒內
                {
                    this.sequence = 0;   //重置當前毫秒計數
                }
                this.lastTimestamp = timestamp;
                //當前毫秒值 | 機器標識值 | 當前毫秒內自增值
                long nextId = ((timestamp - twepoch << timestampLeftShift))
                | (this.workerId << WORKER_ID_SHIFT) | (this.sequence);
                return nextId;
            }
        }
        /// <summary>
        /// 等待下一個毫秒
        /// </summary>
        /// <param name="lastTimestamp"></param>
        /// <returns></returns>
        private long tilNextMillis(long lastTimestamp)
        {
            long timestamp = TimeGen();
            while (timestamp <= lastTimestamp)
            {
                timestamp = TimeGen();
            }
            return timestamp;
        }
        /// <summary>
        /// 獲取當前時間的Unix時間戳
        /// </summary>
        /// <returns></returns>
        private long TimeGen()
        {
            return (DateTime.UtcNow.Ticks - START_TIME.Ticks) / 10000;
        }
    }


調用方式以下github

static void Main(string[] args)
{
    IdWorker worker = new IdWorker(12L);
    for (int i = 0; i < 100000; i++)
    {
        Console.WriteLine(worker.NextId());
    }
}
相關文章
相關標籤/搜索