set err code to timeout if ctx is done

smartcontractkit · Feb 22, 2024 · 528efd0 · 528efd0
1 parent 1cd2a97
commit 528efd0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/core/services/ocr2/plugins/ocr2keeper/evmregistry/v21/mercury/v02/request.go b/core/services/ocr2/plugins/ocr2keeper/evmregistry/v21/mercury/v02/request.go
@@ -173,6 +173,9 @@ func (c *client) singleFeedRequest(ctx context.Context, ch chan<- mercury.Mercur
 				retryable = true
 				state = encoding.MercuryFlakyFailure
 				errCode = encoding.ErrCodeStreamsUnknownError
+				if ctx.Err() != nil {
+					errCode = encoding.ErrCodeStreamsTimeout
+				}
 				return err
 			}
 			defer httpResponse.Body.Close()

diff --git a/core/services/ocr2/plugins/ocr2keeper/evmregistry/v21/mercury/v03/request.go b/core/services/ocr2/plugins/ocr2keeper/evmregistry/v21/mercury/v03/request.go
@@ -145,6 +145,9 @@ func (c *client) multiFeedsRequest(ctx context.Context, ch chan<- mercury.Mercur
 				retryable = true
 				state = encoding.MercuryFlakyFailure
 				errCode = encoding.ErrCodeStreamsUnknownError
+				if ctx.Err() != nil {
+					errCode = encoding.ErrCodeStreamsTimeout
+				}
 				return err
 			}
 			defer resp.Body.Close()