59 lines
1.6 KiB
Go
59 lines
1.6 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"log"
|
|
"net"
|
|
|
|
"omnisocketgo/cmd/internal/latencylog"
|
|
"omnisocketgo/cmd/internal/server"
|
|
"omnisocketgo/cmd/internal/transport"
|
|
)
|
|
|
|
func main() {
|
|
listenAddr := flag.String("listen", ":9001", "UDP server listen address")
|
|
logPath := flag.String("latency-log", "", "optional JSONL file path for latency timestamp logs")
|
|
txTimestampDebugLogPath := flag.String("tx-ts-debug-log", "", "optional JSONL file path for TX errqueue debug records")
|
|
flag.Parse()
|
|
|
|
hubOptions := make([]server.UDPOption, 0, 2)
|
|
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))
|
|
}
|
|
if *txTimestampDebugLogPath != "" {
|
|
logger, err := transport.NewJSONLTXTimestampDebugLogger(*txTimestampDebugLogPath)
|
|
if err != nil {
|
|
log.Fatalf("create tx timestamp debug logger %s: %v", *txTimestampDebugLogPath, err)
|
|
}
|
|
defer logger.Close()
|
|
hubOptions = append(hubOptions, server.WithUDPTXTimestampDebugLogger(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)
|
|
}
|
|
}
|