package main import ( "fmt" "io/ioutil" "net/http" "strings" "github.com/by-cx/lobby/server" "github.com/labstack/echo" ) func listHandler(c echo.Context) error { labels := c.QueryParam("labels") prefixes := c.QueryParam("prefixes") var discoveries []server.Discovery if len(labels) > 0 { labelsFilterSlice := strings.Split(labels, ",") discoveries = discoveryStorage.Filter(labelsFilterSlice) } else if len(prefixes) > 0 { discoveries = discoveryStorage.FilterPrefix(strings.Split(prefixes, ",")) } else { discoveries = discoveryStorage.GetAll() } return c.JSONPretty(200, discoveries, " ") } func prometheusHandler(c echo.Context) error { name := c.Param("name") services := preparePrometheusOutput(name, discoveryStorage.GetAll()) return c.JSONPretty(http.StatusOK, services, " ") } func getIdentificationHandler(c echo.Context) error { discovery, err := localHost.GetIdentification() if err != nil { return c.String(http.StatusInternalServerError, fmt.Sprintf("gathering identification info error: %v\n", err)) } return c.JSONPretty(http.StatusOK, discovery, " ") } func addLabelsHandler(c echo.Context) error { body, err := ioutil.ReadAll(c.Request().Body) if err != nil { return c.String(http.StatusBadRequest, fmt.Sprintf("reading request body error: %v\n", err)) } labels := server.Labels{} for _, label := range strings.Split(string(body), "\n") { labels = append(labels, server.Label(label)) } err = localHost.AddLabels(labels) if err != nil { return c.String(http.StatusBadRequest, err.Error()) } return c.String(http.StatusOK, "OK") } func deleteLabelsHandler(c echo.Context) error { body, err := ioutil.ReadAll(c.Request().Body) if err != nil { return c.String(http.StatusBadRequest, fmt.Sprintf("reading request body error: %v\n", err)) } labels := server.Labels{} for _, label := range strings.Split(string(body), "\n") { labels = append(labels, server.Label(label)) } err = localHost.DeleteLabels(labels) if err != nil { return c.String(http.StatusBadRequest, err.Error()) } return c.String(http.StatusOK, "OK") }