Merge pull request #23 from observeinc/fix-retryable-errors

Fixes retryable errors.

Author: georgios-observeinc

retry.go

@@ -5,6 +5,7 @@ package gosnowflake
 import (
 	"bytes"
 	"context"
+	"errors"
 	"fmt"
 	"io"
 	"math"
@@ -390,6 +391,13 @@ func (r *retryHTTP) execute() (res *http.Response, err error) {
 }
 
 func isRetryableError(req *http.Request, res *http.Response, err error) (bool, error) {
+	if errors.Is(err, context.DeadlineExceeded) {
+		// Don't retry and return an unwrapped DeadlineExceeded error
+		return false, context.DeadlineExceeded
+	} else if errors.Is(err, context.Canceled) {
+		// Don't retry and return an unwrapped Canceled error
+		return false, context.Canceled
+	}
 	if err != nil && res == nil { // Failed http connection. Most probably client timeout.
 		return true, err
 	}