2019-02-01 19:08:21 +00:00
|
|
|
// +build !confonly
|
|
|
|
|
2018-11-03 12:05:23 +00:00
|
|
|
package dispatcher
|
2018-04-16 22:31:10 +00:00
|
|
|
|
|
|
|
import (
|
|
|
|
"v2ray.com/core/common"
|
|
|
|
"v2ray.com/core/common/buf"
|
2018-10-11 19:14:53 +00:00
|
|
|
"v2ray.com/core/features/stats"
|
2018-04-16 22:31:10 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
type SizeStatWriter struct {
|
2018-10-11 19:14:53 +00:00
|
|
|
Counter stats.Counter
|
2018-04-16 22:31:10 +00:00
|
|
|
Writer buf.Writer
|
2021-01-29 17:47:13 +00:00
|
|
|
Record *int64
|
2018-04-16 22:31:10 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func (w *SizeStatWriter) WriteMultiBuffer(mb buf.MultiBuffer) error {
|
2021-01-29 17:47:13 +00:00
|
|
|
bufLen := int64(mb.Len())
|
|
|
|
if w.Record != nil {
|
|
|
|
*w.Record += bufLen
|
|
|
|
}
|
|
|
|
w.Counter.Add(bufLen)
|
2018-04-16 22:31:10 +00:00
|
|
|
return w.Writer.WriteMultiBuffer(mb)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (w *SizeStatWriter) Close() error {
|
|
|
|
return common.Close(w.Writer)
|
|
|
|
}
|
|
|
|
|
2018-12-31 20:25:10 +00:00
|
|
|
func (w *SizeStatWriter) Interrupt() {
|
|
|
|
common.Interrupt(w.Writer)
|
2018-04-16 22:31:10 +00:00
|
|
|
}
|