diff --git a/config.go b/config.go index 27ee69d..b667f61 100644 --- a/config.go +++ b/config.go @@ -16,6 +16,9 @@ type Config struct { InitializeSlack bool SlackToken string SlackChannels string + storage Storage + queue Queue + driver Driver } func newConfig() (Config, error) { @@ -73,5 +76,10 @@ func newConfigFromEnv(getEnv func(string) string) (Config, error) { } else if err := json.Unmarshal(b, &result); err != nil { return Config{}, err } + + result.driver = NewRAM() + result.storage = NewStorage(result.driver) + result.queue = NewQueue(result.driver) + return result, nil } diff --git a/main.go b/main.go index a996087..8b3782c 100644 --- a/main.go +++ b/main.go @@ -23,6 +23,7 @@ func main() { if err != nil { panic(err) } + defer cfg.driver.Close() if err := run(ctx, cfg); err != nil && ctx.Err() == nil { panic(err)