support nkeys

une-tasse-de-cafe · Sep 17, 2024 · 48031d4 · 48031d4
1 parent d65df8b
commit 48031d4
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 7 deletions.
diff --git a/controller/main.go b/controller/main.go
@@ -42,10 +42,19 @@ func main() {
 	coffeeQuantityMap["medium"] = 17
 	coffeeQuantityMap["large"] = 30
 
-	nc, _ := nats.Connect(url)
+	opt, err := nats.NkeyOptionFromSeed("seed.txt")
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	nc, err := nats.Connect(os.Getenv("NATS_URL"), opt)
+	if err != nil {
+		log.Fatal("connect to nats: ", err)
+	}
+
 	defer nc.Drain()
 
-	sub, _ := nc.Subscribe(subjectSub, func(msg *nats.Msg) {
+	sub, err := nc.Subscribe(subjectSub, func(msg *nats.Msg) {
 
 		var order CoffeeOrder
 
@@ -99,6 +108,9 @@ func main() {
 		log.Printf("🗣️ Status: %s, Message: %s\n", response.Status, response.Message)
 		msg.Respond(jsonData)
 	})
+	if err != nil {
+		log.Fatal("connect to nats: ", err)
+	}
 
 	defer sub.Unsubscribe()
 

diff --git a/routes/main.go b/routes/main.go
@@ -134,12 +134,16 @@ func handleHome(w http.ResponseWriter, numberOfPendingOrders int) {
 
 func main() {
 
-	url := os.Getenv("NATS_URL")
-	nc, err := nats.Connect(url)
+	opt, err := nats.NkeyOptionFromSeed("seed.txt")
 	if err != nil {
 		log.Fatal(err)
 	}
 
+	nc, err = nats.Connect(os.Getenv("NATS_URL"), opt)
+	if err != nil {
+		log.Fatal("connect to nats: ", err)
+	}
+
 	defer nc.Drain()
 
 	js, err := jetstream.New(nc)
@@ -169,7 +173,6 @@ func main() {
 		} else {
 			numberOfPendingOrders, err = strconv.Atoi(string(kvValuePendingOrders.Value()))
 			if err != nil {
-
 				numberOfPendingOrders = -1
 			}
 		}

diff --git a/stock-manager/main.go b/stock-manager/main.go
@@ -72,9 +72,14 @@ func main() {
 		url = "192.168.128.51:4222"
 	}
 
-	nc, err := nats.Connect(url)
+	opt, err := nats.NkeyOptionFromSeed("seed.txt")
 	if err != nil {
-		log.Fatal("Failed to connect to NATS:", err)
+		log.Fatal(err)
+	}
+
+	nc, err := nats.Connect(os.Getenv("NATS_URL"), opt)
+	if err != nil {
+		log.Fatal("connect to nats: ", err)
 	}
 
 	defer nc.Close()