Added Close to requestBodyProgress

Azure · Jan 26, 2018 · cb3adc5 · cb3adc5
1 parent aa38b99
commit cb3adc5
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/pipeline/progress.go b/pipeline/progress.go
@@ -42,6 +42,14 @@ func (rbp *requestBodyProgress) Seek(offset int64, whence int) (offsetFromStart
 	return rbp.requestBody.Seek(offset, whence)
 }
 
+// requestBodyProgress supports Close but the underlying stream may not; if it does, Close will close it.
+func (rbp *requestBodyProgress) Close() error {
+	if c, ok := rbp.requestBody.(io.Closer); ok {
+		return c.Close()
+	}
+	return nil
+}
+
 // ********** The following are specific to the response body (a ReadCloser)
 
 // This struct is used when sending a body to the network