fix(generic): binary generic server returning biz err failed (#1648)

cloudwego · Dec 23, 2024 · c50055a · c50055a
1 parent 8a7d120
commit c50055a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pkg/generic/thrift/binary.go b/pkg/generic/thrift/binary.go
@@ -20,6 +20,7 @@ import (
 	"context"
 
 	"github.com/cloudwego/gopkg/bufiox"
+	"github.com/cloudwego/gopkg/protocol/thrift"
 	"github.com/cloudwego/gopkg/protocol/thrift/base"
 )
 
@@ -31,5 +32,10 @@ func NewWriteBinary() *WriteBinary {
 }
 
 func (w *WriteBinary) Write(ctx context.Context, out bufiox.Writer, msg interface{}, method string, isClient bool, requestBase *base.Base) error {
+	bw := thrift.NewBufferWriter(out)
+	defer bw.Recycle()
+	if err := bw.WriteFieldStop(); err != nil {
+		return err
+	}
 	return nil
 }