mirror of
https://github.com/TeaOSLab/EdgeNode.git
synced 2025-11-03 15:00:26 +08:00
33 lines
638 B
Go
33 lines
638 B
Go
// Copyright 2022 GoEdge goedge.cdn@gmail.com. All rights reserved.
|
|
|
|
package writers
|
|
|
|
import "io"
|
|
|
|
type BytesCounterWriter struct {
|
|
writer io.Writer
|
|
count int64
|
|
}
|
|
|
|
func NewBytesCounterWriter(rawWriter io.Writer) *BytesCounterWriter {
|
|
return &BytesCounterWriter{writer: rawWriter}
|
|
}
|
|
|
|
func (this *BytesCounterWriter) RawWriter() io.Writer {
|
|
return this.writer
|
|
}
|
|
|
|
func (this *BytesCounterWriter) Write(p []byte) (n int, err error) {
|
|
n, err = this.writer.Write(p)
|
|
this.count += int64(n)
|
|
return
|
|
}
|
|
|
|
func (this *BytesCounterWriter) Close() error {
|
|
return nil
|
|
}
|
|
|
|
func (this *BytesCounterWriter) TotalBytes() int64 {
|
|
return this.count
|
|
}
|