init

Gopkg.toml

@@ -0,0 +1,38 @@
+# Gopkg.toml example
+#
+# Refer to https://golang.github.io/dep/docs/Gopkg.toml.html
+# for detailed Gopkg.toml documentation.
+#
+# required = ["github.com/user/thing/cmd/thing"]
+# ignored = ["github.com/user/project/pkgX", "bitbucket.org/user/project/pkgA/pkgY"]
+#
+# [[constraint]]
+#   name = "github.com/user/project"
+#   version = "1.0.0"
+#
+# [[constraint]]
+#   name = "github.com/user/project2"
+#   branch = "dev"
+#   source = "github.com/myfork/project2"
+#
+# [[override]]
+#   name = "github.com/x/y"
+#   version = "2.4.0"
+#
+# [prune]
+#   non-go = false
+#   go-tests = true
+#   unused-packages = true
+
+
+[[constraint]]
+  name = "github.com/google/uuid"
+  version = "1.1.1"
+
+[[constraint]]
+  name = "github.com/prometheus/prometheus"
+  version = "2.12.0"
+
+[prune]
+  go-tests = true
+  unused-packages = true

main.go

@@ -0,0 +1,212 @@
+package main
+
+import (
+	"bytes"
+	"crypto/tls"
+	"encoding/json"
+	"flag"
+	"fmt"
+	"github.com/google/uuid"
+	"github.com/prometheus/prometheus/pkg/labels"
+	"io"
+	"io/ioutil"
+	"log"
+	"net/http"
+	"os"
+	"strings"
+	"time"
+)
+
+var logger = log.New(os.Stdout,"[Prmetheus Alert]",log.LstdFlags)
+
+const(
+	ok = 0
+	ng = 1
+
+	indice = "prometheus_alert"
+	template = "prometheus_alert_template"
+)
+
+type AlertHandler struct {
+}
+
+func (th *AlertHandler)ServeHTTP(w http.ResponseWriter, r *http.Request){
+	if r.Body == nil {
+		logger.Println("Got empty request body")
+		return
+	}
+	defer r.Body.Close()
+
+	reqbody, err := ioutil.ReadAll(r.Body)
+	if err != nil {
+		logger.Println(err)
+		return
+	}
+
+	var alerts Alerts
+	json.Unmarshal(reqbody,&alerts)
+
+	//Check indice and template
+	if ok != DoRequest(http.MethodGet,esurl+"/"+indice,nil){
+		logger.Println("Not found indice:",indice,",begin to create...")
+		ret := DoRequest(http.MethodPut,esurl+"/"+indice,nil)
+		if ok != ret{
+			logger.Println("Create indice failed")
+			return
+		}
+	}
+
+	if ok != DoRequest(http.MethodGet,esurl+"/_template/" + template,nil){
+		logger.Println("Not found template:",template,",begin to create...")
+		reqbody := []byte(`
+{
+  "index_patterns": ["prometheus_alert*"],
+  "settings": {
+    "number_of_shards": 1,
+    "number_of_replicas": 1,
+    "index.refresh_interval": "10s",
+    "index.query.default_field": "groupLabels.alertname"
+  },
+  "mappings": {
+    "_doc": {
+      "properties": {
+        "@timestamp": {
+          "type": "date",
+          "doc_values": true
+        }
+      },
+      "dynamic_templates": [
+        {
+          "string_fields": {
+            "match": "*",
+            "match_mapping_type": "string",
+            "mapping": {
+              "type": "text",
+              "fields": {
+                "raw": {
+                  "type":  "keyword",
+                  "ignore_above": 256
+                }
+              }
+            }
+          }
+        }
+      ]
+    }
+  }
+}`)
+		newBody := bytes.NewBuffer(reqbody)
+		ret := DoRequest(http.MethodPost,esurl+"/_template/" + template,newBody)
+		if ok != ret{
+			logger.Println("Create template failed")
+			return
+		}
+	}
+
+	//Send alerts
+	for _,alert := range alerts{
+		jsonalert, err := json.Marshal(alert)
+		if nil != err {
+			logger.Println("Transfor alert to json error,",err)
+		}
+		ret := DoRequest(http.MethodPut,esurl+"/prometheus_alert/_doc/"+uuid.New().String(),bytes.NewBuffer(jsonalert))
+		if ok != ret{
+			logger.Println("Put alter failed,alert:",alerts)
+			return
+		}
+	}
+}
+
+
+//Do request base on elasticsearch of version 6.8
+func DoRequest(method,url string, body io.Reader) int8{
+	tr := &http.Transport{
+		TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
+	}
+
+	client := &http.Client{Transport:tr}
+	req, err := http.NewRequest(method, url, body)
+	if nil != err{
+		logger.Println(err)
+		return ng
+	}
+	req.SetBasicAuth(esusername, espasswd)
+	//We just use GET/PUT/POST
+	req.Header.Set("Content-Type", "application/json")
+	resp,err := client.Do(req)
+	if nil != err{
+		logger.Println(err)
+		return ng
+	}
+	defer resp.Body.Close()
+
+	if 200 <= resp.StatusCode && resp.StatusCode < 300 {
+		return ok
+	}
+
+	logger.Println("Request error,resp.StatusCode =",resp.StatusCode,req)
+	return ng
+}
+
+
+var (
+	esusername, espasswd , esurl string
+	h bool
+)
+
+func usage() {
+	fmt.Fprintf(os.Stderr, `
+Usage: prometheusalert2es [options...]
+
+prometheusalert2es --esurl=${url} --esusername=${username} --espasswd=${passwd}
+
+Options:
+`)
+	flag.PrintDefaults()
+}
+
+func main(){
+	flag.BoolVar(&h, "h", false, "Help info")
+	flag.StringVar(&esusername, "esusername", "", "Elasticsearch username")
+	flag.StringVar(&espasswd, "espasswd", "", "Elasticsearch password")
+	flag.StringVar(&esurl, "esurl", "", "Elasticsearch url")
+	flag.Parse()
+
+	if h{
+		usage()
+		return
+	}
+
+	if "" == esurl || "" == esusername || "" == espasswd {
+		logger.Println("Must specific esusername, espasswd and esurl")
+		usage()
+		return
+	}
+
+	esurl = strings.TrimRight(esurl, "/")
+	serverHandler := http.NewServeMux()
+	serverHandler.Handle("/", &AlertHandler{})
+
+	logger.Println("Start listen on ")
+	http.ListenAndServe(":8888",serverHandler)
+
+}
+
+
+//Copy from github.com/prometheus/prometheus/notifier/notifier.go
+//Different version of prometheus may hava different Alert struct, be careful
+type Alerts []Alert
+// Alert is a generic representation of an alert in the Prometheus eco-system.
+type Alert struct {
+	// Label value pairs for purpose of aggregation, matching, and disposition
+	// dispatching. This must minimally include an "alertname" label.
+	Labels labels.Labels `json:"labels"`
+
+	// Extra key/value information which does not define alert identity.
+	Annotations labels.Labels `json:"annotations"`
+
+	// The known time range for this alert. Both ends are optional.
+	StartsAt     time.Time `json:"startsAt,omitempty"`
+	EndsAt       time.Time `json:"endsAt,omitempty"`
+	GeneratorURL string    `json:"generatorURL,omitempty"`
+}