Skip to content

Commit be25df2

Browse files
committed
Addressing comments
Signed-off-by: Alan Protasio <[email protected]>
1 parent dba6bc4 commit be25df2

File tree

2 files changed

+13
-9
lines changed

2 files changed

+13
-9
lines changed

pkg/alertmanager/alertstore/local/store.go

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -154,17 +154,17 @@ func (f *Store) reloadConfigs() (map[string]alertspb.AlertConfigDesc, error) {
154154
var templates []*alertspb.TemplateDesc
155155

156156
if _, e := os.Stat(userTemplateDir); e == nil {
157-
err = filepath.Walk(filepath.Join(f.cfg.Path, user, templatesDir), func(path string, info os.FileInfo, err error) error {
157+
err = filepath.Walk(userTemplateDir, func(templatePath string, info os.FileInfo, err error) error {
158158
if err != nil {
159-
return errors.Wrapf(err, "unable to walk file path at %s", path)
159+
return errors.Wrapf(err, "unable to walk file path at %s", templatePath)
160160
}
161161
// Ignore files that are directories
162162
if info.IsDir() {
163163
return nil
164164
}
165-
content, err := os.ReadFile(path)
165+
content, err := os.ReadFile(templatePath)
166166
if err != nil {
167-
return errors.Wrapf(err, "unable to read alertmanager templates %s", path)
167+
return errors.Wrapf(err, "unable to read alertmanager templates %s", templatePath)
168168
}
169169

170170
templates = append(templates, &alertspb.TemplateDesc{
@@ -175,8 +175,10 @@ func (f *Store) reloadConfigs() (map[string]alertspb.AlertConfigDesc, error) {
175175
})
176176

177177
if err != nil {
178-
return errors.Wrapf(err, "unable to load alertmanager config %s", path)
178+
return errors.Wrapf(err, "unable to list alertmanager templates: %s", userTemplateDir)
179179
}
180+
} else if !os.IsNotExist(e) {
181+
return errors.Wrapf(e, "unable to read alertmanager templates %s", path)
180182
}
181183

182184
configs[user] = alertspb.AlertConfigDesc{

pkg/alertmanager/alertstore/local/store_test.go

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,7 @@ func TestStore_GetAlertConfigs(t *testing.T) {
8181
// The storage contains some configs.
8282
{
8383
user1Cfg := prepareAlertmanagerConfig("user-1")
84-
user1Dir, user1TemplateDir := prepareUserDir(t, storeDir, "user-1")
84+
user1Dir, user1TemplateDir := prepareUserDir(t, storeDir, true, "user-1")
8585
require.NoError(t, os.WriteFile(filepath.Join(user1Dir, "user-1.yaml"), []byte(user1Cfg), os.ModePerm))
8686

8787
require.NoError(t, os.WriteFile(filepath.Join(user1TemplateDir, "template.tpl"), []byte("testTemplate"), os.ModePerm))
@@ -95,7 +95,7 @@ func TestStore_GetAlertConfigs(t *testing.T) {
9595

9696
// Add another user config.
9797
user2Cfg := prepareAlertmanagerConfig("user-2")
98-
user2Dir, _ := prepareUserDir(t, storeDir, "user-2")
98+
user2Dir, _ := prepareUserDir(t, storeDir, false, "user-2")
9999
require.NoError(t, os.WriteFile(filepath.Join(user2Dir, "user-2.yaml"), []byte(user2Cfg), os.ModePerm))
100100

101101
configs, err = store.GetAlertConfigs(ctx, []string{"user-1", "user-2"})
@@ -107,11 +107,13 @@ func TestStore_GetAlertConfigs(t *testing.T) {
107107
}
108108
}
109109

110-
func prepareUserDir(t *testing.T, storeDir string, user string) (userDir string, templateDir string) {
110+
func prepareUserDir(t *testing.T, storeDir string, createTemplateDir bool, user string) (userDir string, templateDir string) {
111111
userDir = filepath.Join(storeDir, user)
112112
templateDir = filepath.Join(userDir, templatesDir)
113113
require.NoError(t, os.MkdirAll(userDir, os.ModePerm))
114-
require.NoError(t, os.MkdirAll(templateDir, os.ModePerm))
114+
if createTemplateDir {
115+
require.NoError(t, os.MkdirAll(templateDir, os.ModePerm))
116+
}
115117
return
116118
}
117119

0 commit comments

Comments
 (0)