Fix: Send multiple pkgs repetitive by tcp (apache#39)

zaihang365 · Jul 18, 2020 · f3a16c1 · f3a16c1
1 parent 812064c
commit f3a16c1
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/client_test.go b/client_test.go
@@ -140,6 +140,10 @@ func TestTCPClient(t *testing.T) {
 	assert.Equal(t, beforeWriteBytes+20, atomic.LoadUint32(&conn.writeBytes))
 	assert.Nil(t, err)
 	ss.SetCompressType(CompressSnappy)
+	err = ss.WriteBytesArray(pkgs...)
+	assert.Nil(t, err)
+	assert.Equal(t, beforeWritePkgNum+6, atomic.LoadUint32(&conn.writePkgNum))
+	assert.Equal(t, beforeWriteBytes+30, atomic.LoadUint32(&conn.writeBytes))
 	assert.True(t, conn.compress == CompressSnappy)
 
 	clt.Close()

diff --git a/session.go b/session.go
@@ -452,6 +452,7 @@ func (s *session) WriteBytesArray(pkgs ...[]byte) error {
 		if _, err := s.Connection.send(pkgs); err != nil {
 			return perrors.Wrapf(err, "s.Connection.Write(pkgs num:%d)", len(pkgs))
 		}
+		return nil
 	}
 
 	// get len