🎨 update snowflake node generate

itmisx · Dec 23, 2020 · 303926b · 303926b
1 parent 8afb796
commit 303926b
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/internal/pkg/helper.go b/internal/pkg/helper.go
@@ -19,7 +19,12 @@ var Node *snowflake.Node
 func init() {
 	var err error
 	rand.Seed(time.Now().UnixNano())
-	Node, err = snowflake.NewNode(int64(rand.Intn(1024)))
+	if DsfConfig.Tracker.NodeID > 0 {
+		Node, err = snowflake.NewNode(DsfConfig.Tracker.NodeID)
+	} else {
+		Node, err = snowflake.NewNode(int64(rand.Intn(1024)))
+	}
+
 	if err != nil {
 		panic("")
 	}