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
類,它包含了 Twepoch
、WorkerIdBits
、DatacenterIdBits
、SequenceBits
等常量,用于計算 ID 的各個部分。同時,我們還定義了一些私有變量,如 _sequence
、_lastTimestamp
、_workerId
和 _datacenterId
,用于存儲當前的序列號、最后一次生成 ID 的時間戳、工作節點 ID 和數據中心 ID。
Snowflake
類的構造函數接收兩個參數,分別是工作節點 ID 和數據中心 ID,并進行合法性檢查。NextId
方法用于生成下一個 ID,它首先獲取當前的時間戳,然后根據時間戳、工作節點 ID、數據中心 ID 和序列號計算出一個新的 ID。如果當前時間戳小于上一次生成 ID 的時間戳,說明系統時鐘回撥,此時需要等待下一毫秒再生成 ID。
GetCurrentTimestamp
方法用于獲取當前的時間戳(毫秒級),WaitNextMillis
方法用于等待下一毫秒。
這個實現是線程安全的,因為我們使用了 lock
關鍵字來確保在生成 ID 時不會被其他線程打斷。