Give proper timeout error message

1 files changed, 6 insertions, 0 deletions

diff --git a/drive/upload.go b/drive/upload.go
index 6f5edd5..a344b62 100644
--- a/drive/upload.go
+++ b/drive/upload.go
@@ -181,6 +181,9 @@ func (self *Drive) uploadFile(args UploadArgs) (*drive.File, int64, error) {
 
 	f, err := self.service.Files.Create(dstFile).Fields("id", "name", "size", "md5Checksum", "webContentLink").Context(ctx).Media(reader, chunkSize).Do()
 	if err != nil {
+		if isTimeoutError(err) {
+			return nil, 0, fmt.Errorf("Failed to upload file: timeout, no data was transferred for %v", args.Timeout)
+		}
 		return nil, 0, fmt.Errorf("Failed to upload file: %s", err)
 	}
 
@@ -232,6 +235,9 @@ func (self *Drive) UploadStream(args UploadStreamArgs) error {
 
 	f, err := self.service.Files.Create(dstFile).Fields("id", "name", "size", "webContentLink").Context(ctx).Media(reader, chunkSize).Do()
 	if err != nil {
+		if isTimeoutError(err) {
+			return fmt.Errorf("Failed to upload file: timeout, no data was transferred for %v", args.Timeout)
+		}
 		return fmt.Errorf("Failed to upload file: %s", err)
 	}