Update main.yml

.github/workflows/main.yml

@@ -20,30 +20,35 @@ jobs:
       - name: Get Version
         id: get_version
         uses: battila7/get-version-action@v2.0.0
-
-      - name: Build container image
-        env:
-          IMAGE_TAG: ${{ steps.get_version.outputs.version-without-v }}
-        run: docker build -t registry.digitalocean.com/djpianalto/snowflake-server:$IMAGE_TAG .
-
-      - name: Install doctl
-        uses: digitalocean/action-doctl@v2
+      
+      - name: Docker Login
+        # You may pin to the exact commit or the version.
+        # uses: docker/login-action@f054a8b539a109f9f41c372932f1ae047eff08c9
+        uses: docker/login-action@v1.10.0
         with:
-          token: ${{ secrets.DIGITALOCEAN_ACCESS_TOKEN }}
+          registry: ${{ secrets.DR_URL }}
+          # Username used to log against the Docker registry
+          username: ${{ secrets.DH_USERNAME }}
+          # Password or personal access token used to log against the Docker registry
+          password: ${{ secrets.DH_PASSWORD }}
+          # Log out from the Docker registry at the end of a job
+          logout: true
 
-      - name: Login to DigitalOcean Container Registry with short-lived credentials
-        run: doctl registry login --expiry-seconds 600
-
-      - name: Push image to DigitalOcean Container Registry
+      - name: Docker Build & Push
         env:
           IMAGE_TAG: ${{ steps.get_version.outputs.version-without-v }}
-        run: docker push registry.digitalocean.com/djpianalto/snowflake-server:$IMAGE_TAG
+        run: |
+          docker build -t ${{ secrets.DR_URL }}/snowflake-server:$IMAGE_TAG .
+          docker push ${{ secrets.DR_URL }}/snowflake-server:$IMAGE_TAG
+          sed -i 's|<IMAGE>|${{ secrets.DR_URL }}/snowflake-server:'${IMAGE_TAG}'|' $GITHUB_WORKSPACE/deployment.yml
 
-      - name: Update deployment file
-        run: TAG=${{ steps.get_version.outputs.version-without-v }} && sed -i 's|<IMAGE>|registry.digitalocean.com/djpianalto/snowflake-server:'${TAG}'|' $GITHUB_WORKSPACE/deployment.yml
-
-      - name: Save DigitalOcean kubeconfig with short-lived credentials
-        run: doctl kubernetes cluster kubeconfig save --expiry-seconds 600 discord-bots
-
-      - name: Deploy to DigitalOcean Kubernetes
+      - name: Kubernetes set context
+        uses: Azure/k8s-set-context@v1.1
+        with:
+          # Acceptable values: kubeconfig or service-account or service-principal
+          method: kubeconfig
+          # Kubernetes Config
+          kubeconfig: ${{ secrets.KUBE_CONFIG }}
+          
+      - name: Deploy to Kubernetes
         run: kubectl apply -f $GITHUB_WORKSPACE/deployment.yml

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

@@ -2,7 +2,7 @@ kind: Service
 apiVersion: v1
 metadata:
   name: snowflake-server
-  namespace: dca
+  namespace: default
   labels:
     environment: production
 spec:
@@ -25,7 +25,7 @@ apiVersion: apps/v1
 kind: Deployment
 metadata:
   name: snowflake-server
-  namespace: dca
+  namespace: default
   labels:
     app: snowflake-server
 spec: