Fix labels

.gitignore

@@ -1,3 +1,5 @@
 tmp/
 __debug*
 main
+bin/
+nodes.json

.vscode/launch.json

@@ -29,7 +29,7 @@
             "env": {
                 "DUMP_PATH": "../tmp/nodes.json",
                 "CONFIG_PATH": "../tmp/config.json",
-                "NODE_DIR_PATH": "../tmp/node"
+                "NODE_PATH": "../tmp/node.json"
             },
             "args": [
                 "node"

Taskfile.yml

@@ -20,3 +20,8 @@ tasks:
     - go mod tidy
     - mkdir -p bin
     - env CGO_ENABLED=0 go build -o bin/lobby2-{{ .VERSION }}-linux-amd64 cli/*.go
+  deploy:
+    cmds:
+    - task: build
+    - scp bin/lobby2-{{ .VERSION }}-linux-amd64 rosti-db:/usr/local/bin/lobby2.tmp
+    - ssh rosti-db mv /usr/local/bin/lobby2.tmp /usr/local/bin/lobby2

api/main.go

@@ -53,6 +53,7 @@ func (a *API) Run() error {
 	a.e.GET("/nodes", a.listHandler)
 	a.e.GET("/nodes/:hostname", a.getHandler)
 	a.e.POST("/nodes/:hostname", a.refreshHandler)
+	a.e.GET("/prometheus/:service", a.prometheusHandler)
 
 	// Start the server in a goroutine so that it doesn't block the signal listening
 	return a.e.Start(a.listen)
@@ -111,3 +112,18 @@ func (a *API) refreshHandler(c echo.Context) error {
 
 	return c.NoContent(http.StatusNoContent)
 }
+
+// @Summary Prometheus service discovery
+// @Description Return one nodes based on given hostname
+// @Produce  application/json
+// @Success 200 {array} []prometheusDiscovery "Node details"
+// @Failure 401 {object} Message "Forbidden access"
+// @Security Bearer
+// @Router /prometheus/{service} [get]
+func (a *API) prometheusHandler(c echo.Context) error {
+	ss := c.Param("service")
+
+	pds := nodes.GetPrometheusSD(a.np, ss)
+
+	return c.JSONPretty(http.StatusOK, pds, "  ")
+}

cli/actions.go

@@ -52,3 +52,23 @@ func printAction(c *cli.Context) error {
 
 	return nil
 }
+
+func prometheusAction(c *cli.Context) error {
+	cfg := GetConfig()
+
+	np := nodes.NewNodesProcessor(cfg.DumpPath, cfg.DropAfterSeconds)
+
+	pds := nodes.GetPrometheusSD(np, "node")
+
+	for _, pd := range pds {
+		body, err := json.MarshalIndent(pd, "", "  ")
+		if err != nil {
+			fmt.Printf("failed to marshal prometheus discovery: %v\n", err)
+			os.Exit(1)
+		}
+
+		fmt.Println(string(body))
+	}
+
+	return nil
+}

cli/main.go

@@ -27,6 +27,11 @@ func main() {
 				Usage:  "Prints all discovered nodes",
 				Action: printAction,
 			},
+			{
+				Name:   "prometheus",
+				Usage:  "Prints Prometheus Service Discovery",
+				Action: prometheusAction,
+			},
 		},
 	}
 

nodes/prometheus.go

@@ -0,0 +1,45 @@
+package nodes
+
+import (
+	"fmt"
+	"strings"
+)
+
+type prometheusDiscovery struct {
+	Labels  map[string]string `json:"Labels"`
+	Targets []string          `json:"Targets"`
+}
+
+func GetPrometheusSD(p *NodesProcessor, ss string) []prometheusDiscovery {
+	ns := p.List()
+
+	pds := []prometheusDiscovery{
+		{
+			Labels:  make(map[string]string),
+			Targets: []string{},
+		},
+	}
+
+	for _, node := range ns {
+		port, ok := node.KV["prometheus_port"]
+		if !ok {
+			port = "9999"
+		}
+		host, ok := node.KV["prometheus_host"]
+		if !ok {
+			host = node.HostName
+		}
+
+		v, ok := node.KV["prometheus_exporters"]
+		if ok {
+			services := strings.Split(v, ",")
+			for _, service := range services {
+				if ss == service {
+					pds[0].Targets = append(pds[0].Targets, fmt.Sprintf("%s:%s", host, port))
+				}
+			}
+		}
+	}
+
+	return pds
+}