Skip to content

Commit 1b31f5a

Browse files
committed
Address comments
Signed-off-by: Alan Protasio <approtas@amazon.com>
1 parent d69954f commit 1b31f5a

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

pkg/alertmanager/multitenant.go

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -828,20 +828,22 @@ func (am *MultitenantAlertmanager) setConfig(cfg alertspb.AlertConfigDesc) error
828828
var userTemplateDir = filepath.Join(am.getTenantDirectory(cfg.User), templatesDir)
829829
var pathsToRemove = make(map[string]struct{})
830830

831+
// List existing files to keep track the ones to be removed
831832
if oldTemplateFiles, err := ioutil.ReadDir(userTemplateDir); err == nil {
832833
for _, file := range oldTemplateFiles {
833-
filesToRemove[file.Name()] = filepath.Join(userTemplateDir, file.Name())
834+
pathsToRemove[filepath.Join(userTemplateDir, file.Name())] = struct{}{}
834835
}
835836
}
836837

837838
for _, tmpl := range cfg.Templates {
838-
templateFilepath, err := safeTemplateFilepath(userTemplateDir, tmpl.Filename)
839+
templateFilePath, err := safeTemplateFilepath(userTemplateDir, tmpl.Filename)
839840
if err != nil {
840841
return err
841842
}
842843

844+
// Removing from pathsToRemove map the files that still exists in the config
843845
delete(pathsToRemove, templateFilePath)
844-
hasChanged, err := storeTemplateFile(templateFilepath, tmpl.Body)
846+
hasChanged, err := storeTemplateFile(templateFilePath, tmpl.Body)
845847
if err != nil {
846848
return err
847849
}
@@ -851,7 +853,7 @@ func (am *MultitenantAlertmanager) setConfig(cfg alertspb.AlertConfigDesc) error
851853
}
852854
}
853855

854-
for _, pathToRemove := range pathsToRemove {
856+
for pathToRemove := range pathsToRemove {
855857
err := os.Remove(pathToRemove)
856858
if err != nil {
857859
level.Warn(am.logger).Log("msg", "failed to remove file", "file", pathToRemove, "err", err)

0 commit comments

Comments
 (0)