configuration for metrics.yaml and templates

docker/n9eetc/server.conf

@@ -54,6 +54,7 @@ Interval = 1000
 [Alerting]
 NotifyScriptPath = "./etc/script/notify.py"
 NotifyConcurrency = 100
+TemplatesDir = "./etc/template"
 
 [Alerting.RedisPub]
 Enable = false

docker/n9eetc/webapi.conf

@@ -7,6 +7,9 @@ RunMode = "release"
 # do not change
 AdminRole = "Admin"
 
+# metrics descriptions
+MetricsYamlFile = "./etc/metrics.yaml"
+
 # Linkage with notify.py script
 NotifyChannels = [ "email", "dingtalk", "wecom", "feishu" ]
 

etc/server.conf

@@ -57,6 +57,7 @@ Interval = 1000
 [Alerting]
 NotifyScriptPath = "./etc/script/notify.py"
 NotifyConcurrency = 100
+TemplatesDir = "./etc/template"
 
 [Alerting.RedisPub]
 Enable = false

etc/webapi.conf

@@ -7,6 +7,9 @@ RunMode = "release"
 # do not change
 AdminRole = "Admin"
 
+# metrics descriptions
+MetricsYamlFile = "./etc/metrics.yaml"
+
 # Linkage with notify.py script
 NotifyChannels = [ "email", "dingtalk", "wecom", "feishu" ]
 

src/server/config/config.go

@@ -113,6 +113,7 @@ type HeartbeatConfig struct {
 type Alerting struct {
 	NotifyScriptPath  string
 	NotifyConcurrency int
+	TemplatesDir      string
 	RedisPub          RedisPub
 }
 

src/server/engine/notify.go

@@ -44,15 +44,17 @@ var fns = template.FuncMap{
 }
 
 func initTpls() error {
-	tplDir := path.Join(runner.Cwd, "etc", "template")
+	if config.C.Alerting.TemplatesDir == "" {
+		config.C.Alerting.TemplatesDir = path.Join(runner.Cwd, "etc", "template")
+	}
 
-	filenames, err := file.FilesUnder(tplDir)
+	filenames, err := file.FilesUnder(config.C.Alerting.TemplatesDir)
 	if err != nil {
 		return errors.WithMessage(err, "failed to exec FilesUnder")
 	}
 
 	if len(filenames) == 0 {
-		return errors.New("no tpl files under " + tplDir)
+		return errors.New("no tpl files under " + config.C.Alerting.TemplatesDir)
 	}
 
 	tplFiles := make([]string, 0, len(filenames))
@@ -63,11 +65,11 @@ func initTpls() error {
 	}
 
 	if len(tplFiles) == 0 {
-		return errors.New("no tpl files under " + tplDir)
+		return errors.New("no tpl files under " + config.C.Alerting.TemplatesDir)
 	}
 
 	for i := 0; i < len(tplFiles); i++ {
-		tplpath := path.Join(tplDir, tplFiles[i])
+		tplpath := path.Join(config.C.Alerting.TemplatesDir, tplFiles[i])
 
 		tpl, err := template.New(tplFiles[i]).Funcs(fns).ParseFiles(tplpath)
 		if err != nil {

src/webapi/config/config.go

@@ -77,6 +77,7 @@ type Config struct {
 	RunMode         string
 	I18N            string
 	AdminRole       string
+	MetricsYamlFile string
 	ContactKeys     []ContactKey
 	NotifyChannels  []string
 	Log             logx.Config

src/webapi/router/router_metric_desc.go

@@ -12,7 +12,11 @@ import (
 )
 
 func metricsDescGetFile(c *gin.Context) {
-	fp := path.Join(runner.Cwd, "etc", "metrics.yaml")
+	fp := config.C.MetricsYamlFile
+	if fp == "" {
+		fp = path.Join(runner.Cwd, "etc", "metrics.yaml")
+	}
+
 	if !file.IsExist(fp) {
 		c.String(404, "%s not found", fp)
 		return