mirror of https://github.com/v2ray/v2ray-core
47 lines
925 B
Go
47 lines
925 B
Go
package buf
|
|
|
|
import "io"
|
|
|
|
// BufferToBytesWriter is a Writer that writes alloc.Buffer into underlying writer.
|
|
type BufferToBytesWriter struct {
|
|
writer io.Writer
|
|
}
|
|
|
|
// Write implements Writer.Write(). Write() takes ownership of the given buffer.
|
|
func (v *BufferToBytesWriter) Write(buffer *Buffer) error {
|
|
defer buffer.Release()
|
|
for {
|
|
nBytes, err := v.writer.Write(buffer.Bytes())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if nBytes == buffer.Len() {
|
|
break
|
|
}
|
|
buffer.SliceFrom(nBytes)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
type BytesToBufferWriter struct {
|
|
writer Writer
|
|
}
|
|
|
|
func (v *BytesToBufferWriter) Write(payload []byte) (int, error) {
|
|
bytesWritten := 0
|
|
size := len(payload)
|
|
for size > 0 {
|
|
buffer := New()
|
|
nBytes, _ := buffer.Write(payload)
|
|
size -= nBytes
|
|
payload = payload[nBytes:]
|
|
bytesWritten += nBytes
|
|
err := v.writer.Write(buffer)
|
|
if err != nil {
|
|
return bytesWritten, err
|
|
}
|
|
}
|
|
|
|
return bytesWritten, nil
|
|
}
|