THRIFT-3292 Integrate new Zlib transport for Go into test suite
Client: Go
Patch: Paul Magrath <paul@swiftkey.com>
This closes #580
diff --git a/test/go/src/common/client.go b/test/go/src/common/client.go
index 267273e..e55dc6d 100644
--- a/test/go/src/common/client.go
+++ b/test/go/src/common/client.go
@@ -20,6 +20,7 @@
package common
import (
+ "compress/zlib"
"crypto/tls"
"flag"
"fmt"
@@ -82,6 +83,11 @@
trans = thrift.NewTFramedTransport(trans)
case "buffered":
trans = thrift.NewTBufferedTransport(trans, 8192)
+ case "zlib":
+ trans, err = thrift.NewTZlibTransport(trans, zlib.BestCompression)
+ if err != nil {
+ return nil, err
+ }
case "":
trans = trans
default:
diff --git a/test/go/src/common/server.go b/test/go/src/common/server.go
index d354b32..dc380b2 100644
--- a/test/go/src/common/server.go
+++ b/test/go/src/common/server.go
@@ -20,6 +20,7 @@
package common
import (
+ "compress/zlib"
"crypto/tls"
"flag"
"fmt"
@@ -99,6 +100,8 @@
transportFactory = thrift.NewTFramedTransportFactory(transportFactory)
case "buffered":
transportFactory = thrift.NewTBufferedTransportFactory(8192)
+ case "zlib":
+ transportFactory = thrift.NewTZlibTransportFactory(zlib.BestCompression)
case "":
transportFactory = thrift.NewTTransportFactory()
default: