fix: adding monitoring

v1/examples/server/main.go

@@ -1,8 +1,12 @@
 package main
 
 import (
+	"context"
 	"encoding/json"
 	"log"
+	"os"
+	"os/signal"
+	"syscall"
 	"time"
 
 	"git.neurocipta.com/rogerferdinan/safe-web-socket/v1/server"
@@ -14,34 +18,66 @@ type ExampleData struct {
 }
 
 func main() {
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
+	sigChan := make(chan os.Signal, 1)
+	signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM)
+	go func() {
+		<-sigChan
+		log.Println("Received shutdown signal")
+		cancel()
+	}()
+
 	s, err := server.NewSafeWebsocketServerBuilder().
 		BaseHost("localhost").
 		BasePort(8080).
 		ApiKey("abcd").
-		HandleFuncWebsocket("/ws/test/", "data_1", func(c chan []byte) {
+		Context(ctx).
+		HandleFuncWebsocket("/ws/test/", "data_1", 5_000, func(ctx context.Context, c chan []byte) {
 			ticker := time.NewTicker(100 * time.Millisecond)
-			for range ticker.C {
-				jsonBytes, err := json.Marshal(ExampleData{
-					Time: time.Now(),
-					Data: "data_1",
-				})
-				if err != nil {
-					continue
+			defer ticker.Stop()
+			for {
+				select {
+				case <-ctx.Done():
+					return
+				case <-ticker.C:
+					jsonBytes, err := json.Marshal(ExampleData{
+						Time: time.Now(),
+						Data: "data_1",
+					})
+					if err != nil {
+						continue
+					}
+					select {
+					case c <- jsonBytes:
+					case <-ctx.Done():
+						return
+					}
 				}
-				c <- jsonBytes
 			}
 		}).
-		HandleFuncWebsocket("/ws/test/", "data_2", func(c chan []byte) {
+		HandleFuncWebsocket("/ws/test/", "data_2", 5_000, func(ctx context.Context, c chan []byte) {
 			ticker := time.NewTicker(100 * time.Millisecond)
-			for range ticker.C {
-				jsonBytes, err := json.Marshal(ExampleData{
-					Time: time.Now(),
-					Data: "data_2",
-				})
-				if err != nil {
-					continue
+			defer ticker.Stop()
+			for {
+				select {
+				case <-ctx.Done():
+					return
+				case <-ticker.C:
+					jsonBytes, err := json.Marshal(ExampleData{
+						Time: time.Now(),
+						Data: "data_2",
+					})
+					if err != nil {
+						continue
+					}
+					select {
+					case c <- jsonBytes:
+					case <-ctx.Done():
+						return
+					}
 				}
-				c <- jsonBytes
 			}
 		}).
 		Build()