save

history/history.go

@@ -29,17 +29,19 @@ var quit chan struct{}
 
 func init() {
 	Tunnel = make(chan *History)
-	quit = make(chan struct{})
+	quit = make(chan struct{}, 1)
 }
 
 func Start(r Receiver) {
 	go func() {
+		for {
 			select {
 			case h := <-Tunnel:
 				r(h)
 			case <-quit:
 				return
 			}
+		}
 	}()
 }
 
@@ -47,6 +49,10 @@ func Stop() {
 	quit <- struct{}{}
 }
 
+func Send(h *History) {
+	Tunnel <- h
+}
+
 type Receiver func(*History)
 
 var DBReceiver Receiver = func(h *History) {

history/history_test.go

@@ -3,8 +3,30 @@ package history
 import (
 	"testing"
 	"time"
+
+	"kumoly.io/kumoly/app/util"
 )
 
 func TestHistory(t *testing.T) {
-	time.Sleep(time.Second * 30)
+	t.Log("start")
+	Start(func(h *History) {
+		t.Logf("%+v\n", h)
+	})
+	go func() {
+		for {
+			Send(&History{
+				Module:  "test",
+				Type:    INFO,
+				Message: "testing",
+				Body:    time.Now().String(),
+				Caller:  util.Caller(2),
+			})
+			time.Sleep(time.Second)
+		}
+	}()
+
+	time.Sleep(time.Second * 20)
+	Stop()
+	t.Log("stoped")
+	time.Sleep(time.Second * 2)
 }