From b0cb481ead077d6ec7441a94d26889bb54cfdb3a Mon Sep 17 00:00:00 2001
From: Dustin Pianalto <dustin@djpianalto.com>
Date: Fri, 9 Jul 2021 01:15:45 -0800
Subject: [PATCH] Get worker id from env

---
 cmd/snowflake/main.go | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/cmd/snowflake/main.go b/cmd/snowflake/main.go
index 38c2ec0..d9ab8a1 100644
--- a/cmd/snowflake/main.go
+++ b/cmd/snowflake/main.go
@@ -1,6 +1,9 @@
 package main
 
 import (
+	"log"
+	"os"
+	"strconv"
 	"sync"
 
 	"github.com/dustinpianalto/snowflake/internal/generator"
@@ -10,7 +13,12 @@ import (
 
 func main() {
 
-	generator.CreateGenerator(1)
+	workerID, err := strconv.ParseUint(os.Getenv("WORKER_ID"), 10, 64)
+	if err != nil {
+		log.Fatal("Not a valid worker id")
+	}
+
+	generator.CreateGenerator(workerID)
 
 	go generator.Generator.Run()