42 lines
627 B
Go
42 lines
627 B
Go
package config
|
|
|
|
import (
|
|
"os"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
"gopkg.in/yaml.v2"
|
|
)
|
|
|
|
var Cfg Config
|
|
|
|
type Config struct {
|
|
Debug bool
|
|
ScyllaDB ScyllaDB `yaml:"scylladb"`
|
|
RabbitMq RabbitMq
|
|
}
|
|
|
|
type ScyllaDB struct {
|
|
Host string
|
|
Username string
|
|
Password string
|
|
Keyspace string
|
|
}
|
|
|
|
type RabbitMq struct {
|
|
Host string
|
|
Username string
|
|
Password string
|
|
}
|
|
|
|
func LoadConfig() {
|
|
data, err := os.ReadFile("config.yml")
|
|
|
|
if err != nil {
|
|
logrus.Fatalln("Failed to read config file, err:", err)
|
|
}
|
|
|
|
if err := yaml.Unmarshal(data, &Cfg); err != nil {
|
|
logrus.Fatalln("Failed to unmarshal config file, err:", err)
|
|
}
|
|
}
|