mirror of https://github.com/XTLS/Xray-core
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
25 lines
514 B
25 lines
514 B
package dispatcher |
|
|
|
import ( |
|
"github.com/xtls/xray-core/common" |
|
"github.com/xtls/xray-core/common/buf" |
|
"github.com/xtls/xray-core/features/stats" |
|
) |
|
|
|
type SizeStatWriter struct { |
|
Counter stats.Counter |
|
Writer buf.Writer |
|
} |
|
|
|
func (w *SizeStatWriter) WriteMultiBuffer(mb buf.MultiBuffer) error { |
|
w.Counter.Add(int64(mb.Len())) |
|
return w.Writer.WriteMultiBuffer(mb) |
|
} |
|
|
|
func (w *SizeStatWriter) Close() error { |
|
return common.Close(w.Writer) |
|
} |
|
|
|
func (w *SizeStatWriter) Interrupt() { |
|
common.Interrupt(w.Writer) |
|
}
|
|
|