introduce a network error class so not all Lambda client errors are t…

…reated as AWS errors
grafana · Apr 29, 2024 · 4cbe995 · 4cbe995
1 parent db491d4
commit 4cbe995
Show file tree

Hide file tree

Showing 26 changed files with 39 additions and 23 deletions.
diff --git a/dist/aws.js b/dist/aws.js
diff --git a/dist/aws.js.map b/dist/aws.js.map
diff --git a/dist/event-bridge.js b/dist/event-bridge.js
diff --git a/dist/event-bridge.js.map b/dist/event-bridge.js.map
diff --git a/dist/index.js b/dist/index.js
diff --git a/dist/index.js.map b/dist/index.js.map
diff --git a/dist/kinesis.js b/dist/kinesis.js
diff --git a/dist/kinesis.js.map b/dist/kinesis.js.map
diff --git a/dist/kms.js b/dist/kms.js
diff --git a/dist/kms.js.map b/dist/kms.js.map
diff --git a/dist/lambda.js b/dist/lambda.js
diff --git a/dist/lambda.js.map b/dist/lambda.js.map
diff --git a/dist/s3.js b/dist/s3.js
diff --git a/dist/s3.js.map b/dist/s3.js.map
diff --git a/dist/secrets-manager.js b/dist/secrets-manager.js
diff --git a/dist/secrets-manager.js.map b/dist/secrets-manager.js.map
diff --git a/dist/signature.js b/dist/signature.js
diff --git a/dist/signature.js.map b/dist/signature.js.map
diff --git a/dist/sqs.js b/dist/sqs.js
diff --git a/dist/sqs.js.map b/dist/sqs.js.map
diff --git a/dist/ssm.js b/dist/ssm.js
diff --git a/dist/ssm.js.map b/dist/ssm.js.map
diff --git a/src/index.ts b/src/index.ts
@@ -19,3 +19,4 @@ export { SQSClient } from './sqs'
 export { KinesisClient } from './internal/kinesis'
 export { EventBridgeClient } from './internal/event-bridge'
 export { LambdaClient, LambdaInvocationError } from './lambda'
+export { AWSError, NetworkError} from './internal/error'
diff --git a/src/internal/error.ts b/src/internal/error.ts
@@ -53,3 +53,13 @@ export class AWSError extends Error {
         }
     }
 }
+
+export class NetworkError extends Error {
+    code: number;
+
+    constructor(message: string, code: number) {
+        super(message)
+        this.name = 'NetworkError'
+        this.code = code
+    }
+}
diff --git a/src/internal/lambda.ts b/src/internal/lambda.ts
@@ -3,7 +3,7 @@ import encoding from 'k6/encoding'
 
 import { AWSClient } from './client'
 import { AWSConfig } from './config'
-import { AWSError } from './error'
+import { AWSError, NetworkError } from './error'
 import { InvalidSignatureError, SignatureV4 } from './signature'
 import { AMZ_TARGET_HEADER } from './constants'
 import { HTTPHeaders, HTTPMethod, QueryParameterBag } from './http'
@@ -107,6 +107,10 @@ export class LambdaClient extends AWSClient {
             return
         }
 
+        if (errorCode < 1400 || errorCode >= 1600 ) {
+            throw new NetworkError(errorMessage, errorCode)
+        }
+
         const awsError = AWSError.parse(response)
         switch (awsError.code) {
             case 'AuthorizationHeaderMalformed':

diff --git a/src/lambda.ts b/src/lambda.ts
@@ -5,3 +5,4 @@ export {
     LambdaInvocationError,
     LambdaClient
 } from './internal/lambda'
+export { AWSError, NetworkError} from './internal/error'