Feat: UDP 框架

cmd/udpserver/main.go

@@ -0,0 +1,48 @@
+package main
+
+import (
+	"flag"
+	"log"
+	"net"
+
+	"omnisocketgo/cmd/internal/latencylog"
+	"omnisocketgo/cmd/internal/server"
+)
+
+func main() {
+	listenAddr := flag.String("listen", ":9001", "UDP server listen address")
+	logPath := flag.String("latency-log", "", "optional JSONL file path for latency timestamp logs")
+	flag.Parse()
+
+	hubOptions := make([]server.UDPOption, 0, 1)
+	if *logPath != "" {
+		logger, err := latencylog.NewJSONLLogger(*logPath)
+		if err != nil {
+			log.Fatalf("create latency logger %s: %v", *logPath, err)
+		}
+		defer logger.Close()
+		hubOptions = append(hubOptions, server.WithUDPLogger(logger))
+	}
+
+	udpAddr, err := net.ResolveUDPAddr("udp", *listenAddr)
+	if err != nil {
+		log.Fatalf("resolve udp address %s: %v", *listenAddr, err)
+	}
+
+	conn, err := net.ListenUDP("udp", udpAddr)
+	if err != nil {
+		log.Fatalf("listen udp on %s: %v", *listenAddr, err)
+	}
+	defer conn.Close()
+
+	hub, err := server.NewUDPHub(conn, hubOptions...)
+	if err != nil {
+		log.Fatalf("create udp hub: %v", err)
+	}
+
+	log.Printf("udp server listening on %s", conn.LocalAddr())
+
+	if err := hub.Serve(); err != nil {
+		log.Fatalf("udp hub serve: %v", err)
+	}
+}