A mirror service, for better functional tests.

Author: Mathieu Lecarme

event/event.go

@@ -0,0 +1,31 @@
+package event
+
+import (
+	"time"
+)
+
+type Event struct {
+	Ts     time.Time              `json:"ts"`
+	Record map[string]interface{} `json:"record"`
+}
+
+func New(ts time.Time, record map[string]interface{}) Event {
+	return Event{
+		Ts:     ts,
+		Record: record,
+	}
+}
+
+type Events []Event
+
+func (e Events) Len() int {
+	return len(e)
+}
+
+func (e Events) Less(i, j int) bool {
+	return e[i].Ts.After(e[j].Ts)
+}
+
+func (e Events) Swap(i, j int) {
+	e[i], e[j] = e[j], e[i]
+}

main.go

@@ -2,18 +2,15 @@ package main
 
 import (
 	"fmt"
+	"net/http"
 	"os"
-	"time"
 
+	"github.com/factorysh/fluent-server/mirror"
 	"github.com/factorysh/fluent-server/server"
 )
 
-func handler(tag string, ts *time.Time, record map[string]interface{}) error {
-	fmt.Println(tag, ts, record)
-	return nil
-}
-
 func main() {
+	m := mirror.New()
 	var s *server.Server
 	caCrt := os.Getenv("CA_CRT")
 	if caCrt != "" {
@@ -26,9 +23,14 @@ ca.crt: %s
 server.crt: %s
 server.key: %s
 `, caCrt, os.Getenv("SRV_CRT"), os.Getenv("SRV_KEY"))
-		s = server.NewTLS(handler, cfg)
+		s = server.NewTLS(m.Handler, cfg)
 	} else {
-		s = server.New(handler)
+		s = server.New(m.Handler)
+	}
+	ll := os.Getenv("MIRROR_LISTEN")
+	if ll != "" {
+		go http.ListenAndServe(ll, m)
+		fmt.Println("mirror listen ", ll)
 	}
 
 	l := os.Getenv("LISTEN")

mirror/mirror.go

@@ -0,0 +1,46 @@
+package mirror
+
+import (
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"sync"
+	"time"
+
+	"github.com/factorysh/fluent-server/event"
+)
+
+type Mirror struct {
+	lock   *sync.Mutex
+	events map[string]event.Events
+}
+
+func New() *Mirror {
+	return &Mirror{
+		lock:   &sync.Mutex{},
+		events: make(map[string]event.Events),
+	}
+}
+
+func (t *Mirror) Handler(tag string, ts *time.Time, record map[string]interface{}) error {
+	fmt.Println(tag, ts, record)
+	t.lock.Lock()
+	defer t.lock.Unlock()
+	evts, ok := t.events[tag]
+	if !ok {
+		t.events[tag] = event.Events{
+			event.New(*ts, record),
+		}
+	} else {
+		t.events[tag] = append(evts, event.New(*ts, record))
+	}
+	return nil
+}
+
+func (t *Mirror) ServeHTTP(w http.ResponseWriter, r *http.Request) {
+	w.Header().Set("Content-Type", "application/json")
+	err := json.NewEncoder(w).Encode(t.events)
+	if err != nil {
+		panic(err)
+	}
+}