feat(benchmarks/rps): Add unix support.

Author: jochumdev

benchmarks/rps/cmd/orb-rps-client/config.go

@@ -9,14 +9,13 @@ import (
 const (
 	configSection = "benchClient"
 
-	defaultBypassRegistry = 1
-	defaultConnections    = 256
-	defaultPoolSize       = 256
-	defaultDuration       = 15
-	defaultTimeout        = 15
-	defaultTransport      = "grpc"
-	defaultPackageSize    = 1000
-	defaultContentType    = "application/x-protobuf"
+	defaultConnections = 256
+	defaultPoolSize    = 256
+	defaultDuration    = 15
+	defaultTimeout     = 15
+	defaultTransport   = "grpc"
+	defaultPackageSize = 1000
+	defaultContentType = "application/x-protobuf"
 )
 
 //nolint:gochecknoglobals
@@ -27,14 +26,6 @@ var (
 func flags() []*cli.Flag {
 	var flags []*cli.Flag
 
-	flags = append(flags, cli.NewFlag(
-		"bypass_registry",
-		defaultBypassRegistry,
-		cli.FlagConfigPaths([]string{configSection, "bypassRegistry"}),
-		cli.FlagUsage("Bypasses the registry by caching it, set to 0 to disable"),
-		cli.FlagEnvVars("BYPASS_REGISTRY"),
-	))
-
 	flags = append(flags, cli.NewFlag(
 		"pool_size",
 		defaultPoolSize,
@@ -103,13 +94,12 @@ func flags() []*cli.Flag {
 }
 
 type clientConfig struct {
-	BypassRegistry int    `json:"bypassRegistry" yaml:"bypassRegistry"`
-	PoolSize       int    `json:"poolSize"       yaml:"poolSize"`
-	Connections    int    `json:"connections"    yaml:"connections"`
-	Duration       int    `json:"duration"       yaml:"duration"`
-	Timeout        int    `json:"timeout"        yaml:"timeout"`
-	Threads        int    `json:"threads"        yaml:"threads"`
-	Transport      string `json:"transport"      yaml:"transport"`
-	PackageSize    int    `json:"packageSize"    yaml:"packageSize"`
-	ContentType    string `json:"contentType"    yaml:"contentType"`
+	PoolSize    int    `json:"poolSize"       yaml:"poolSize"`
+	Connections int    `json:"connections"    yaml:"connections"`
+	Duration    int    `json:"duration"       yaml:"duration"`
+	Timeout     int    `json:"timeout"        yaml:"timeout"`
+	Threads     int    `json:"threads"        yaml:"threads"`
+	Transport   string `json:"transport"      yaml:"transport"`
+	PackageSize int    `json:"packageSize"    yaml:"packageSize"`
+	ContentType string `json:"contentType"    yaml:"contentType"`
 }

benchmarks/rps/cmd/orb-rps-client/main.go

@@ -101,21 +101,6 @@ func setupClientOptions(ctx context.Context, cfg *clientConfig, cli client.Type,
 		client.WithContentType(cfg.ContentType),
 	}
 
-	// Handle direct connection (bypass registry)
-	if cfg.BypassRegistry == 1 {
-		logger.Debug("Resolving service", "server", serverName)
-
-		// Resolve service nodes
-		address, _, err := cli.SelectService(ctx, serverName, client.WithPreferredTransports(cfg.Transport))
-		if err != nil {
-			return nil, fmt.Errorf("failed to resolve service: %w", err)
-		}
-
-		// Add direct URL to options
-		opts = append(opts, client.WithURL(fmt.Sprintf("%s://%s", cfg.Transport, address)))
-		logger.Info("Using transport", "transport", cfg.Transport)
-	}
-
 	return opts, nil
 }
 
@@ -157,7 +142,6 @@ func runBenchmark(
 func bench(ctx context.Context, cfg *clientConfig, logger log.Logger, cli client.Type) error {
 	// Log configuration
 	logger.Info("Configuration",
-		"bypass_registry", cfg.BypassRegistry,
 		"connections", cfg.Connections,
 		"duration", cfg.Duration,
 		"timeout", cfg.Timeout,

benchmarks/rps/cmd/orb-rps-client/wire.go

@@ -70,15 +70,14 @@ func wireRun(
 
 func provideClientConfig(svcCtx *cli.ServiceContextWithConfig) (*clientConfig, error) {
 	cfg := &clientConfig{
-		BypassRegistry: defaultBypassRegistry,
-		PoolSize:       defaultPoolSize,
-		Connections:    defaultConnections,
-		Duration:       defaultDuration,
-		Timeout:        defaultTimeout,
-		Threads:        defaultThreads,
-		Transport:      defaultTransport,
-		PackageSize:    defaultPackageSize,
-		ContentType:    defaultContentType,
+		PoolSize:    defaultPoolSize,
+		Connections: defaultConnections,
+		Duration:    defaultDuration,
+		Timeout:     defaultTimeout,
+		Threads:     defaultThreads,
+		Transport:   defaultTransport,
+		PackageSize: defaultPackageSize,
+		ContentType: defaultContentType,
 	}
 
 	if err := config.Parse(nil, configSection, svcCtx.Config(), &cfg); err != nil && !errors.Is(err, config.ErrNoSuchKey) {

benchmarks/rps/cmd/orb-rps-client/wire_gen.go

@@ -65,6 +65,21 @@ func provideServerOpts() ([]server.ConfigOption, error) {
 	opts = append(opts, server.WithEntrypointConfig("drpc", drpc.NewConfig(
 		drpc.WithHandlers(hRegister),
 	)))
+	opts = append(opts, server.WithEntrypointConfig("unix+drpc", drpc.NewConfig(
+		drpc.WithNetwork("unix"),
+		drpc.WithAddress("/tmp/orb-rps-server-drpc.sock"),
+		drpc.WithHandlers(hRegister),
+	)))
+	opts = append(opts, server.WithEntrypointConfig("unix+grpc", mgrpc.NewConfig(
+		mgrpc.WithNetwork("unix"),
+		mgrpc.WithAddress("/tmp/orb-rps-server-grpc.sock"),
+		mgrpc.WithHandlers(hRegister),
+	)))
+	opts = append(opts, server.WithEntrypointConfig("unix+http", mhttp.NewConfig(
+		mhttp.WithNetwork("unix"),
+		mhttp.WithAddress("/tmp/orb-rps-server-http.sock"),
+		mhttp.WithHandlers(hRegister),
+	)))
 
 	return opts, nil
 }