config.go

package rpcx

import (
	"fmt"
	"time"

	"github.com/meshplus/bitxhub-kit/crypto"
	"github.com/meshplus/bitxhub-kit/fileutil"
	"github.com/meshplus/bitxhub-kit/log"
)

const (
	blockChanNumber = 1024
	defaultTimeout  = 1 * time.Second
	defaultPoolSize = 4
)

type config struct {
	logger       Logger
	poolSize     int
	privateKey   crypto.PrivateKey
	nodesInfo    []*NodeInfo
	ipfsAddrs    []string
	timeoutLimit time.Duration // timeout limit config for dialing grpc
}

type NodeInfo struct {
	Addr       string
	EnableTLS  bool
	CertPath   string
	CommonName string
	AccessCert string
	AccessKey  string
}

type Option func(*config)

func WithNodesInfo(nodesInfo ...*NodeInfo) Option {
	return func(config *config) {
		config.nodesInfo = nodesInfo
	}
}

func WithLogger(logger Logger) Option {
	return func(config *config) {
		config.logger = logger
	}
}

func WithPrivateKey(key crypto.PrivateKey) Option {
	return func(config *config) {
		config.privateKey = key
	}
}

func WithIPFSInfo(addrs []string) Option {
	return func(config *config) {
		config.ipfsAddrs = addrs
	}
}

func WithTimeoutLimit(limit time.Duration) Option {
	return func(config *config) {
		config.timeoutLimit = limit
	}
}

func WithPoolSize(size int) Option {
	return func(config *config) {
		config.poolSize = size
	}
}

func generateConfig(opts ...Option) (*config, error) {
	config := &config{}
	for _, opt := range opts {
		opt(config)
	}

	if err := checkConfig(config); err != nil {
		return nil, err
	}

	return config, nil
}

func checkConfig(config *config) error {
	if config.privateKey == nil {
		return fmt.Errorf("private key is empty")
	}

	if len(config.nodesInfo) == 0 {
		return fmt.Errorf("bitxhub addrs cant not be 0")
	}

	if config.logger == nil {
		config.logger = log.NewWithModule("rpcx")
	}

	if config.timeoutLimit == 0 {
		config.timeoutLimit = defaultTimeout
	}

	if config.poolSize == 0 {
		config.poolSize = defaultPoolSize
	}

	// if EnableTLS is set, then tls certs must be provided
	for _, nodeInfo := range config.nodesInfo {
		if nodeInfo.EnableTLS {
			if !fileutil.Exist(nodeInfo.CertPath) {
				return fmt.Errorf("ca cert file %s is not found while tls is enabled", nodeInfo.CertPath)
			}
			if !fileutil.Exist(nodeInfo.AccessCert) {
				return fmt.Errorf("access cert file %s is not found while tls is enabled", nodeInfo.AccessCert)
			}
			if !fileutil.Exist(nodeInfo.AccessKey) {
				return fmt.Errorf("access key file %s is not found while tls is enabled", nodeInfo.AccessKey)
			}
		}
	}
	return nil
}