Add filter configuration and improve URL handling

_meta/beat.yml

@@ -42,3 +42,15 @@ icingabeat:
     types:
       - CheckResult
       - StateChange
+
+    # Event streams can be filtered by attributes using the prefix 'event.'
+    #
+    # Example for the CheckResult type with the exit_code set to 2:
+    # filter: "event.check_result.exit_status==2"
+    #
+    # Example for the CheckResult type with the service matching the string
+    # pattern "mysql*":
+    # filter: 'match("mysql*", event.service)'
+    #
+    # Comment out this line if you don't want to set a filter at all
+    filter: ""

beater/eventstream.go

@@ -4,7 +4,8 @@ import (
 	"bufio"
 	"encoding/json"
 	"io"
-	"strings"
+	"net/url"
+	"strconv"
 	"sync"
 	"time"
 
@@ -36,15 +37,34 @@ func NewEventstream(bt *Icingabeat, cfg config.Config) *Eventstream {
 
 // Run evenstream receiver
 func (es *Eventstream) Run() error {
-	types := strings.Join(es.config.Eventstream.Types, "&types=")
-	logp.Info(types)
+	queue := "icingabeat"
+	host := es.config.Host + ":" + strconv.Itoa(es.config.Port)
+	var URL *url.URL
+
+	URL, err := url.Parse("https://" + host)
+	if err != nil {
+		logp.Info("Invalid request URL")
+	}
+
+	URL.Path += "/v1/events/"
+
+	parameters := url.Values{}
+	parameters.Add("queue", queue)
+
+	if es.config.Eventstream.Filter != "" {
+		parameters.Add("filter", es.config.Eventstream.Filter)
+	}
+
+	for _, eventType := range es.config.Eventstream.Types {
+		parameters.Add("types", eventType)
+	}
+
+	URL.RawQuery = parameters.Encode()
+
 	for {
 
 		ticker := time.NewTicker(es.config.RetryInterval)
-		response, responseErr := requestURL(
-			es.icingabeat,
-			"POST",
-			"/v1/events?queue=icingabeat&types="+types)
+		response, responseErr := requestURL(es.icingabeat, "POST", URL)
 
 		if responseErr == nil {
 			reader := bufio.NewReader(response.Body)

beater/http.go

@@ -3,13 +3,13 @@ package beater
 import (
 	"crypto/tls"
 	"errors"
-	"fmt"
 	"net/http"
+	"net/url"
 
 	"github.com/elastic/beats/libbeat/logp"
 )
 
-func requestURL(bt *Icingabeat, method, path string) (*http.Response, error) {
+func requestURL(bt *Icingabeat, method string, URL *url.URL) (*http.Response, error) {
 	transport := &http.Transport{
 		TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
 	}
@@ -18,12 +18,9 @@ func requestURL(bt *Icingabeat, method, path string) (*http.Response, error) {
 		Transport: transport,
 	}
 
-	url := fmt.Sprintf(
-		"https://%s:%v%s",
-		bt.config.Host,
-		bt.config.Port,
-		path)
-	request, err := http.NewRequest(method, url, nil)
+	logp.Info("Requested URL: %v", URL.String())
+
+	request, err := http.NewRequest(method, URL.String(), nil)
 
 	if err != nil {
 		logp.Info("Request: %v", err)

beater/icingabeat.go

@@ -40,7 +40,6 @@ func (bt *Icingabeat) Run(b *beat.Beat) error {
 	bt.client = b.Publisher.Connect()
 
 	eventstream = NewEventstream(bt, bt.config)
-	logp.Info("hostname: %v", bt.config.Host)
 	go eventstream.Run()
 
 	for {

config/config.go

@@ -18,6 +18,7 @@ type Config struct {
 // EventstreamConfig optoins
 type EventstreamConfig struct {
 	Types  []string `config:"types"`
+	Filter string   `config:"filter"`
 }
 
 // DefaultConfig values

icingabeat.full.yml

@@ -43,6 +43,18 @@ icingabeat:
       - CheckResult
       - StateChange
 
+    # Event streams can be filtered by attributes using the prefix event.
+    #
+    # Example for the CheckResult type with the exit_code set to 2:
+    # filter: "event.check_result.exit_status==2"
+    #
+    # Example for the CheckResult type with the service matching the string
+    # pattern "mysql*":
+    # filter: 'match("mysql*", event.service)'
+    #
+    # Comment out this line if you don't want to set a filter at all
+    filter: ""
+
 #================================ General ======================================
 
 # The name of the shipper that publishes the network data. It can be used to group

icingabeat.yml

@@ -43,6 +43,18 @@ icingabeat:
       - CheckResult
       - StateChange
 
+    # Event streams can be filtered by attributes using the prefix event.
+    #
+    # Example for the CheckResult type with the exit_code set to 2:
+    # filter: "event.check_result.exit_status==2"
+    #
+    # Example for the CheckResult type with the service matching the string
+    # pattern "mysql*":
+    # filter: 'match("mysql*", event.service)'
+    #
+    # Comment out this line if you don't want to set a filter at all
+    filter: ""
+
 #================================ General =====================================
 
 # The name of the shipper that publishes the network data. It can be used to group