continuous send for simulators

sim.go

@@ -4,10 +4,14 @@ package main
 
 import (
 	"encoding/json"
+	"fmt"
 	"log"
+	"math/rand"
 	"net/url"
 	"os"
 	"os/signal"
+	"sync"
+	"sync/atomic"
 	"syscall"
 	"time"
 
@@ -22,6 +26,12 @@ const (
 	channelID = "123456789"
 )
 
+// send interval range (random between minInterval and maxInterval)
+var (
+	minInterval = 500 * time.Millisecond
+	maxInterval = 5 * time.Second
+)
+
 type Handshake struct {
 	Type string          `json:"type"`
 	Data json.RawMessage `json:"data"`
@@ -59,6 +69,8 @@ type GatewayMessageIn struct {
 }
 
 func main() {
+	rand.Seed(time.Now().UnixNano())
+
 	interrupt := make(chan os.Signal, 1)
 	signal.Notify(interrupt, os.Interrupt, syscall.SIGTERM)
 
@@ -80,19 +92,19 @@ func main() {
 
 	log.Println("Connected to gateway")
 
-	// respond to pings (server ping -> client must pong). Using SetPingHandler is fine,
-	// but WriteControl for Pong is acceptable too.
+	var writeMu sync.Mutex
+
 	conn.SetPingHandler(func(appData string) error {
 		log.Println("Received ping from server, sending pong")
-		err := conn.WriteControl(websocket.PongMessage, []byte(appData), time.Now().Add(5*time.Second))
-		if err != nil {
+		writeMu.Lock()
+		defer writeMu.Unlock()
+		if err := conn.WriteControl(websocket.PongMessage, []byte(appData), time.Now().Add(5*time.Second)); err != nil {
 			log.Printf("Failed to send pong: %v", err)
 			return err
 		}
 		return nil
 	})
 
-	// Build and send handshake including channel id
 	modHS := ModHandshake{
 		ServerID:  serverID,
 		ChannelID: channelID,
@@ -107,12 +119,14 @@ func main() {
 		Data: modHSData,
 	}
 
+	writeMu.Lock()
 	if err := conn.WriteJSON(handshake); err != nil {
+		writeMu.Unlock()
 		log.Fatalf("Failed to send handshake: %v", err)
 	}
+	writeMu.Unlock()
 	log.Println("Handshake sent")
 
-	// Read acknowledgment (some servers might not reply with JSON; handle errors)
 	var ack GatewayAck
 	if err := conn.ReadJSON(&ack); err != nil {
 		log.Fatalf("Failed to read acknowledgment: %v", err)
@@ -129,62 +143,115 @@ func main() {
 				if websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway, websocket.CloseNormalClosure) {
 					log.Printf("WebSocket error: %v", err)
 				} else {
-					log.Printf("Connection closed: %v", err)
+					log.Printf("Connection closed/read error: %v", err)
 				}
 				return
 			}
-			if messageType == websocket.TextMessage || messageType == websocket.BinaryMessage {
+			switch messageType {
+			case websocket.TextMessage, websocket.BinaryMessage:
 				log.Printf("Received from server: %s", string(message))
+			default:
 			}
 		}
 	}()
 
-	// Optional: send a test message after connecting
-	time.Sleep(1 * time.Second)
-	testMsg := GatewayMessageIn{
-		MsgID: "test-msg-001",
-		ID:    serverID,
-		Destination: Destination{
-			ID: channelID,
-		},
-		Author: User{
-			ID:   "player-uuid-123",
-			Name: "TestPlayer",
-		},
-		Content: "Hello from simulated mod!",
-		Ts:      time.Now().UTC(),
-	}
+	var msgCounter uint64 = 1
 
-	if err := conn.WriteJSON(testMsg); err != nil {
-		log.Printf("Failed to send test message: %v", err)
-	} else {
-		log.Println("Sent test message to gateway")
-	}
+	func() {
+		testMsg := GatewayMessageIn{
+			MsgID: fmt.Sprintf("test-msg-%06d", atomic.AddUint64(&msgCounter, 1)),
+			ID:    serverID,
+			Destination: Destination{
+				ID: channelID,
+			},
+			Author: User{
+				ID:   "player-uuid-123",
+				Name: "TestPlayer",
+			},
+			Content: "Hello from simulated mod!",
+			Ts:      time.Now().UTC(),
+		}
 
-	log.Println("Connection established. Responding to pings. Press Ctrl+C to disconnect.")
+		writeMu.Lock()
+		if err := conn.WriteJSON(testMsg); err != nil {
+			log.Printf("Failed to send test message: %v", err)
+		} else {
+			log.Println("Sent initial test message to gateway")
+		}
+		writeMu.Unlock()
+	}()
+
+	go func() {
+		for {
+			d := randomDuration(minInterval, maxInterval)
+
+			select {
+			case <-done:
+				return
+			case <-time.After(d):
+				// build message
+				msgNum := atomic.AddUint64(&msgCounter, 1)
+				msg := GatewayMessageIn{
+					MsgID: fmt.Sprintf("sim-msg-%06d", msgNum),
+					ID:    serverID,
+					Destination: Destination{
+						ID: channelID,
+					},
+					Author: User{
+						ID:   fmt.Sprintf("sim-user-%d", msgNum%1000),
+						Name: fmt.Sprintf("SimUser%d", msgNum%1000),
+					},
+					Content: fmt.Sprintf("Random interval message #%d (delay %s)", msgNum, d),
+					Ts:      time.Now().UTC(),
+				}
+
+				writeMu.Lock()
+				if err := conn.WriteJSON(msg); err != nil {
+					log.Printf("Failed to send simulated message: %v", err)
+					writeMu.Unlock()
+					return
+				}
+				writeMu.Unlock()
+
+				log.Printf("Sent simulated message %s (next wait up to %s)", msg.MsgID, maxInterval)
+			}
+		}
+	}()
+
+	log.Println("Connection established. Sending simulated messages at random intervals. Press Ctrl+C to disconnect.")
 
 	for {
 		select {
 		case <-done:
-			log.Println("Connection closed")
+			log.Println("Connection read loop closed, exiting")
 			return
 		case <-interrupt:
 			log.Println("Interrupt received, closing connection...")
 
+			writeMu.Lock()
 			err := conn.WriteMessage(
 				websocket.CloseMessage,
 				websocket.FormatCloseMessage(websocket.CloseNormalClosure, ""),
 			)
+			writeMu.Unlock()
 			if err != nil {
 				log.Printf("Write close error: %v", err)
-				return
 			}
 
 			select {
 			case <-done:
-			case <-time.After(time.Second):
+			case <-time.After(1 * time.Second):
 			}
 			return
 		}
 	}
 }
+
+func randomDuration(min, max time.Duration) time.Duration {
+	if max <= min {
+		return min
+	}
+	diff := int64(max - min)
+	n := rand.Int63n(diff)
+	return min + time.Duration(n)
+}