mirror of
https://github.com/TeaOSLab/EdgeNode.git
synced 2025-12-16 07:46:36 +08:00
重构对HTTP请求的处理方法:缓存、压缩、WebP、限速
This commit is contained in:
26
internal/utils/readers/bytes_counter_reader.go
Normal file
26
internal/utils/readers/bytes_counter_reader.go
Normal file
@@ -0,0 +1,26 @@
|
||||
// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
|
||||
|
||||
package readers
|
||||
|
||||
import "io"
|
||||
|
||||
type BytesCounterReader struct {
|
||||
rawReader io.Reader
|
||||
count int64
|
||||
}
|
||||
|
||||
func NewBytesCounterReader(rawReader io.Reader) *BytesCounterReader {
|
||||
return &BytesCounterReader{
|
||||
rawReader: rawReader,
|
||||
}
|
||||
}
|
||||
|
||||
func (this *BytesCounterReader) Read(p []byte) (n int, err error) {
|
||||
n, err = this.rawReader.Read(p)
|
||||
this.count += int64(n)
|
||||
return
|
||||
}
|
||||
|
||||
func (this *BytesCounterReader) TotalBytes() int64 {
|
||||
return this.count
|
||||
}
|
||||
Reference in New Issue
Block a user