1 files changed, 10 insertions, 0 deletions
diff --git a/drive/timeout_reader.go b/drive/timeout_reader.go
index ba2bb83..878911b 100644
--- a/drive/timeout_reader.go
+++ b/drive/timeout_reader.go
@@ -10,6 +10,16 @@ import (
 const MaxIdleTimeout = time.Second * 120
 const TimeoutTimerInterval = time.Second * 10
 
+type timeoutReaderWrapper func(io.Reader) io.Reader
+
+func getTimeoutReaderWrapperContext() (timeoutReaderWrapper, context.Context) {
+    ctx, cancel := context.WithCancel(context.TODO())
+    wrapper := func(r io.Reader) io.Reader {
+         return getTimeoutReader(r, cancel)
+    }
+    return wrapper, ctx
+}
+
 func getTimeoutReaderContext(r io.Reader) (io.Reader, context.Context) {
     ctx, cancel := context.WithCancel(context.TODO())
     return getTimeoutReader(r, cancel), ctx