From 651f15d9b8bde1b024f97a42e92634afdc1eeda7 Mon Sep 17 00:00:00 2001 From: Emile Date: Mon, 27 Jan 2020 20:13:15 +0100 Subject: moved the source into an own folder --- src/http.go | 52 ++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 52 insertions(+) create mode 100644 src/http.go (limited to 'src/http.go') diff --git a/src/http.go b/src/http.go new file mode 100644 index 0000000..7aac0c7 --- /dev/null +++ b/src/http.go @@ -0,0 +1,52 @@ +package main + +// locationHandlerEndpoint handles requests to the /locations endpoint +// This is used by the grafana worldmap plugin to find out where to draw the +// fancy circles +func locationHandlerEndpoint(w http.ResponseWriter, r *http.Request) { + + // set some headers + w.Header().Set("Content-Type", "application/json") + w.Header().Set("Access-Control-Allow-Origin", "https://grafana.nbg1.emile.space") + + // start building json (yes, this is not a nice implementation, PRs welcome!) + fmt.Fprintf(w, "%s", "[") + + var i int = 0 + for _, v := range cities { + + // print the "json" object containing the metrics needed + fmt.Fprintf(w, "{") + fmt.Fprintf(w, "\"key\": \"%s\",", v.key) + fmt.Fprintf(w, "\"latitude\": %f,", v.latitude) + fmt.Fprintf(w, "\"longitude\": %f,", v.longitude) + fmt.Fprintf(w, "\"name\": \"%s\"", v.name) + + // close the object (this handles the trailing comma problem) + if i == len(cities) - 1 { + fmt.Fprintf(w, "}") + } else { + fmt.Fprintf(w, "},") + } + i++ + } + fmt.Fprintf(w, "%s", "]") +} + +// indexHandler handles the request to the / endpoint +// It simply returns a link to the /metrics page +func indexHandler(w http.ResponseWriter, req *http.Request) { + _, _ = fmt.Fprintf(w, "metrics") +} + +// Handle HTTP requests to the /metrics endpoint +func metricsHandler(w http.ResponseWriter, req *http.Request) { + + // return the overall amount of passwords catched + fmt.Fprintf(w, "num_passwords %d\n", metrics_num_passwords) + + // return the amount of passwords catched from a given city + for k, v := range metrics_city_num { + fmt.Fprintf(w, "a_metric{city=\"%s\"} %d\n", strings.ToLower(k), v) + } +} \ No newline at end of file -- cgit 1.4.1