fix: update code for new go SDK

Author: raksiv

src/pages/guides/go/realtime-messaging.mdx

@@ -77,28 +77,31 @@ import (
   "context"
   "fmt"
 
-  "github.com/nitrictech/go-sdk/handler"
   "github.com/nitrictech/go-sdk/nitric"
+  "github.com/nitrictech/go-sdk/nitric/keyvalue"
+  "github.com/nitrictech/go-sdk/nitric/websockets"
 )
 
 func main() {
-  ws, err := nitric.NewWebsocket("public")
-  if err != nil {
-    fmt.Println("Error creating WebSocket:", err)
-    return
-  }
+  // Create a WebSocket endpoint named "public".
+  ws := nitric.NewWebsocket("public")
 
-  connections, err := nitric.NewKv("connections").Allow(nitric.KvStoreGet, nitric.KvStoreSet, nitric.KvStoreDelete)
-  if err != nil {
-    fmt.Println("Error creating KV store:", err)
-    return
-  }
+  // Initialize a KV store named "connections" with Get, Set, and Delete permissions.
+  connections := nitric.NewKv("connections").Allow(keyvalue.KvStoreGet, keyvalue.KvStoreSet, keyvalue.KvStoreDelete)
+
+  // Handle new WebSocket connections by storing the connection ID in the KV store.
+  ws.On(websockets.EventType_Connect, func(ctx *websockets.Ctx) {
+    err := connections.Set(context.TODO(), ctx.Request.ConnectionID(), map[string]interface{}{
+      "connectionId": ctx.Request.ConnectionID(),
+    })
+    if err != nil {
+      return
+    }
+  })
 
   // Add event handlers here
 
-  if err := nitric.Run(); err != nil {
-    fmt.Println("Error running Nitric service:", err)
-  }
+  nitric.Run()
 }
 ```
 
@@ -117,40 +120,35 @@ From here, let's add some features to that function that allow us to manage conn
 ### Register connections on connect
 
 ```go
-ws.On(handler.WebsocketConnect, func(ctx *handler.WebsocketContext, next handler.WebsocketHandler) (*handler.WebsocketContext, error) {
+ws.On(websockets.EventType_Connect, func(ctx *websockets.Ctx) {
   err := connections.Set(context.TODO(), ctx.Request.ConnectionID(), map[string]interface{}{
     "connectionId": ctx.Request.ConnectionID(),
   })
   if err != nil {
-    return ctx, err
+    return
   }
-
-  return next(ctx)
 })
 ```
 
 ### Remove connections on disconnect
 
 ```go
-ws.On(handler.WebsocketDisconnect, func(ctx *handler.WebsocketContext, next handler.WebsocketHandler) (*handler.WebsocketContext, error) {
+ws.On(websockets.EventType_Disconnect, func(ctx *websockets.Ctx) {
   err := connections.Delete(context.TODO(), ctx.Request.ConnectionID())
   if err != nil {
-    return ctx, err
+    return
   }
-
-  return next(ctx)
 })
 ```
 
 ### Broadcast messages to all connected clients
 
 ```go
-ws.On(handler.WebsocketMessage, func(ctx *handler.WebsocketContext, next handler.WebsocketHandler) (*handler.WebsocketContext, error) {
+ws.On(websockets.EventType_Message, func(ctx *websockets.Ctx) {
   connectionStream, err := connections.Keys(context.TODO())
   if err != nil {
-    return ctx, err
+    return
   }
-
   senderId := ctx.Request.ConnectionID()
 
   for {
@@ -166,11 +164,9 @@ ws.On(handler.WebsocketMessage, func(ctx *handler.WebsocketContext, next handler
     message := fmt.Sprintf("%s: %s", senderId, ctx.Request.Message())
     err = ws.Send(context.TODO(), connectionId, []byte(message))
     if err != nil {
-      return ctx, err
+      return
     }
   }
-
-  return next(ctx)
 })
 ```
 
@@ -183,77 +179,74 @@ ws.On(handler.WebsocketMessage, func(ctx *handler.WebsocketContext, next handler
 package main
 
 import (
-	"context"
-	"fmt"
+  "context"
+  "fmt"
 
-	"github.com/nitrictech/go-sdk/handler"
-	"github.com/nitrictech/go-sdk/nitric"
+  "github.com/nitrictech/go-sdk/nitric"
+  "github.com/nitrictech/go-sdk/nitric/keyvalue"
+  "github.com/nitrictech/go-sdk/nitric/websockets"
 )
 
 func main() {
-	ws, err := nitric.NewWebsocket("public")
-	if err != nil {
-		fmt.Println("Error creating WebSocket:", err)
-		return
-	}
-
-	connections, err := nitric.NewKv("connections").Allow(nitric.KvStoreGet, nitric.KvStoreSet, nitric.KvStoreDelete)
-	if err != nil {
-		fmt.Println("Error creating KV store:", err)
-		return
-	}
-
-	ws.On(handler.WebsocketConnect, func(ctx *handler.WebsocketContext, next handler.WebsocketHandler) (*handler.WebsocketContext, error) {
-		err := connections.Set(context.TODO(), ctx.Request.ConnectionID(), map[string]interface{}{
-			"connectionId": ctx.Request.ConnectionID(),
-		})
-		if err != nil {
-			return ctx, err
-		}
-
-		return next(ctx)
-	})
-
-	ws.On(handler.WebsocketDisconnect, func(ctx *handler.WebsocketContext, next handler.WebsocketHandler) (*handler.WebsocketContext, error) {
-		err := connections.Delete(context.TODO(), ctx.Request.ConnectionID())
-		if err != nil {
-			return ctx, err
-		}
-
-		return next(ctx)
-	})
-
-	ws.On(handler.WebsocketMessage, func(ctx *handler.WebsocketContext, next handler.WebsocketHandler) (*handler.WebsocketContext, error) {
-		connectionStream, err := connections.Keys(context.TODO())
-		if err != nil {
-			return ctx, err
-		}
-
-		senderId := ctx.Request.ConnectionID()
-
-		for {
-			connectionId, err := connectionStream.Recv()
-			if err != nil {
-				break
-			}
-
-			if connectionId == senderId {
-				continue
-			}
-
-			message := fmt.Sprintf("%s: %s", senderId, ctx.Request.Message())
-			err = ws.Send(context.TODO(), connectionId, []byte(message))
-			if err != nil {
-				return ctx, err
-			}
-		}
-
-		return next(ctx)
-	})
-
-	if err := nitric.Run(); err != nil {
-		fmt.Println("Error running Nitric service:", err)
-	}
+  // Create a WebSocket endpoint named "public".
+  ws := nitric.NewWebsocket("public")
+
+  // Initialize a KV store named "connections" with Get, Set, and Delete permissions.
+  connections := nitric.NewKv("connections").Allow(keyvalue.KvStoreGet, keyvalue.KvStoreSet, keyvalue.KvStoreDelete)
+
+  // Handle new WebSocket connections by storing the connection ID in the KV store.
+  ws.On(websockets.EventType_Connect, func(ctx *websockets.Ctx) {
+    err := connections.Set(context.TODO(), ctx.Request.ConnectionID(), map[string]interface{}{
+      "connectionId": ctx.Request.ConnectionID(),
+    })
+    if err != nil {
+      return
+    }
+  })
+
+  ws.On(websockets.EventType_Connect, func(ctx *websockets.Ctx) {
+    err := connections.Set(context.TODO(), ctx.Request.ConnectionID(), map[string]interface{}{
+      "connectionId": ctx.Request.ConnectionID(),
+    })
+    if err != nil {
+      return
+    }
+  })
+
+  ws.On(websockets.EventType_Disconnect, func(ctx *websockets.Ctx) {
+    err := connections.Delete(context.TODO(), ctx.Request.ConnectionID())
+    if err != nil {
+      return
+    }
+  })
+
+  ws.On(websockets.EventType_Message, func(ctx *websockets.Ctx) {
+    connectionStream, err := connections.Keys(context.TODO())
+    if err != nil {
+      return
+    }
+
+    senderId := ctx.Request.ConnectionID()
+
+    for {
+      connectionId, err := connectionStream.Recv()
+      if err != nil {
+        break
+      }
+
+      if connectionId == senderId {
+        continue
+      }
+
+      message := fmt.Sprintf("%s: %s", senderId, ctx.Request.Message())
+      err = ws.Send(context.TODO(), connectionId, []byte(message))
+      if err != nil {
+        return
+      }
+    }
+  })
+
+  nitric.Run()
 }
 ```