136 lines
2.7 KiB
Go
136 lines
2.7 KiB
Go
package order
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/meshplus/bitxhub-kit/crypto"
|
|
"github.com/meshplus/bitxhub-kit/types"
|
|
"github.com/meshplus/bitxhub-model/pb"
|
|
"github.com/meshplus/bitxhub/pkg/peermgr"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
type Config struct {
|
|
ID uint64
|
|
IsNew bool
|
|
RepoRoot string
|
|
StoragePath string
|
|
PluginPath string
|
|
PeerMgr peermgr.PeerManager
|
|
PrivKey crypto.PrivateKey
|
|
Logger logrus.FieldLogger
|
|
Nodes map[uint64]*pb.VpInfo
|
|
Applied uint64
|
|
Digest string
|
|
GetTransactionFunc func(hash *types.Hash) (*pb.Transaction, error)
|
|
GetChainMetaFunc func() *pb.ChainMeta
|
|
GetBlockByHeight func(height uint64) (*pb.Block, error)
|
|
}
|
|
|
|
type Option func(*Config)
|
|
|
|
func WithID(id uint64) Option {
|
|
return func(config *Config) {
|
|
config.ID = id
|
|
}
|
|
}
|
|
|
|
func WithIsNew(isNew bool) Option {
|
|
return func(config *Config) {
|
|
config.IsNew = isNew
|
|
}
|
|
}
|
|
|
|
func WithRepoRoot(path string) Option {
|
|
return func(config *Config) {
|
|
config.RepoRoot = path
|
|
}
|
|
}
|
|
|
|
func WithStoragePath(path string) Option {
|
|
return func(config *Config) {
|
|
config.StoragePath = path
|
|
}
|
|
}
|
|
|
|
func WithPluginPath(path string) Option {
|
|
return func(config *Config) {
|
|
config.PluginPath = path
|
|
}
|
|
}
|
|
|
|
func WithPeerManager(peerMgr peermgr.PeerManager) Option {
|
|
return func(config *Config) {
|
|
config.PeerMgr = peerMgr
|
|
}
|
|
}
|
|
|
|
func WithPrivKey(privKey crypto.PrivateKey) Option {
|
|
return func(config *Config) {
|
|
config.PrivKey = privKey
|
|
}
|
|
}
|
|
|
|
func WithLogger(logger logrus.FieldLogger) Option {
|
|
return func(config *Config) {
|
|
config.Logger = logger
|
|
}
|
|
}
|
|
|
|
func WithNodes(nodes map[uint64]*pb.VpInfo) Option {
|
|
return func(config *Config) {
|
|
config.Nodes = nodes
|
|
}
|
|
}
|
|
|
|
func WithApplied(height uint64) Option {
|
|
return func(config *Config) {
|
|
config.Applied = height
|
|
}
|
|
}
|
|
|
|
func WithDigest(digest string) Option {
|
|
return func(config *Config) {
|
|
config.Digest = digest
|
|
}
|
|
}
|
|
|
|
func WithGetChainMetaFunc(f func() *pb.ChainMeta) Option {
|
|
return func(config *Config) {
|
|
config.GetChainMetaFunc = f
|
|
}
|
|
}
|
|
|
|
func WithGetTransactionFunc(f func(hash *types.Hash) (*pb.Transaction, error)) Option {
|
|
return func(config *Config) {
|
|
config.GetTransactionFunc = f
|
|
}
|
|
}
|
|
|
|
func WithGetBlockByHeightFunc(f func(height uint64) (*pb.Block, error)) Option {
|
|
return func(config *Config) {
|
|
config.GetBlockByHeight = f
|
|
}
|
|
}
|
|
|
|
func checkConfig(config *Config) error {
|
|
if config.Logger == nil {
|
|
return fmt.Errorf("logger is nil")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func GenerateConfig(opts ...Option) (*Config, error) {
|
|
config := &Config{}
|
|
for _, opt := range opts {
|
|
opt(config)
|
|
}
|
|
|
|
if err := checkConfig(config); err != nil {
|
|
return nil, fmt.Errorf("create p2p: %w", err)
|
|
}
|
|
|
|
return config, nil
|
|
}
|