init
This commit is contained in:
windyboy
@@ -0,0 +1,164 @@
|
||||
package config
|
||||
|
||||
import (
|
||||
"caatsm/pkg/utils"
|
||||
"fmt"
|
||||
"os"
|
||||
"regexp"
|
||||
"strings"
|
||||
|
||||
"github.com/spf13/viper"
|
||||
// Adjust this import based on your project structure
|
||||
)
|
||||
|
||||
var MyConfig *Config
|
||||
|
||||
type Config struct {
|
||||
Nats NatsConfig
|
||||
Subscription SubscriptionConfig
|
||||
Timeouts TimeoutsConfig
|
||||
Body []BodyConfig
|
||||
}
|
||||
|
||||
type NatsConfig struct {
|
||||
Client string
|
||||
URL string
|
||||
Cluster string
|
||||
}
|
||||
|
||||
type SubscriptionConfig struct {
|
||||
Topic string
|
||||
QueueGroup string `mapstructure:"queue_group"`
|
||||
}
|
||||
|
||||
type TimeoutsConfig struct {
|
||||
ServerTimeout string `mapstructure:"server_timeout"`
|
||||
ReconnectWait string `mapstructure:"reconnect_wait"`
|
||||
CloseTimeout string `mapstructure:"close_timeout"`
|
||||
AckWaitTimeout string `mapstructure:"ack_wait_timeout"`
|
||||
}
|
||||
|
||||
type BodyConfig struct {
|
||||
Name string
|
||||
Patterns []PatternConfig
|
||||
}
|
||||
|
||||
type PatternConfig struct {
|
||||
Pattern string
|
||||
Comments string
|
||||
Expression *regexp.Regexp
|
||||
}
|
||||
|
||||
func SetMyConfig(cfg *Config) {
|
||||
MyConfig = cfg
|
||||
}
|
||||
|
||||
func GetMyConfig() *Config {
|
||||
if MyConfig == nil {
|
||||
cfg, err := LoadConfig()
|
||||
if err != nil {
|
||||
utils.Logger.Fatalf("error loading config: %v", err)
|
||||
}
|
||||
MyConfig = cfg
|
||||
}
|
||||
return MyConfig
|
||||
}
|
||||
|
||||
// LoadConfig loads the configuration from a file
|
||||
func LoadConfig() (*Config, error) {
|
||||
log := utils.Logger
|
||||
env := os.Getenv("GO_ENV")
|
||||
if env == "" {
|
||||
env = "dev"
|
||||
}
|
||||
log.Infof("Environment: %s", env)
|
||||
|
||||
viper.SetConfigType("toml")
|
||||
viper.SetConfigName("config." + env)
|
||||
viper.AddConfigPath("configs")
|
||||
viper.SetEnvPrefix("tele")
|
||||
viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_"))
|
||||
|
||||
if err := viper.ReadInConfig(); err != nil {
|
||||
errMsg := fmt.Sprintf("error reading config file for environment '%s': %v", env, err)
|
||||
log.Error(errMsg)
|
||||
return nil, fmt.Errorf(errMsg)
|
||||
}
|
||||
|
||||
log.Debug("Config file read successfully")
|
||||
log.Debugf("Config keys: %v", viper.AllKeys())
|
||||
|
||||
var config Config
|
||||
if err := viper.Unmarshal(&config); err != nil {
|
||||
errMsg := fmt.Sprintf("unable to decode config into struct for environment '%s': %v", env, err)
|
||||
log.Error(errMsg)
|
||||
return nil, fmt.Errorf(errMsg)
|
||||
}
|
||||
|
||||
log.Debugf("Config loaded before regex compilation: %+v", config)
|
||||
|
||||
// Compile regex patterns
|
||||
for i := range config.Body {
|
||||
for j := range config.Body[i].Patterns {
|
||||
name := config.Body[i].Name
|
||||
pattern := config.Body[i].Patterns[j].Pattern
|
||||
expr, err := regexp.Compile(pattern)
|
||||
if err != nil {
|
||||
errMsg := fmt.Sprintf("error compiling regex for body '%s', pattern '%s': %v", name, pattern, err)
|
||||
log.Error(errMsg)
|
||||
return nil, fmt.Errorf(errMsg)
|
||||
}
|
||||
config.Body[i].Patterns[j].Expression = expr
|
||||
}
|
||||
}
|
||||
|
||||
log.Debugf("Final config after regex compilation: %+v", config)
|
||||
return &config, nil
|
||||
}
|
||||
|
||||
// ValidateConfig validates the loaded configuration
|
||||
func ValidateConfig(cfg *Config) error {
|
||||
log := utils.Logger
|
||||
|
||||
if cfg.Nats.Client == "" {
|
||||
err := "nats client is required"
|
||||
log.Error(err)
|
||||
return fmt.Errorf(err)
|
||||
}
|
||||
if cfg.Nats.URL == "" {
|
||||
err := "nats URL is required"
|
||||
log.Error(err)
|
||||
return fmt.Errorf(err)
|
||||
}
|
||||
if cfg.Subscription.Topic == "" {
|
||||
err := "subscription topic is required"
|
||||
log.Error(err)
|
||||
return fmt.Errorf(err)
|
||||
}
|
||||
if len(cfg.Body) == 0 {
|
||||
err := "at least one body configuration is required"
|
||||
log.Error(err)
|
||||
return fmt.Errorf(err)
|
||||
}
|
||||
for _, body := range cfg.Body {
|
||||
if body.Name == "" {
|
||||
err := "body name is required"
|
||||
log.Error(err)
|
||||
return fmt.Errorf(err)
|
||||
}
|
||||
for _, pattern := range body.Patterns {
|
||||
if pattern.Pattern == "" {
|
||||
err := fmt.Sprintf("pattern is required for body '%s'", body.Name)
|
||||
log.Error(err)
|
||||
return fmt.Errorf(err)
|
||||
}
|
||||
if pattern.Expression == nil {
|
||||
err := fmt.Sprintf("compiled expression is missing for pattern '%s' in body '%s'", pattern.Pattern, body.Name)
|
||||
log.Error(err)
|
||||
return fmt.Errorf(err)
|
||||
}
|
||||
}
|
||||
}
|
||||
log.Info("config validation passed")
|
||||
return nil
|
||||
}
|
||||
@@ -0,0 +1,128 @@
|
||||
package config
|
||||
|
||||
import (
|
||||
"os"
|
||||
"path/filepath"
|
||||
"testing"
|
||||
|
||||
. "github.com/onsi/ginkgo/v2"
|
||||
. "github.com/onsi/gomega"
|
||||
)
|
||||
|
||||
func TestConfig(t *testing.T) {
|
||||
RegisterFailHandler(Fail)
|
||||
RunSpecs(t, "Config Suite")
|
||||
}
|
||||
|
||||
var _ = Describe("Config", func() {
|
||||
var originalEnv string
|
||||
|
||||
BeforeEach(func() {
|
||||
// Save the original GO_ENV value
|
||||
originalEnv = os.Getenv("GO_ENV")
|
||||
})
|
||||
|
||||
AfterEach(func() {
|
||||
// Restore the original GO_ENV value
|
||||
os.Setenv("GO_ENV", originalEnv)
|
||||
})
|
||||
|
||||
// Helper function to create temporary TOML files in the "configs" directory
|
||||
createTempConfigFile := func(env, content string) string {
|
||||
dir := "configs"
|
||||
err := os.MkdirAll(dir, 0755)
|
||||
Expect(err).NotTo(HaveOccurred(), "failed to create config directory")
|
||||
|
||||
filename := filepath.Join(dir, "config."+env+".toml")
|
||||
tmpfile, err := os.Create(filename)
|
||||
Expect(err).NotTo(HaveOccurred(), "failed to create temp config file")
|
||||
_, err = tmpfile.Write([]byte(content))
|
||||
Expect(err).NotTo(HaveOccurred(), "failed to write to temp config file")
|
||||
err = tmpfile.Close()
|
||||
Expect(err).NotTo(HaveOccurred(), "failed to close temp config file")
|
||||
return filename
|
||||
}
|
||||
|
||||
Context("Loading and validating a valid TOML file", func() {
|
||||
It("should load and validate the configuration correctly", func() {
|
||||
content := `
|
||||
[nats]
|
||||
client = "test-client"
|
||||
url = "nats://localhost:4222"
|
||||
cluster = "test-cluster"
|
||||
|
||||
[subscription]
|
||||
topic = "example-topic"
|
||||
queue_group = "example-group"
|
||||
|
||||
[timeouts]
|
||||
server_timeout = "30s"
|
||||
reconnect_wait = "10s"
|
||||
close_timeout = "10s"
|
||||
ack_wait_timeout = "5s"
|
||||
|
||||
[[body]]
|
||||
name = "FPL"
|
||||
[[body.patterns]]
|
||||
pattern = "^\\((?P<type>[A-Z]{3})\\-(?P<number>[A-Z]+\\d+)\\-(?P<indicator>[A-Z]{2})(?:.*\\s*)?\\-(?P<aircraft>[A-Z]+\\d+/?[A-Z]?)\\s*\\-(?P<surve>.*)\\s*\\-(?P<departure>[A-Z]{4})(?P<departure_time>\\d{4})\\s*\\-(?P<speed>[A-Z]+\\d+)(?P<level>[A-Z0-9]+)\\s(?P<route>.*)\\s*\\-(?P<destination>[A-Z]{4})(?P<estt>\\d{4})\\s(?P<alter>[A-Z]{4})\\s*\\-(?P<pbn>PBN\\/[A-Z0-9]+)\\s(?P<nav>NAV\\/\\w+)\\sREG\\/(?P<reg>[A-Z0-9]+)\\sEET\\/(?P<eet>\\w{4}\\d{4})\\sSEL\\/(?P<sel>\\w+)\\sPER\\/(?P<performance>\\w)\\sRIF\\/(?P<rif>\\w+\\s[A-Z0-9]+\\s[A-Z]+)\\s*RMK\\/(?P<remark>.*)\\)$"
|
||||
comments = "FPL multi line expression"
|
||||
`
|
||||
os.Setenv("GO_ENV", "test")
|
||||
createTempConfigFile("test", content)
|
||||
defer os.RemoveAll("configs")
|
||||
|
||||
config, err := LoadConfig()
|
||||
Expect(err).NotTo(HaveOccurred(), "failed to load valid config")
|
||||
Expect(config).NotTo(BeNil(), "config should not be nil")
|
||||
Expect(config.Nats.Client).To(Equal("test-client"), "nats.client should be 'test-client'")
|
||||
Expect(config.Subscription.Topic).To(Equal("example-topic"), "subscription.topic should be 'example-topic'")
|
||||
|
||||
// Validate the config
|
||||
err = ValidateConfig(config)
|
||||
Expect(err).NotTo(HaveOccurred(), "validation should pass for valid config")
|
||||
})
|
||||
})
|
||||
|
||||
Context("Loading and validating a non-existent file", func() {
|
||||
It("should return an error", func() {
|
||||
os.Setenv("GO_ENV", "nonexistent")
|
||||
defer os.RemoveAll("configs")
|
||||
|
||||
_, err := LoadConfig()
|
||||
Expect(err).To(HaveOccurred(), "expected error for non-existent config file")
|
||||
})
|
||||
})
|
||||
|
||||
Context("Loading and validating a file with invalid TOML format", func() {
|
||||
It("should return an error", func() {
|
||||
content := `
|
||||
invalid TOML content
|
||||
`
|
||||
os.Setenv("GO_ENV", "invalid")
|
||||
createTempConfigFile("invalid", content)
|
||||
defer os.RemoveAll("configs")
|
||||
|
||||
_, err := LoadConfig()
|
||||
Expect(err).To(HaveOccurred(), "expected error for invalid TOML format")
|
||||
})
|
||||
})
|
||||
|
||||
Context("Validating an invalid config structure", func() {
|
||||
It("should return an error for missing required fields", func() {
|
||||
invalidConfig := &Config{
|
||||
Nats: NatsConfig{
|
||||
Client: "",
|
||||
URL: "",
|
||||
},
|
||||
Subscription: SubscriptionConfig{
|
||||
Topic: "",
|
||||
},
|
||||
Body: []BodyConfig{},
|
||||
}
|
||||
|
||||
err := ValidateConfig(invalidConfig)
|
||||
Expect(err).To(HaveOccurred(), "expected validation error for invalid config")
|
||||
Expect(err.Error()).To(ContainSubstring("nats client is required"), "expected error for missing nats client")
|
||||
})
|
||||
})
|
||||
})
|
||||
Reference in New Issue
Block a user