fix(serverless-aws): cubejsProcess agent doesn't collect all events after process has been finished

Author: paveltiunov

packages/cubejs-server-core/core/index.js

@@ -323,6 +323,7 @@ class CubejsServerCore {
   initAgent() {
     if (process.env.CUBEJS_AGENT_ENDPOINT_URL) {
       const oldLogger = this.logger;
+      this.preAgentLogger = oldLogger;
       this.logger = (msg, params) => {
         oldLogger(msg, params);
         agentCollect(
@@ -337,6 +338,16 @@ class CubejsServerCore {
     }
   }
 
+  async flushAgent() {
+    if (process.env.CUBEJS_AGENT_ENDPOINT_URL) {
+      await agentCollect(
+        { msg: 'Flush Agent' },
+        process.env.CUBEJS_AGENT_ENDPOINT_URL,
+        this.preAgentLogger
+      );
+    }
+  }
+
   static create(options) {
     return new CubejsServerCore(options);
   }

packages/cubejs-serverless-aws/index.js

@@ -29,6 +29,7 @@ class AWSHandlers extends Handlers {
         const message = JSON.parse(record.Sns.Message);
         await this.processMessage(message);
       }));
+      await this.serverCore.flushAgent();
     } else {
       this.serverCore.logger('Invalid Lambda Process Message', {
         warning: `Event doesn't contain Records field. Skipping.`,