Function copyWithBufferAndMetrics

pkg/shared/network.go:162–196

copyWithBufferAndMetrics performs optimized copying with metrics tracking

(dst io.Writer, src io.Reader, bufferSize int, recordBytes func(int64))

// copyWithBufferAndMetrics performs optimized copying with metrics tracking
func copyWithBufferAndMetrics(dst io.Writer, src io.Reader, bufferSize int, recordBytes func(int64)) (written int64, err error) {
	buf := make([]byte, bufferSize)
	for {
		nr, er := src.Read(buf)
		if nr > 0 {
			nw, ew := dst.Write(buf[0:nr])
			if nw < 0 || nr < nw {
				nw = 0
				if ew == nil {
					ew = fmt.Errorf("invalid write result")
				}
			}
			written += int64(nw)
			// Record bytes transferred for metrics
			if recordBytes != nil && nw > 0 {
				recordBytes(int64(nw))
			}
			if ew != nil {
				err = ew
				break
			}
			if nr != nw {
				err = io.ErrShortWrite
				break
			}
		}
		if er != nil {
			if er != io.EOF {
				err = er
			}
			break
		}
	}
	return written, err
}

// OptimizedCopyWithContext performs high-performance bidirectional copying between two connections with context support
func OptimizedCopyWithContext(ctx context.Context, dst, src net.Conn) {