pgconn.SafeToRetry checks for wrapped errors

Use errors.As instead of type assertion. Port 4e2e7a040579c1999c0766642d836eb28c6e2018 to v5 Credit to tjasko
jackc · May 9, 2024 · 579a320 · 579a320
1 parent 01d649b
commit 579a320
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pgconn/errors.go b/pgconn/errors.go
@@ -12,8 +12,9 @@ import (
 
 // SafeToRetry checks if the err is guaranteed to have occurred before sending any data to the server.
 func SafeToRetry(err error) bool {
-	if e, ok := err.(interface{ SafeToRetry() bool }); ok {
-		return e.SafeToRetry()
+	var retryableErr interface{ SafeToRetry() bool }
+	if errors.As(err, &retryableErr) {
+		return retryableErr.SafeToRetry()
 	}
 	return false
 }