fix: update code for new go SDK

Author: raksiv

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

@@ -74,31 +74,34 @@ Let's begin by setting up the WebSocket application. First, create a new folder
 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
-  }
+	// 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)
 })
 ```
 
@@ -186,8 +182,8 @@ import (
 	"context"
 	"fmt"
 
-	"github.com/nitrictech/go-sdk/handler"
 	"github.com/nitrictech/go-sdk/nitric"
+	"github.com/nitrictech/go-sdk/nitric/websockets"
 )
 
 func main() {
@@ -203,30 +199,26 @@ func main() {
 		return
 	}
 
-	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)
 	})
 
-	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)
 	})
 
-	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()
@@ -244,16 +236,12 @@ func main() {
 			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)
 	})
 
-	if err := nitric.Run(); err != nil {
-		fmt.Println("Error running Nitric service:", err)
-	}
+	nitric.Run()
 }
 ```