mirror of
https://github.com/cmz0228/hysteria-dev.git
synced 2025-06-15 08:49:52 +00:00
26 lines
393 B
Go
26 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
|
|
}
|
|
}
|
|
}
|