Update zlib_transport.go

donnie4w · Mar 25, 2024 · 90ec2e1 · 90ec2e1
1 parent c247fad
commit 90ec2e1
Showing 1 changed file with 109 additions and 109 deletions.
diff --git a/thrift/zlib_transport.go b/thrift/zlib_transport.go
@@ -26,112 +26,112 @@ import (
 )
 
 // TZlibTransportFactory is a factory for TZlibTransport instances
-type TZlibTransportFactory struct {
-	level   int
-	factory TTransportFactory
-}
-
-// TZlibTransport is a TTransport implementation that makes use of zlib compression.
-type TZlibTransport struct {
-	reader    io.ReadCloser
-	transport TTransport
-	writer    *zlib.Writer
-}
-
-// GetTransport constructs a new instance of NewTZlibTransport
-func (p *TZlibTransportFactory) GetTransport(trans TTransport) (TTransport, error) {
-	if p.factory != nil {
-		// wrap other factory
-		var err error
-		trans, err = p.factory.GetTransport(trans)
-		if err != nil {
-			return nil, err
-		}
-	}
-	return NewTZlibTransport(trans, p.level)
-}
-
-// NewTZlibTransportFactory constructs a new instance of NewTZlibTransportFactory
-func NewTZlibTransportFactory(level int) *TZlibTransportFactory {
-	return &TZlibTransportFactory{level: level, factory: nil}
-}
-
-// NewTZlibTransportFactoryWithFactory constructs a new instance of TZlibTransportFactory
-// as a wrapper over existing transport factory
-func NewTZlibTransportFactoryWithFactory(level int, factory TTransportFactory) *TZlibTransportFactory {
-	return &TZlibTransportFactory{level: level, factory: factory}
-}
-
-// NewTZlibTransport constructs a new instance of TZlibTransport
-func NewTZlibTransport(trans TTransport, level int) (*TZlibTransport, error) {
-	w, err := zlib.NewWriterLevel(trans, level)
-	if err != nil {
-		return nil, err
-	}
-
-	return &TZlibTransport{
-		writer:    w,
-		transport: trans,
-	}, nil
-}
-
-// Close closes the reader and writer (flushing any unwritten data) and closes
-// the underlying transport.
-func (z *TZlibTransport) Close() error {
-	if z.reader != nil {
-		if err := z.reader.Close(); err != nil {
-			return err
-		}
-	}
-	if err := z.writer.Close(); err != nil {
-		return err
-	}
-	return z.transport.Close()
-}
-
-// Flush flushes the writer and its underlying transport.
-func (z *TZlibTransport) Flush(ctx context.Context) error {
-	if err := z.writer.Flush(); err != nil {
-		return err
-	}
-	return z.transport.Flush(ctx)
-}
-
-// IsOpen returns true if the transport is open
-func (z *TZlibTransport) IsOpen() bool {
-	return z.transport.IsOpen()
-}
-
-// Open opens the transport for communication
-func (z *TZlibTransport) Open() error {
-	return z.transport.Open()
-}
-
-func (z *TZlibTransport) Read(p []byte) (int, error) {
-	if z.reader == nil {
-		r, err := zlib.NewReader(z.transport)
-		if err != nil {
-			return 0, NewTTransportExceptionFromError(err)
-		}
-		z.reader = r
-	}
-
-	return z.reader.Read(p)
-}
-
-// RemainingBytes returns the size in bytes of the data that is still to be
-// read.
-func (z *TZlibTransport) RemainingBytes() uint64 {
-	return z.transport.RemainingBytes()
-}
-
-func (z *TZlibTransport) Write(p []byte) (int, error) {
-	return z.writer.Write(p)
-}
-
-// SetTConfiguration implements TConfigurationSetter for propagation.
-func (z *TZlibTransport) SetTConfiguration(conf *TConfiguration) {
-	PropagateTConfiguration(z.transport, conf)
-}
-
-var _ TConfigurationSetter = (*TZlibTransport)(nil)
+ type TZlibTransportFactory struct {
+	 level   int
+	 factory TTransportFactory
+ }
+ 
+ // TZlibTransport is a TTransport implementation that makes use of zlib compression.
+ type TZlibTransport struct {
+	 reader    io.ReadCloser
+	 transport TTransport
+	 writer    *zlib.Writer
+ }
+ 
+ // GetTransport constructs a new instance of NewTZlibTransport
+ func (p *TZlibTransportFactory) GetTransport(trans TTransport) (TTransport, error) {
+	 if p.factory != nil {
+		 // wrap other factory
+		 var err error
+		 trans, err = p.factory.GetTransport(trans)
+		 if err != nil {
+			 return nil, err
+		 }
+	 }
+	 return NewTZlibTransport(trans, p.level)
+ }
+ 
+ // NewTZlibTransportFactory constructs a new instance of NewTZlibTransportFactory
+ func NewTZlibTransportFactory(level int) *TZlibTransportFactory {
+	 return &TZlibTransportFactory{level: level, factory: nil}
+ }
+ 
+ // NewTZlibTransportFactoryWithFactory constructs a new instance of TZlibTransportFactory
+ // as a wrapper over existing transport factory
+ func NewTZlibTransportFactoryWithFactory(level int, factory TTransportFactory) *TZlibTransportFactory {
+	 return &TZlibTransportFactory{level: level, factory: factory}
+ }
+ 
+ // NewTZlibTransport constructs a new instance of TZlibTransport
+ func NewTZlibTransport(trans TTransport, level int) (*TZlibTransport, error) {
+	 w, err := zlib.NewWriterLevel(trans, level)
+	 if err != nil {
+		 return nil, err
+	 }
+ 
+	 return &TZlibTransport{
+		 writer:    w,
+		 transport: trans,
+	 }, nil
+ }
+ 
+ // Close closes the reader and writer (flushing any unwritten data) and closes
+ // the underlying transport.
+ func (z *TZlibTransport) Close() error {
+	 if z.reader != nil {
+		 if err := z.reader.Close(); err != nil {
+			 return err
+		 }
+	 }
+	 if err := z.writer.Close(); err != nil {
+		 return err
+	 }
+	 return z.transport.Close()
+ }
+ 
+ // Flush flushes the writer and its underlying transport.
+ func (z *TZlibTransport) Flush(ctx context.Context) error {
+	 if err := z.writer.Flush(); err != nil {
+		 return err
+	 }
+	 return z.transport.Flush(ctx)
+ }
+ 
+ // IsOpen returns true if the transport is open
+ func (z *TZlibTransport) IsOpen() bool {
+	 return z.transport.IsOpen()
+ }
+ 
+ // Open opens the transport for communication
+ func (z *TZlibTransport) Open() error {
+	 return z.transport.Open()
+ }
+ 
+ func (z *TZlibTransport) Read(p []byte) (int, error) {
+	 if z.reader == nil {
+		 r, err := zlib.NewReader(z.transport)
+		 if err != nil {
+			 return 0, NewTTransportExceptionFromError(err)
+		 }
+		 z.reader = r
+	 }
+ 
+	 return z.reader.Read(p)
+ }
+ 
+ // RemainingBytes returns the size in bytes of the data that is still to be
+ // read.
+ func (z *TZlibTransport) RemainingBytes() uint64 {
+	 return z.transport.RemainingBytes()
+ }
+ 
+ func (z *TZlibTransport) Write(p []byte) (int, error) {
+	 return z.writer.Write(p)
+ }
+ 
+ // SetTConfiguration implements TConfigurationSetter for propagation.
+ func (z *TZlibTransport) SetTConfiguration(conf *TConfiguration) {
+	 PropagateTConfiguration(z.transport, conf)
+ }
+ 
+ var _ TConfigurationSetter = (*TZlibTransport)(nil)