forked from TrueCloudLab/frostfs-node
8060735732
Generalize single gRPC interface of the storage node to a group of interfaces. Each interface calls the same RPC handler. Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
100 lines
2.2 KiB
Go
100 lines
2.2 KiB
Go
package grpcconfig
|
|
|
|
import (
|
|
"errors"
|
|
"strconv"
|
|
|
|
"github.com/nspcc-dev/neofs-node/cmd/neofs-node/config"
|
|
)
|
|
|
|
var (
|
|
errEndpointNotSet = errors.New("empty/not set endpoint, see `grpc.endpoint` section")
|
|
errTLSKeyNotSet = errors.New("empty/not set TLS key file path, see `grpc.tls.key` section")
|
|
errTLSCertNotSet = errors.New("empty/not set TLS certificate file path, see `grpc.tls.certificate` section")
|
|
)
|
|
|
|
// Config is a wrapper over the config section
|
|
// which provides access to gRPC server configurations.
|
|
type Config config.Config
|
|
|
|
// Endpoint returns value of "endpoint" config parameter.
|
|
//
|
|
// Panics if value is not a non-empty string.
|
|
func (x *Config) Endpoint() string {
|
|
v := config.StringSafe(
|
|
(*config.Config)(x),
|
|
"endpoint")
|
|
if v == "" {
|
|
panic(errEndpointNotSet)
|
|
}
|
|
|
|
return v
|
|
}
|
|
|
|
// TLS returns "tls" subsection as a TLSConfig.
|
|
//
|
|
// Returns nil if "enabled" value of "tls" subsection is false.
|
|
func (x *Config) TLS() *TLSConfig {
|
|
sub := (*config.Config)(x).
|
|
Sub("tls")
|
|
|
|
if !config.BoolSafe(sub, "enabled") {
|
|
return nil
|
|
}
|
|
|
|
return &TLSConfig{
|
|
cfg: sub,
|
|
}
|
|
}
|
|
|
|
// TLSConfig is a wrapper over the config section
|
|
// which provides access to TLS configurations
|
|
// of the gRPC server.
|
|
type TLSConfig struct {
|
|
cfg *config.Config
|
|
}
|
|
|
|
// KeyFile returns value of "key" config parameter.
|
|
//
|
|
// Panics if value is not a non-empty string.
|
|
func (tls TLSConfig) KeyFile() string {
|
|
v := config.StringSafe(tls.cfg, "key")
|
|
if v == "" {
|
|
panic(errTLSKeyNotSet)
|
|
}
|
|
|
|
return v
|
|
}
|
|
|
|
// CertificateFile returns value of "certificate" config parameter.
|
|
//
|
|
// Panics if value is not a non-empty string.
|
|
func (tls TLSConfig) CertificateFile() string {
|
|
v := config.StringSafe(tls.cfg, "certificate")
|
|
if v == "" {
|
|
panic(errTLSCertNotSet)
|
|
}
|
|
|
|
return v
|
|
}
|
|
|
|
// IterateEndpoints iterates over subsections ["0":"N") (N - "num" value)
|
|
// of "grpc" section of c, wrap them into Config and passes to f.
|
|
//
|
|
// Panics if N is not a positive number.
|
|
func IterateEndpoints(c *config.Config, f func(*Config)) {
|
|
c = c.Sub("grpc")
|
|
|
|
num := config.Uint(c, "num")
|
|
if num == 0 {
|
|
panic("no gRPC server configured")
|
|
}
|
|
|
|
for i := uint64(0); i < num; i++ {
|
|
si := strconv.FormatUint(i, 10)
|
|
|
|
sc := (*Config)(c.Sub(si))
|
|
|
|
f(sc)
|
|
}
|
|
}
|