phares3757/gatus
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Improve test coverage

Browse Source
This commit is contained in:
TwinProduction
2021-05-19 00:55:03 -04:00
parent 263b2f0f94
commit 63b0ac8b35
2 changed files with 41 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
storage/storage.go
View File

@ -37,6 +37,10 @@ func Get() store.Store {
func Initialize(cfg *Config) error {
initialized = true
var err error
if cancelFunc != nil {
// Stop the active autoSave task
cancelFunc()
}
if cfg == nil || len(cfg.File) == 0 {
log.Println("[storage][Initialize] Creating storage provider")
provider, err = memory.NewStore("")
@ -44,10 +48,6 @@ func Initialize(cfg *Config) error {
return err
}
} else {
if cancelFunc != nil {
// Stop the active autoSave task
cancelFunc()
}
ctx, cancelFunc = context.WithCancel(context.Background())
log.Printf("[storage][Initialize] Creating storage provider with file=%s", cfg.File)
provider, err = memory.NewStore(cfg.File)