mirror of
https://github.com/apernet/hysteria.git
synced 2025-04-05 05:27:40 +03:00
25 lines
393 B
Go
25 lines
393 B
Go
package utils
|
|
|
|
import (
|
|
"io"
|
|
"sync/atomic"
|
|
)
|
|
|
|
const pipeBufferSize = 16384
|
|
|
|
func Pipe(src, dst io.ReadWriter, atomicCounter *uint64) error {
|
|
buf := make([]byte, pipeBufferSize)
|
|
for {
|
|
rn, err := src.Read(buf)
|
|
if rn > 0 {
|
|
wn, err := dst.Write(buf[:rn])
|
|
atomic.AddUint64(atomicCounter, uint64(wn))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|