connection

type ClientConfig struct {
	Endpoint *Endpoint     `mapstructure:"endpoint"  yaml:"endpoint"`
	TLS      TLSConfig     `mapstructure:"tls"       yaml:"tls"`
	Retry    *RetryProfile `mapstructure:"reconnect" yaml:"reconnect"`
}

type ClientParameters struct {
	Address        string
	Creds          credentials.TransportCredentials
	Retry          *RetryProfile
	AdditionalOpts []grpc.DialOption
}

type Endpoint struct {
	Host string `mapstructure:"host" json:"host,omitempty" yaml:"host,omitempty"`
	Port int    `mapstructure:"port" json:"port,omitempty" yaml:"port,omitempty"`
}

type MultiClientConfig struct {
	Endpoints []*Endpoint   `mapstructure:"endpoints" yaml:"endpoints"`
	TLS       TLSConfig     `mapstructure:"tls"       yaml:"tls"`
	Retry     *RetryProfile `mapstructure:"reconnect" yaml:"reconnect"`
}

type RateLimitConfig struct {
	// RequestsPerSecond is the maximum number of requests per second allowed.
	// Set to 0 or negative to disable rate limiting.
	RequestsPerSecond int `mapstructure:"requests-per-second"`
	// Burst is the maximum number of requests allowed in a single burst.
	// This allows handling sudden spikes of concurrent requests from multiple clients.
	// Must be greater than 0 and less than or equal to RequestsPerSecond when rate limiting is enabled.
	Burst int `mapstructure:"burst"`
}

type RetryProfile struct {
	InitialInterval     time.Duration `mapstructure:"initial-interval" yaml:"initial-interval"`
	RandomizationFactor float64       `mapstructure:"randomization-factor" yaml:"randomization-factor"`
	Multiplier          float64       `mapstructure:"multiplier" yaml:"multiplier"`
	MaxInterval         time.Duration `mapstructure:"max-interval" yaml:"max-interval"`
	// After MaxElapsedTime the ExponentialBackOff returns RetryStopDuration.
	// It never stops if MaxElapsedTime == 0.
	MaxElapsedTime time.Duration `mapstructure:"max-elapsed-time" yaml:"max-elapsed-time"`
}

type ServerConfig struct {
	Endpoint  Endpoint               `mapstructure:"endpoint"`
	TLS       TLSConfig              `mapstructure:"tls"`
	KeepAlive *ServerKeepAliveConfig `mapstructure:"keep-alive"`
	RateLimit RateLimitConfig        `mapstructure:"rate-limit"`
	// contains filtered or unexported fields
}

type ServerKeepAliveConfig struct {
	Params            *ServerKeepAliveParamsConfig            `mapstructure:"params"`
	EnforcementPolicy *ServerKeepAliveEnforcementPolicyConfig `mapstructure:"enforcement-policy"`
}

type ServerKeepAliveEnforcementPolicyConfig struct {
	MinTime             time.Duration `mapstructure:"min-time"`
	PermitWithoutStream bool          `mapstructure:"permit-without-stream"`
}

type ServerKeepAliveParamsConfig struct {
	MaxConnectionIdle     time.Duration `mapstructure:"max-connection-idle"`
	MaxConnectionAge      time.Duration `mapstructure:"max-connection-age"`
	MaxConnectionAgeGrace time.Duration `mapstructure:"max-connection-age-grace"`
	Time                  time.Duration `mapstructure:"time"`
	Timeout               time.Duration `mapstructure:"timeout"`
}

type Service interface {
	// Run executes the service until the context is done.
	Run(ctx context.Context) error
	// WaitForReady waits for the service resources to initialize.
	// If the context ended before the service is ready, returns false.
	WaitForReady(ctx context.Context) bool
	// RegisterService registers the supported APIs for this service.
	RegisterService(server *grpc.Server)
}

type TLSConfig struct {
	Mode string `mapstructure:"mode"`
	// CertPath is the path to the certificate file (public key).
	CertPath string `mapstructure:"cert-path"`
	// KeyPath is the path to the key file (private key).
	KeyPath     string   `mapstructure:"key-path"`
	CACertPaths []string `mapstructure:"ca-cert-paths"`
}

type TLSMaterials struct {
	Mode    string
	Cert    []byte
	Key     []byte
	CACerts [][]byte
}

type WithAddress interface {
	Address() string
}