feat: add tenant id to invoke context

Author: Ramisa Alam

deps/aws-lambda-cpp-0.2.8.tar.gz

@@ -0,0 +1,40 @@
+diff --git a/include/aws/lambda-runtime/runtime.h b/include/aws/lambda-runtime/runtime.h
+index f7cb8ef..833b69d 100644
+--- a/include/aws/lambda-runtime/runtime.h
++++ b/include/aws/lambda-runtime/runtime.h
+@@ -56,6 +56,11 @@ struct invocation_request {
+      */
+     std::string function_arn;
+ 
++    /**
++     * The Tenant ID of the current invocation.
++     */
++    std::string tenant_id;
++
+     /**
+      * Function execution deadline counted in milliseconds since the Unix epoch.
+      */
+diff --git a/src/runtime.cpp b/src/runtime.cpp
+index d1e655f..bcc217d 100644
+--- a/src/runtime.cpp
++++ b/src/runtime.cpp
+@@ -40,6 +40,7 @@ static constexpr auto CLIENT_CONTEXT_HEADER = "lambda-runtime-client-context";
+ static constexpr auto COGNITO_IDENTITY_HEADER = "lambda-runtime-cognito-identity";
+ static constexpr auto DEADLINE_MS_HEADER = "lambda-runtime-deadline-ms";
+ static constexpr auto FUNCTION_ARN_HEADER = "lambda-runtime-invoked-function-arn";
++static constexpr auto TENANT_ID_HEADER = "lambda-runtime-aws-tenant-id";
+ 
+ enum Endpoints {
+     INIT,
+@@ -296,6 +297,11 @@ runtime::next_outcome runtime::get_next()
+         req.function_arn = std::move(out).get_result();
+     }
+ 
++    out = resp.get_header(TENANT_ID_HEADER);
++    if (out.is_success()) {
++        req.tenant_id = std::move(out).get_result();
++    }
++
+     out = resp.get_header(DEADLINE_MS_HEADER);
+     if (out.is_success()) {
+         auto const& deadline_string = std::move(out).get_result();

deps/patches/aws-lambda-cpp-add-tenant-id.patch

@@ -29,7 +29,8 @@ wget -c https://github.com/awslabs/aws-lambda-cpp/archive/refs/tags/v$AWS_LAMBDA
 # Apply patches
 (
   cd aws-lambda-cpp-$AWS_LAMBDA_CPP_RELEASE && \
-    patch -p1 < ../patches/aws-lambda-cpp-add-xray-response.patch
+    patch -p1 < ../patches/aws-lambda-cpp-add-xray-response.patch && \
+    patch -p1 < ../patches/aws-lambda-cpp-add-tenant-id.patch
 )
 
 # Pack again and remove the folder

scripts/update_dependencies.sh

@@ -18,6 +18,7 @@ const INVOKE_HEADER = {
   AWSRequestId: 'lambda-runtime-aws-request-id',
   DeadlineMs: 'lambda-runtime-deadline-ms',
   XRayTrace: 'lambda-runtime-trace-id',
+  TenantId: 'lambda-runtime-aws-tenant-id',
 };
 
 module.exports = class InvokeContext {
@@ -91,6 +92,7 @@ module.exports = class InvokeContext {
       ),
       invokedFunctionArn: this.headers[INVOKE_HEADER.ARN],
       awsRequestId: this.headers[INVOKE_HEADER.AWSRequestId],
+      tenantId: this.headers[INVOKE_HEADER.TenantId],
       getRemainingTimeInMillis: function () {
         return deadline - Date.now();
       },

src/InvokeContext.js

@@ -82,6 +82,11 @@ class RuntimeApiNextPromiseWorker : public Napi::AsyncWorker {
                     Napi::String::New(env, "lambda-runtime-cognito-identity"),
                     Napi::String::New(env, response.cognito_identity.c_str()));
         }
+        if (response.tenant_id != "") {
+            headers.Set(
+                    Napi::String::New(env, "lambda-runtime-aws-tenant-id"),
+                    Napi::String::New(env, response.tenant_id.c_str()));
+        }
 
         auto ret = Napi::Object::New(env);
         ret.Set(Napi::String::New(env, "bodyJson"), response_data);

src/rapid-client.cc

@@ -35,3 +35,32 @@ describe('Getting remaining invoke time', () => {
     remainingTime.should.lessThanOrEqual(1000);
   });
 });
+
+describe('Verifying tenant id', () => {
+  it('should return undefined if tenant id is not set', () => {
+    let ctx = new InvokeContext({});
+
+    (ctx._headerData().tenantId === undefined).should.be.true();
+  });
+  it('should return undefined if tenant id is set to undefined', () => {
+    let ctx = new InvokeContext({
+      'lambda-runtime-aws-tenant-id': undefined,
+    });
+
+    (ctx._headerData().tenantId === undefined).should.be.true();
+  });
+  it('should return empty if tenant id is set to empty string', () => {
+    let ctx = new InvokeContext({
+      'lambda-runtime-aws-tenant-id': '',
+    });
+
+    (ctx._headerData().tenantId === '').should.be.true();
+  });
+  it('should return the same id if a valid tenant id is set', () => {
+    let ctx = new InvokeContext({
+      'lambda-runtime-aws-tenant-id': 'blue',
+    });
+
+    ctx._headerData().tenantId.should.equal('blue');
+  });
+});