|
|
@@ -1,9 +1,6 @@
|
|
|
package buf
|
|
|
|
|
|
-import (
|
|
|
- "io"
|
|
|
- "sync"
|
|
|
-)
|
|
|
+import "io"
|
|
|
|
|
|
// BufferToBytesWriter is a Writer that writes alloc.Buffer into underlying writer.
|
|
|
type BufferToBytesWriter struct {
|
|
|
@@ -32,17 +29,10 @@ func (v *BufferToBytesWriter) Release() {
|
|
|
}
|
|
|
|
|
|
type BytesToBufferWriter struct {
|
|
|
- sync.Mutex
|
|
|
writer Writer
|
|
|
}
|
|
|
|
|
|
func (v *BytesToBufferWriter) Write(payload []byte) (int, error) {
|
|
|
- v.Lock()
|
|
|
- defer v.Unlock()
|
|
|
- if v.writer == nil {
|
|
|
- return 0, io.ErrClosedPipe
|
|
|
- }
|
|
|
-
|
|
|
bytesWritten := 0
|
|
|
size := len(payload)
|
|
|
for size > 0 {
|
|
|
@@ -62,8 +52,5 @@ func (v *BytesToBufferWriter) Write(payload []byte) (int, error) {
|
|
|
|
|
|
// Release implements Releasable.Release()
|
|
|
func (v *BytesToBufferWriter) Release() {
|
|
|
- v.Lock()
|
|
|
v.writer.Release()
|
|
|
- v.writer = nil
|
|
|
- v.Unlock()
|
|
|
}
|