Get worker id from env

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()
 

deployment.yml

@@ -21,30 +21,37 @@ status:
 
 ---
 
-apiVersion: v1
+apiVersion: apps/v1
-apiVersion: batch/v1beta1
+kind: Deployment
-kind: CronJob
 metadata:
   name: snowflake-server
-  namespace: default
+  namespace: dca
   labels:
     app: snowflake-server
 spec:
-  schedule: "0 0 * * *"
+  replicas: 1
-  jobTemplate:
+  selector:
+    matchLabels:
+      app: snowflake-server
+  strategy:
+    rollingUpdate:
+      maxSurge: 1
+      maxUnavailable: 1
+  minReadySeconds: 30
+  template:
+    metadata:
+      labels:
+        app: snowflake-server
     spec:
-      template:
+      containers:
-        spec:
+        - name: snowflake-server
-          restartPolicy: "OnFailure"
+          image: <IMAGE>
-          containers:
+          ports:
-            - name: snowflake-server
+            - containerPort: 50051
-              image: <IMAGE>
+            - containerPort: 50052
-              ports:
+          env:
-                - containerPort: 50051
+            - name: WORKER_ID
-                - containerPort: 50052
+              valueFrom:
-              env:
+                secretKeyRef:
-                - name: WORKER_ID
+                  name: snowflake
-                  valueFrom:
+                  key: worker_id
-                    secretKeyRef:
-                      name: snowflake
-                      key: worker_id