feat(go): add 100k ingest hardening baseline

go/vehicle-gateway/internal/loadsim/config.go

@@ -0,0 +1,82 @@
+package loadsim
+
+import (
+	"errors"
+	"flag"
+	"fmt"
+	"strings"
+	"time"
+)
+
+type Protocol string
+
+const (
+	ProtocolGB32960    Protocol = "gb32960"
+	ProtocolJT808      Protocol = "jt808"
+	ProtocolYutongMQTT Protocol = "yutong-mqtt"
+)
+
+type Config struct {
+	Protocol             Protocol
+	Addr                 string
+	Connections          int
+	ConnectRatePerSecond int
+	SendInterval         time.Duration
+	Duration             time.Duration
+	Template             string
+}
+
+type FlagConfig struct {
+	protocol     string
+	addr         string
+	connections  int
+	connectRate  int
+	sendInterval time.Duration
+	duration     time.Duration
+	template     string
+}
+
+func RegisterFlags(fs *flag.FlagSet) *FlagConfig {
+	cfg := &FlagConfig{}
+	fs.StringVar(&cfg.protocol, "protocol", string(ProtocolJT808), "protocol: jt808, gb32960, yutong-mqtt")
+	fs.StringVar(&cfg.addr, "addr", "", "target host:port")
+	fs.IntVar(&cfg.connections, "connections", 1, "target concurrent connections")
+	fs.IntVar(&cfg.connectRate, "connect-rate", 100, "new connections per second")
+	fs.DurationVar(&cfg.sendInterval, "send-interval", 10*time.Second, "frame send interval per connection")
+	fs.DurationVar(&cfg.duration, "duration", 10*time.Minute, "load test duration")
+	fs.StringVar(&cfg.template, "template", "", "frame template name")
+	return cfg
+}
+
+func (c *FlagConfig) Build() (Config, error) {
+	protocol := Protocol(strings.ToLower(strings.TrimSpace(c.protocol)))
+	switch protocol {
+	case ProtocolGB32960, ProtocolJT808, ProtocolYutongMQTT:
+	default:
+		return Config{}, fmt.Errorf("unsupported protocol %q", c.protocol)
+	}
+	if strings.TrimSpace(c.addr) == "" {
+		return Config{}, errors.New("addr is required")
+	}
+	if c.connections <= 0 {
+		return Config{}, errors.New("connections must be positive")
+	}
+	if c.connectRate <= 0 {
+		return Config{}, errors.New("connect-rate must be positive")
+	}
+	if c.sendInterval <= 0 {
+		return Config{}, errors.New("send-interval must be positive")
+	}
+	if c.duration <= 0 {
+		return Config{}, errors.New("duration must be positive")
+	}
+	return Config{
+		Protocol:             protocol,
+		Addr:                 strings.TrimSpace(c.addr),
+		Connections:          c.connections,
+		ConnectRatePerSecond: c.connectRate,
+		SendInterval:         c.sendInterval,
+		Duration:             c.duration,
+		Template:             strings.TrimSpace(c.template),
+	}, nil
+}