Update memory_buffer.go

donnie4w · Nov 22, 2023 · ae10cf5 · ae10cf5
1 parent 15ee811
commit ae10cf5
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/thrift/memory_buffer.go b/thrift/memory_buffer.go
@@ -75,6 +75,14 @@ func (p *TMemoryBuffer) Flush(ctx context.Context) error {
 	return nil
 }
 
+func (p *TMemoryBuffer) Bytes() []byte {
+	return p.Buffer.Bytes()
+}
+
+func (p *TMemoryBuffer) Write(buf []byte) (n int, e error) {
+	return p.Buffer.Write(buf)
+}
+
 func (p *TMemoryBuffer) RemainingBytes() (num_bytes uint64) {
 	return uint64(p.Buffer.Len())
-}
+}