mirror of
https://github.com/eosswedenorg/thalos
synced 2026-06-17 04:30:03 +02:00
85 lines
2.5 KiB
Go
85 lines
2.5 KiB
Go
package config
|
|
|
|
import (
|
|
"reflect"
|
|
"time"
|
|
|
|
"github.com/eosswedenorg/thalos/internal/log"
|
|
|
|
shipclient "github.com/eosswedenorg-go/antelope-ship-client"
|
|
)
|
|
|
|
type RedisConfig struct {
|
|
Addr string `yaml:"addr"`
|
|
User string `yaml:"user"`
|
|
Password string `yaml:"password"`
|
|
DB int `yaml:"db"`
|
|
Prefix string `yaml:"prefix"`
|
|
}
|
|
|
|
type TelegramConfig struct {
|
|
Id string `yaml:"id" mapstructure:"id"`
|
|
Channel int64 `yaml:"channel" mapstructure:"channel"`
|
|
}
|
|
|
|
type ShipConfig struct {
|
|
Url string `yaml:"url" mapstructure:"url"`
|
|
IrreversibleOnly bool `yaml:"irreversible_only" mapstructure:"irreversible_only"`
|
|
MaxMessagesInFlight uint32 `yaml:"max_messages_in_flight" mapstructure:"max_messages_in_flight"`
|
|
StartBlockNum uint32 `yaml:"start_block_num" mapstructure:"start_block_num"`
|
|
EndBlockNum uint32 `yaml:"end_block_num" mapstructure:"end_block_num"`
|
|
Chain string `yaml:"chain" mapstructure:"chain"`
|
|
}
|
|
|
|
type Config struct {
|
|
Name string `yaml:"name" mapstructure:"name"`
|
|
Ship ShipConfig `yaml:"ship" mapstructure:"ship"`
|
|
Api string `yaml:"api" mapstructure:"api"`
|
|
|
|
Log log.Config `yaml:"log" mapstructure:"log"`
|
|
|
|
Redis RedisConfig `yaml:"redis" mapstructure:"redis"`
|
|
MessageCodec string `yaml:"message_codec" mapstructure:"message_codec"`
|
|
|
|
Telegram TelegramConfig `yaml:"telegram" mapstructure:"telegram"`
|
|
}
|
|
|
|
// Create a new Config object with default values
|
|
func New() Config {
|
|
return Config{
|
|
MessageCodec: "json",
|
|
Log: log.Config{
|
|
MaxFileSize: 10 * 1000 * 1000, // 10 mb
|
|
MaxTime: time.Hour * 24,
|
|
},
|
|
Ship: defaultShipConfig(""),
|
|
Redis: RedisConfig{
|
|
Addr: "localhost:6379",
|
|
Prefix: "ship",
|
|
},
|
|
}
|
|
}
|
|
|
|
func defaultShipConfig(url string) ShipConfig {
|
|
return ShipConfig{
|
|
Url: url,
|
|
StartBlockNum: shipclient.NULL_BLOCK_NUMBER,
|
|
EndBlockNum: shipclient.NULL_BLOCK_NUMBER,
|
|
MaxMessagesInFlight: 10,
|
|
IrreversibleOnly: false,
|
|
}
|
|
}
|
|
|
|
// mapstructure DecodeHook that can parse a shorthand ship config (only string instead of struct.)
|
|
func decodeShorthandShipConfig(from reflect.Value, to reflect.Value) (interface{}, error) {
|
|
shipType := reflect.TypeOf(ShipConfig{})
|
|
|
|
// If to is a struct and is assignable to a ShipConfig and from is a string.
|
|
// Then we treat the from value as ShipConfig.Url
|
|
if to.Kind() == reflect.Struct && to.Type().AssignableTo(shipType) && from.Kind() == reflect.String {
|
|
return defaultShipConfig(from.String()), nil
|
|
}
|
|
|
|
// If not, decode as normal.
|
|
return from.Interface(), nil
|
|
}
|