93 lines
2.6 KiB
Go
93 lines
2.6 KiB
Go
package common
|
|
|
|
import (
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
// Common config for all services
|
|
|
|
type Config struct {
|
|
ConfigFilePath string `env:"CONFIG_FILE_PATH"`
|
|
MessageSizeLimit int `env:"QUEUE_MESSAGE_SIZE_LIMIT,default=1048576"`
|
|
MaxMemoryUsage uint64 `env:"MAX_MEMORY_USAGE,default=80"`
|
|
MemoryLimitMB uint64 `env:"MEMORY_LIMIT_MB,default=0"` // 0 means take limit from OS (cgroup)
|
|
}
|
|
|
|
type Configer interface {
|
|
GetConfigPath() string
|
|
}
|
|
|
|
func (c *Config) GetConfigPath() string {
|
|
return c.ConfigFilePath
|
|
}
|
|
|
|
// Postgres config
|
|
|
|
type Postgres struct {
|
|
Postgres string `env:"POSTGRES_STRING,required"`
|
|
ApplicationName string `env:"SERVICE_NAME,default='worker'"`
|
|
}
|
|
|
|
func (cfg *Postgres) String() string {
|
|
str := cfg.Postgres
|
|
if !strings.Contains(cfg.Postgres, "application_name") {
|
|
if strings.Contains(cfg.Postgres, "?") {
|
|
str += "&"
|
|
} else {
|
|
str += "?"
|
|
}
|
|
str += "application_name=" + cfg.ApplicationName
|
|
}
|
|
return str
|
|
}
|
|
|
|
// Redshift config
|
|
|
|
type Redshift struct {
|
|
ConnectionString string `env:"REDSHIFT_STRING"`
|
|
Host string `env:"REDSHIFT_HOST"`
|
|
Port int `env:"REDSHIFT_PORT"`
|
|
User string `env:"REDSHIFT_USER"`
|
|
Password string `env:"REDSHIFT_PASSWORD"`
|
|
Database string `env:"REDSHIFT_DATABASE"`
|
|
Bucket string `env:"REDSHIFT_BUCKET,default=rdshftbucket"`
|
|
}
|
|
|
|
// Clickhouse config
|
|
|
|
type Clickhouse struct {
|
|
URL string `env:"CLICKHOUSE_STRING"`
|
|
Database string `env:"CLICKHOUSE_DATABASE,default=default"`
|
|
UserName string `env:"CLICKHOUSE_USERNAME,default=default"`
|
|
Password string `env:"CLICKHOUSE_PASSWORD,default="`
|
|
LegacyUserName string `env:"CH_USERNAME,default=default"`
|
|
LegacyPassword string `env:"CH_PASSWORD,default="`
|
|
}
|
|
|
|
func (cfg *Clickhouse) GetTrimmedURL() string {
|
|
chUrl := strings.TrimPrefix(cfg.URL, "tcp://")
|
|
chUrl = strings.TrimSuffix(chUrl, "/default")
|
|
return chUrl
|
|
}
|
|
|
|
// ElasticSearch config
|
|
|
|
type ElasticSearch struct {
|
|
URLs string `env:"ELASTICSEARCH_URLS"`
|
|
UseAWS bool `env:"ELASTICSEARCH_IN_AWS,default=false"`
|
|
}
|
|
|
|
func (cfg *ElasticSearch) GetURLs() []string {
|
|
return strings.Split(cfg.URLs, ",")
|
|
}
|
|
|
|
type HTTP struct {
|
|
HTTPHost string `env:"HTTP_HOST,default="`
|
|
HTTPPort string `env:"HTTP_PORT,required"`
|
|
HTTPTimeout time.Duration `env:"HTTP_TIMEOUT,default=60s"`
|
|
JsonSizeLimit int64 `env:"JSON_SIZE_LIMIT,default=131072"` // 128KB, 1000 for HTTP service
|
|
UseAccessControlHeaders bool `env:"USE_CORS,default=false"`
|
|
JWTSecret string `env:"JWT_SECRET"`
|
|
JWTSpotSecret string `env:"JWT_SPOT_SECRET"`
|
|
}
|