d7f3a5f811
* internal,lib,uplink: add monkit task to missing places Change-Id: I490053eee4ed517502f9fe00c6394f0095bd13d0 * Include Monkit * Add missing context * Another missing ctx * More ctx missing * Linting * go imports Change-Id: Ibf0ed072eba339f027727ed8039f7bce1f223fa7 * fix semantic merge conflict Change-Id: I67fb1f4e7b6cd5e89d69987ed7b3966b7d30ee37
33 lines
675 B
Go
33 lines
675 B
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information
|
|
|
|
package sync2
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
|
|
monkit "gopkg.in/spacemonkeygo/monkit.v2"
|
|
)
|
|
|
|
var mon = monkit.Package()
|
|
|
|
type readerFunc func(p []byte) (n int, err error)
|
|
|
|
func (rf readerFunc) Read(p []byte) (n int, err error) { return rf(p) }
|
|
|
|
// Copy implements copying with cancellation
|
|
func Copy(ctx context.Context, dst io.Writer, src io.Reader) (written int64, err error) {
|
|
defer mon.Task()(&ctx)(&err)
|
|
written, err = io.Copy(dst, readerFunc(func(p []byte) (int, error) {
|
|
select {
|
|
case <-ctx.Done():
|
|
return 0, ctx.Err()
|
|
default:
|
|
return src.Read(p)
|
|
}
|
|
}))
|
|
|
|
return written, err
|
|
}
|