88 lines
2.4 KiB
Go
88 lines
2.4 KiB
Go
package network
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/CityOfZion/neo-go/config"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
type (
|
|
// ServerConfig holds the server configuration.
|
|
ServerConfig struct {
|
|
// MinPeers is the minimum number of peers for normal operation,
|
|
// when the node has less than this number of peers it tries to
|
|
// connect with some new ones.
|
|
MinPeers int
|
|
|
|
// AttemptConnPeers it the number of connection to try to
|
|
// establish when the connection count drops below the MinPeers
|
|
// value.
|
|
AttemptConnPeers int
|
|
|
|
// MaxPeers it the maximum numbers of peers that can
|
|
// be connected to the server.
|
|
MaxPeers int
|
|
|
|
// The user agent of the server.
|
|
UserAgent string
|
|
|
|
// Address. Example: "127.0.0.1".
|
|
Address string
|
|
|
|
// Port. Example: 20332.
|
|
Port uint16
|
|
|
|
// The network mode the server will operate on.
|
|
// ModePrivNet docker private network.
|
|
// ModeTestNet NEO test network.
|
|
// ModeMainNet NEO main network.
|
|
Net config.NetMode
|
|
|
|
// Relay determines whether the server is forwarding its inventory.
|
|
Relay bool
|
|
|
|
// Seeds are a list of initial nodes used to establish connectivity.
|
|
Seeds []string
|
|
|
|
// Maximum duration a single dial may take.
|
|
DialTimeout time.Duration
|
|
|
|
// The duration between protocol ticks with each connected peer.
|
|
// When this is 0, the default interval of 5 seconds will be used.
|
|
ProtoTickInterval time.Duration
|
|
|
|
// Level of the internal logger.
|
|
LogLevel log.Level
|
|
|
|
// Wallet is a wallet configuration.
|
|
Wallet *config.WalletConfig
|
|
}
|
|
)
|
|
|
|
// NewServerConfig creates a new ServerConfig struct
|
|
// using the main applications config.
|
|
func NewServerConfig(cfg config.Config) ServerConfig {
|
|
appConfig := cfg.ApplicationConfiguration
|
|
protoConfig := cfg.ProtocolConfiguration
|
|
|
|
var wc *config.WalletConfig
|
|
if appConfig.UnlockWallet.Path != "" {
|
|
wc = &appConfig.UnlockWallet
|
|
}
|
|
|
|
return ServerConfig{
|
|
UserAgent: cfg.GenerateUserAgent(),
|
|
Address: appConfig.Address,
|
|
Port: appConfig.NodePort,
|
|
Net: protoConfig.Magic,
|
|
Relay: appConfig.Relay,
|
|
Seeds: protoConfig.SeedList,
|
|
DialTimeout: appConfig.DialTimeout * time.Second,
|
|
ProtoTickInterval: appConfig.ProtoTickInterval * time.Second,
|
|
MaxPeers: appConfig.MaxPeers,
|
|
AttemptConnPeers: appConfig.AttemptConnPeers,
|
|
MinPeers: appConfig.MinPeers,
|
|
Wallet: wc,
|
|
}
|
|
}
|