中文字幕av专区_日韩电影在线播放_精品国产精品久久一区免费式_av在线免费观看网站

溫馨提示×

C# Snowflake算法的源碼解析

c#
小樊
85
2024-09-02 12:46:21
欄目: 編程語言

Snowflake 算法是 Twitter 開源的一種分布式 ID 生成策略,它可以在不依賴數據庫或其他存儲設備的情況下生成全局唯一的 ID。Snowflake 算法的 ID 結構包括時間戳、數據中心 ID、機器 ID 和序列號等部分。

以下是一個簡單的 C# 實現 Snowflake 算法的示例代碼:

using System;
using System.Threading;

public class Snowflake
{
    private const long Twepoch = 1288834974657L;
    private const int WorkerIdBits = 5;
    private const int DatacenterIdBits = 5;
    private const int SequenceBits = 12;
    private const long MaxWorkerId = -1L ^ (-1L<< WorkerIdBits);
    private const long MaxDatacenterId = -1L ^ (-1L<< DatacenterIdBits);
    private const int WorkerIdShift = SequenceBits;
    private const int DatacenterIdShift = SequenceBits + WorkerIdBits;
    private const int TimestampLeftShift = SequenceBits + WorkerIdBits + DatacenterIdBits;
    private const long SequenceMask = -1L ^ (-1L << SequenceBits);

    private static long _sequence = 0L;
    private static long _lastTimestamp = -1L;

    private readonly long _workerId;
    private readonly long _datacenterId;
    private readonly object _lock = new object();

    public Snowflake(long workerId, long datacenterId)
    {
        if (workerId > MaxWorkerId || workerId < 0)
        {
            throw new ArgumentException($"Worker Id can't be greater than {MaxWorkerId} or less than 0");
        }

        if (datacenterId > MaxDatacenterId || datacenterId < 0)
        {
            throw new ArgumentException($"Datacenter Id can't be greater than {MaxDatacenterId} or less than 0");
        }

        _workerId = workerId;
        _datacenterId = datacenterId;
    }

    public long NextId()
    {
        lock (_lock)
        {
            var timestamp = GetCurrentTimestamp();

            if (timestamp > _lastTimestamp)
            {
                _sequence = 0;
                _lastTimestamp = timestamp;
            }
            else
            {
                _sequence = (_sequence + 1) & SequenceMask;

                if (_sequence == 0)
                {
                    timestamp = WaitNextMillis(_lastTimestamp);
                    _lastTimestamp = timestamp;
                }
            }

            return ((timestamp - Twepoch)<< TimestampLeftShift) |
                   (_datacenterId<< DatacenterIdShift) |
                   (_workerId<< WorkerIdShift) |
                   _sequence;
        }
    }

    private long GetCurrentTimestamp()
    {
        return (long)(DateTime.UtcNow - new DateTime(1970, 1, 1, 0, 0, 0, DateTimeKind.Utc)).TotalMilliseconds;
    }

    private long WaitNextMillis(long lastTimestamp)
    {
        var timestamp = GetCurrentTimestamp();

        while (timestamp <= lastTimestamp)
        {
            Thread.Sleep(1);
            timestamp = GetCurrentTimestamp();
        }

        return timestamp;
    }
}

這個實現中,我們定義了一個 Snowflake 類,它包含了 TwepochWorkerIdBitsDatacenterIdBitsSequenceBits 等常量,用于計算 ID 的各個部分。同時,我們還定義了一些私有變量,如 _sequence_lastTimestamp_workerId_datacenterId,用于存儲當前的序列號、最后一次生成 ID 的時間戳、工作節點 ID 和數據中心 ID。

Snowflake 類的構造函數接收兩個參數,分別是工作節點 ID 和數據中心 ID,并進行合法性檢查。NextId 方法用于生成下一個 ID,它首先獲取當前的時間戳,然后根據時間戳、工作節點 ID、數據中心 ID 和序列號計算出一個新的 ID。如果當前時間戳小于上一次生成 ID 的時間戳,說明系統時鐘回撥,此時需要等待下一毫秒再生成 ID。

GetCurrentTimestamp 方法用于獲取當前的時間戳(毫秒級),WaitNextMillis 方法用于等待下一毫秒。

這個實現是線程安全的,因為我們使用了 lock 關鍵字來確保在生成 ID 時不會被其他線程打斷。

0
静安区| 波密县| 武安市| 深圳市| 定西市| 山东省| 阿勒泰市| 镇雄县| 苏尼特右旗| 大安市| 皮山县| 平和县| 鲁甸县| 肃宁县| 逊克县| 钦州市| 凤庆县| 永清县| 庆元县| 曲沃县| 鄂托克前旗| 临武县| 南平市| 岱山县| 繁峙县| 毕节市| 公安县| 呼伦贝尔市| 博野县| 绥阳县| 凌海市| 万山特区| 武冈市| 柳河县| 开江县| 乐清市| 青海省| 睢宁县| 平阴县| 贵阳市| 江口县|