2016-02-26 17:45:33 -05:00
|
|
|
package io
|
|
|
|
|
|
|
|
import (
|
|
|
|
"io"
|
2016-05-20 07:56:19 -04:00
|
|
|
"sync"
|
2016-02-26 17:45:33 -05:00
|
|
|
|
|
|
|
"github.com/v2ray/v2ray-core/common/alloc"
|
|
|
|
)
|
|
|
|
|
|
|
|
type BufferedWriter struct {
|
2016-05-20 07:56:19 -04:00
|
|
|
sync.Mutex
|
2016-02-26 17:45:33 -05:00
|
|
|
writer io.Writer
|
|
|
|
buffer *alloc.Buffer
|
|
|
|
cached bool
|
|
|
|
}
|
|
|
|
|
|
|
|
func NewBufferedWriter(rawWriter io.Writer) *BufferedWriter {
|
|
|
|
return &BufferedWriter{
|
|
|
|
writer: rawWriter,
|
|
|
|
buffer: alloc.NewBuffer().Clear(),
|
|
|
|
cached: true,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (this *BufferedWriter) Write(b []byte) (int, error) {
|
2016-05-20 07:56:19 -04:00
|
|
|
this.Lock()
|
|
|
|
defer this.Unlock()
|
|
|
|
|
|
|
|
if this.writer == nil {
|
|
|
|
return 0, io.EOF
|
|
|
|
}
|
|
|
|
|
2016-02-26 17:45:33 -05:00
|
|
|
if !this.cached {
|
|
|
|
return this.writer.Write(b)
|
|
|
|
}
|
|
|
|
nBytes, _ := this.buffer.Write(b)
|
|
|
|
if this.buffer.IsFull() {
|
2016-02-27 05:02:42 -05:00
|
|
|
err := this.Flush()
|
2016-02-26 17:45:33 -05:00
|
|
|
if err != nil {
|
|
|
|
return nBytes, err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return nBytes, nil
|
|
|
|
}
|
|
|
|
|
2016-02-27 05:02:42 -05:00
|
|
|
func (this *BufferedWriter) Flush() error {
|
2016-05-20 07:56:19 -04:00
|
|
|
this.Lock()
|
|
|
|
defer this.Unlock()
|
|
|
|
if this.writer == nil {
|
|
|
|
return io.EOF
|
|
|
|
}
|
|
|
|
|
2016-05-12 20:20:07 -04:00
|
|
|
defer this.buffer.Clear()
|
|
|
|
for !this.buffer.IsEmpty() {
|
|
|
|
nBytes, err := this.writer.Write(this.buffer.Value)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2016-02-26 17:45:33 -05:00
|
|
|
this.buffer.SliceFrom(nBytes)
|
|
|
|
}
|
2016-05-12 20:20:07 -04:00
|
|
|
return nil
|
2016-02-26 17:45:33 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
func (this *BufferedWriter) Cached() bool {
|
|
|
|
return this.cached
|
|
|
|
}
|
|
|
|
|
|
|
|
func (this *BufferedWriter) SetCached(cached bool) {
|
|
|
|
this.cached = cached
|
|
|
|
if !cached && !this.buffer.IsEmpty() {
|
2016-02-27 05:02:42 -05:00
|
|
|
this.Flush()
|
2016-02-26 17:45:33 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (this *BufferedWriter) Release() {
|
2016-05-20 07:56:19 -04:00
|
|
|
this.Lock()
|
|
|
|
defer this.Unlock()
|
|
|
|
|
2016-05-12 20:20:07 -04:00
|
|
|
this.Flush()
|
2016-02-26 17:45:33 -05:00
|
|
|
this.buffer.Release()
|
2016-03-10 10:56:19 -05:00
|
|
|
this.buffer = nil
|
|
|
|
this.writer = nil
|
2016-02-26 17:45:33 -05:00
|
|
|
}
|