update client

client OnExit -> OnDone

Author: stefanhengl

internal/streaming/client.go

@@ -31,6 +31,7 @@ type Decoder struct {
 	OnAlert    func(*EventAlert)
 	OnError    func(*EventError)
 	OnUnknown  func(event, data []byte)
+	OnDone     func()
 }
 
 func (rr Decoder) ReadAll(r io.Reader) error {
@@ -125,6 +126,9 @@ func (rr Decoder) ReadAll(r io.Reader) error {
 			rr.OnError(&d)
 		} else if bytes.Equal(event, []byte("done")) {
 			// Always the last event
+			if rr.OnDone != nil {
+				rr.OnDone()
+			}
 			break
 		} else {
 			if rr.OnUnknown == nil {

internal/streaming/client_test.go

@@ -106,6 +106,11 @@ func TestDecoder(t *testing.T) {
 		OnUnknown: func(event, data []byte) {
 			t.Fatalf("got unexpected event: %s %s", event, data)
 		},
+		OnDone: func() {
+			if len(got) != len(want) {
+				t.Fatalf("OnDone was called before the last event was processed.")
+			}
+		},
 	}.ReadAll(resp.Body)
 	if err != nil {
 		t.Fatal(err)