config.go

package rpc

import "fmt"

// 'localhost' gets interpreted as ipv6
// TODO: revisit this
const localhost = "127.0.0.1"

type RPCConfig struct {
	Info     *ServerConfig  `json:",omitempty" toml:",omitempty"`
	Profiler *ServerConfig  `json:",omitempty" toml:",omitempty"`
	GRPC     *ServerConfig  `json:",omitempty" toml:",omitempty"`
	Metrics  *MetricsConfig `json:",omitempty" toml:",omitempty"`
}

type ServerConfig struct {
	Enabled       bool
	ListenAddress string
}

type MetricsConfig struct {
	ServerConfig
	MetricsPath     string
	BlockSampleSize int
}

func DefaultRPCConfig() *RPCConfig {
	return &RPCConfig{
		Info:     DefaultInfoConfig(),
		Profiler: DefaultProfilerConfig(),
		GRPC:     DefaultGRPCConfig(),
		Metrics:  DefaultMetricsConfig(),
	}
}

func DefaultInfoConfig() *ServerConfig {
	return &ServerConfig{
		Enabled:       true,
		ListenAddress: fmt.Sprintf("tcp://%s:26658", localhost),
	}
}

func DefaultGRPCConfig() *ServerConfig {
	return &ServerConfig{
		Enabled:       true,
		ListenAddress: fmt.Sprintf("tcp://%s:10997", localhost),
	}
}

func DefaultProfilerConfig() *ServerConfig {
	return &ServerConfig{
		Enabled:       false,
		ListenAddress: fmt.Sprintf("tcp://%s:6060", localhost),
	}
}

func DefaultMetricsConfig() *MetricsConfig {
	return &MetricsConfig{
		ServerConfig: ServerConfig{
			Enabled:       false,
			ListenAddress: fmt.Sprintf("tcp://%s:9102", localhost),
		},
		MetricsPath:     "/metrics",
		BlockSampleSize: 100,
	}
}