fix: Add UNKNOWN to retry code, the same set was used for request level retries. (#3034)

yirutang · web-flow · commit 60fb0c743a15 · 2025-07-08T16:48:25.000-07:00
* Update WriteToDefaultStream.java

* add unknown to retry code

* remove unwanted change.
diff --git a/google-cloud-bigquerystorage/src/main/java/com/google/cloud/bigquery/storage/v1/ConnectionWorker.java b/google-cloud-bigquerystorage/src/main/java/com/google/cloud/bigquery/storage/v1/ConnectionWorker.java
@@ -1254,7 +1254,8 @@ private boolean isConnectionErrorRetriable(Code statusCode) {
         || statusCode == Code.UNAVAILABLE
         || statusCode == Code.CANCELLED
         || statusCode == Code.INTERNAL
-        || statusCode == Code.DEADLINE_EXCEEDED;
+        || statusCode == Code.DEADLINE_EXCEEDED
+        || statusCode == Code.UNKNOWN;
   }
 
   private boolean useBackoffForError(Code statusCode, String streamName) {

Original file line number	Diff line number	Diff line change
`@@ -1254,7 +1254,8 @@ private boolean isConnectionErrorRetriable(Code statusCode) {`
`1254`	`1254`	`\|\| statusCode == Code.UNAVAILABLE`
`1255`	`1255`	`\|\| statusCode == Code.CANCELLED`
`1256`	`1256`	`\|\| statusCode == Code.INTERNAL`
`1257`		`- \|\| statusCode == Code.DEADLINE_EXCEEDED;`
	`1257`	`+ \|\| statusCode == Code.DEADLINE_EXCEEDED`
	`1258`	`+ \|\| statusCode == Code.UNKNOWN;`
`1258`	`1259`	`}`
`1259`	`1260`
`1260`	`1261`	`private boolean useBackoffForError(Code statusCode, String streamName) {`