Go 雪花算法生成唯一ID（Snowflake）&分布式

在使用 Go 语言实现的雪花算法（Snowflake）时，每个分布式系统中的节点通常都会有一个独立的 Node 实例。这个 Node 实例会根据节点的配置生成唯一的ID。

以下是几个关键点：

1. 节点唯一性：每个节点（或服务实例）都应该有一个唯一的工作节点ID（worker ID）。这是雪花算法保证全局唯一性的关键。

2. 独立 Node 实例 ：在实际部署中，每个节点应该创建自己的 Node 实例，并为其分配一个唯一的工作节点ID。

3. 不要共享 Node 实例 ：尽管雪花算法是线程安全的，可以在同一节点内被多个 Goroutine 安全地调用，但不同节点之间不应该共享同一个 Node 实例。

4. 配置工作节点ID ：在创建 Node 实例时，你需要为每个节点配置一个不同的工作节点ID。这个ID通常在节点启动时配置，并在节点的整个生命周期内保持不变。

package main

import (
	"fmt"
	"github.com/bwmarrin/snowflake"
	"sync"
)

func main() {
	// 假设有5个服务实例，每个实例有一个唯一的节点ID
	nodeIDs := []int64{1, 2, 3, 4, 5}
	var wg sync.WaitGroup

	// 并发生成ID的函数
	generateIDs := func(nodeID int64) {
		defer wg.Done()
		node, err := snowflake.NewNode(nodeID)
		if err != nil {
			fmt.Printf("创建ID为的节点时出错%d: %v\n", nodeID, err)
			return
		}

		for i := 0; i < 200; i++ { // 每个节点生成10个ID
			id := node.Generate().Int64()
			fmt.Println(id)
			// 模拟工作负载
			//time.Sleep(10 * time.Millisecond)
		}
	}

	// 为每个服务实例创建一个Goroutine
	for _, nodeID := range nodeIDs {
		wg.Add(1)
		go generateIDs(nodeID)
	}

	// 等待所有Goroutine完成
	wg.Wait()
}

package main

import (
	"fmt"
	"sync"
	"time"

	"github.com/bwmarrin/snowflake"
)

func main() {
	var wg sync.WaitGroup

	// 模拟多个节点
	nodes := []int64{1, 2, 3, 4, 5} // 节点ID
	numIDs := 200                   // 每个节点生成的ID数量

	// 启动多个goroutines模拟多个节点
	for _, nodeID := range nodes {
		wg.Add(1)
		go func(nid int64) {
			defer wg.Done()
			node, err := snowflake.NewNode(nid)
			if err != nil {
				fmt.Println("创建节点时出错:", err)
				return
			}

			for i := 0; i < numIDs; i++ {
				id := node.Generate().Int64()
				fmt.Println(id) // 模拟工作负载
				time.Sleep(1 * time.Millisecond)
			}
		}(nodeID)
	}

	// 等待所有goroutines完成
	wg.Wait()
}

package main

import (
	"fmt"
	"github.com/bwmarrin/snowflake"
	"log"
	"sync"
)

func GenerateSnowflakeID(wg *sync.WaitGroup, i int) {
	defer wg.Done()
	// 需要设置节点的机器ID
	node, err := snowflake.NewNode(int64(i))
	if err != nil {
		log.Fatal(err)
	}
	for i := 0; i < 100; i++ {
		fmt.Println(node.Generate().Int64())
	}

}
func main() {
	var wg sync.WaitGroup
	for i := 0; i < 20; i++ {
		wg.Add(1)
		go GenerateSnowflakeID(&wg, i)
	}
	wg.Wait()
}

注意事项

• 确保每个服务实例的节点ID是唯一的，不要重复。
• 雪花算法的Node实例是线程安全的，所以你可以在同一实例中并发地生成ID，而不必担心线程安全问题。
• 在实际部署中，节点ID通常会通过配置文件、环境变量或配置管理系统来分配和获取