feat: introduce timeout to s3 read requests

graphql-hive · Aug 24, 2024 · 20da8c7 · 20da8c7
1 parent dacc302
commit 20da8c7
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/packages/services/cdn-worker/src/artifact-storage-reader.ts b/packages/services/cdn-worker/src/artifact-storage-reader.ts
@@ -21,6 +21,9 @@ type SDLArtifactTypes = `sdl${'.graphql' | '.graphqls' | ''}`;
 
 export type ArtifactsType = SDLArtifactTypes | 'metadata' | 'services' | 'supergraph';
 
+/** Timeout in milliseconds for S3 read calls. */
+const READ_TIMEOUT_MS = 20_000;
+
 const OperationS3BucketKeyModel = zod.tuple([
   zod.string().uuid(),
   zod.string().min(1),
@@ -85,6 +88,7 @@ export class ArtifactStorageReader {
         headers: {
           'X-Amz-Expires': String(presignedUrlExpirationSeconds),
         },
+        timeout: READ_TIMEOUT_MS,
       },
     );
 
@@ -126,6 +130,7 @@ export class ArtifactStorageReader {
         aws: {
           signQuery: true,
         },
+        timeout: READ_TIMEOUT_MS,
       },
     );
     this.analytics?.track(
@@ -164,6 +169,7 @@ export class ArtifactStorageReader {
         aws: {
           signQuery: true,
         },
+        timeout: READ_TIMEOUT_MS,
       },
     );
     this.analytics?.track(
@@ -200,6 +206,7 @@ export class ArtifactStorageReader {
           signQuery: true,
         },
         headers,
+        timeout: READ_TIMEOUT_MS,
       },
     );