fix: client wait bug

core/client.go

@@ -5,6 +5,7 @@ import (
 	"context"
 	"errors"
 	"fmt"
+	"io"
 	"reflect"
 	"runtime"
 	"time"
@@ -266,6 +267,12 @@ func (c *Client) handleFrameError(err error, reconnection chan<- struct{}) {
 
 	c.errorfn(err)
 
+	// exit client program if stream has be closed.
+	if err == io.EOF {
+		c.ctxCancel()
+		return
+	}
+
 	// always attempting to reconnect if an error is encountered,
 	// the error is mostly network error.
 	select {

core/metadata_test.go

@@ -31,7 +31,6 @@ func TestMetadata(t *testing.T) {
 func TestMetadataSlogAttr(t *testing.T) {
 	md := metadata.New(map[string]string{
 		"aaaa": "bbbb",
-		"cccc": "dddd",
 	})
 
 	buf := bytes.NewBuffer(nil)
@@ -50,5 +49,5 @@ func TestMetadataSlogAttr(t *testing.T) {
 
 	logger.Debug("test metadata", MetadataSlogAttr(md))
 
-	assert.Equal(t, "level=DEBUG msg=\"test metadata\" metadata.aaaa=bbbb metadata.cccc=dddd\n", buf.String())
+	assert.Equal(t, "level=DEBUG msg=\"test metadata\" metadata.aaaa=bbbb\n", buf.String())
 }