@izara_project/izara-market-library-service-schemas 1.0.35 → 1.0.37

package/src/reStructure/TemplateData/flowSchema/createRecordByStatusType/complete/mainFunction/template.ejs

@@ -18,7 +18,7 @@ along with this program. If not, see
 
 'use strict';
 
-const hash = require("object-hash");
+const hash = require('@izara_project/izara-shared-core').objectHash;
 const { getObjectSchema } = require('@izara_project/izara-core-library-service-schemas');
 
 const dynamodbSharedLib = require('@izara_project/izara-core-library-dynamodb');
@@ -29,16 +29,16 @@ const lambdaSharedLib = require('@izara_project/izara-core-library-lambda');
 
 const externalRequest = require('@izara_project/izara-core-library-external-request');
 const sns = externalRequest.sns
-const { postToConnection } = require("../../../../libs/source/GenerateCodeLibs");
+const { postToConnection } = require("../../../libs/source/GenerateCodeLibs");
 
 const utils = require('@izara_project/izara-market-library-service-schemas/src/MainLibs/src/Utils')
-const NoRetryError = require('@izara_project/izara-core-library-core').NoRetryError
+const { NoRetryError } = require('@izara_project/izara-core-library-core')
 const { v4: uuidv4 } = require('uuid')
 const coreConsts = require("@izara_project/izara-core-library-core/src/Consts");
-const { TOPIC_NAME_GENERATE_CODE } = require('@izara_project/izara-market-library-service-schemas/src/GenerateCodeLibs/src/Consts')
+// const { TOPIC_NAME_GENERATE_CODE } = require('@izara_project/izara-market-library-service-schemas/src/GenerateCodeLibs/src/Consts')
+const { generateCodeLibs: { consts: TOPIC_NAME_GENERATE_CODE } } = require('@izara_project/izara-market-library-service-schemas')
+
 /**
- *
- *
  * description of function.
  * @param {Object} _izContext
  * @param {CorrelationIds} _izContext.correlationIds - property of _izContext
@@ -47,11 +47,9 @@ const { TOPIC_NAME_GENERATE_CODE } = require('@izara_project/izara-market-librar
  * @param {Object} requestParams.identifiers - identifiers for get data
  * @param {Object} requestParams.additionalRequest - additionalRequest
  *
- *
  * @returns {object} description of return value
  */
-
-module.exports.<%- functionMain%> = async (
+module.exports.registerCompleteMain = async (
   _izContext,
   requestParams,
   callingFlowConfig = {},
@@ -63,14 +61,14 @@ module.exports.<%- functionMain%> = async (
     _izContext.logger.debug("WebSocketInvoke requestParams", requestParams)
     _izContext.logger.debug("WebSocketInvoke callingFlowConfig", callingFlowConfig)
 
-    const connectionId = event.requestContext.connectionId;
-    let correlationId = event._izContext.correlationIds.get(consts.X_CORRELATION_ID);
+    const connectionId = requestContext.connectionId;
+    let correlationId = _izContext.correlationIds.get(consts.X_CORRELATION_ID);
 
     const getRegisterRecords = await dynamodbSharedLib.getItem(
       _izContext,
       await dynamodbSharedLib.tableName(_izContext, "RegisterRecords"),
       {
-        taskKey: correlationId,
+        identifierTask: correlationId,
         connectionId: connectionId,
       }
     );
@@ -90,40 +88,93 @@ module.exports.<%- functionMain%> = async (
       throw new NoRetryError(`flowTag not found ${flowTag}`)
     }
 
-    switch (flowSchema.statusType) {
+    // get data from dynamoDB of objectType of flowSchema
+    const getStoredCache = async (objectType) => {
+      const objectSchema = await getObjectSchema.getObjSchemaS3WithCache(
+        _izContext,
+        {
+          objectType: objectType,
+          serviceTag: process.env.iz_serviceTag,
+        })
+
+      _izContext.logger.debug("==> objectSchema", objectSchema)
+
+      // this returns the table name of the first storage resource with storageType "dynamoDB"
+      const tableName = Object.values(objectSchema.storageResources).find(
+        (storageResource) => storageResource.storageType === "dynamoDB"
+      )?.tableName
+      _izContext.logger.debug("==> tableName", tableName)
+
+      let identifierPK = null
+      let identifierSK = null
+      objectSchema.identifiers.map((identifier) => {
+        if (identifier.type === "partitionKey") {
+          identifierPK = identifier.fieldName
+        } else if (identifier.type === "sortKey") {
+          identifierSK = identifier.fieldName
+        }
+      })
+
+      _izContext.logger.debug("==> identifierPK", identifierPK)
+      _izContext.logger.debug("==> identifierSK", identifierSK)
+
+      const payloadGetStoredCache = {
+        [identifierPK]: requestParams.identifiers[identifierPK],
+      }
+      if (identifierSK) {
+        payloadGetStoredCache[identifierSK] = requestParams.identifiers[identifierSK]
+      }
+
+      _izContext.logger.debug("==> payloadGetStoredCache", payloadGetStoredCache)
+
+      // const getStoredCache = await dynamodbSharedLib.getItem(
+      //   _izContext,
+      //   await dynamodbSharedLib.tableName(_izContext, tableName),
+      //   payloadGetStoredCache
+      // )
+
+      const getStoredCache = await dynamodbSharedLib.query(
+        _izContext,
+        await dynamodbSharedLib.tableName(_izContext, tableName),
+        payloadGetStoredCache
+      )
+      _izContext.logger.debug("==> getStoredCache:::", getStoredCache)
+      return ({
+        data: getStoredCache.Items[0],
+        identifierPK: identifierPK,
+      })
+    }
+
+    switch (getFlowSchema.statusType) {
       case "statusField":
         let payload = {
           identifiers: requestParams.identifiers,
-          objectType: flowSchema.objType.objectType
+          objectType: getFlowSchema.objType.objectType
         }
         let result = await externalRequest.lambda.invokeSync(
           _izContext,
-          await lambdaSharedLib.lambdaFunctionName(_izContext, "GetHdrInv", flowSchema.objType.serviceTag),
+          await lambdaSharedLib.lambdaFunctionName(_izContext, "GetHdrInv", getFlowSchema.objType.serviceTag),
           payload
         )
         if (!result) {
           throw new NoRetryError(`statusField not found ${requestParams.identifiers}`)
         }
+        await postToConnection({message:result}, connectionId)
         return result
-        case "storedCache":
+      case "storedCache":
         // if statusType=storedCache then query storedCache > post message to client -> remove connectionId if status = error || complete
-        const getStoredCache = await dynamodbSharedLib.getItem(
-          _izContext,
-          await dynamodbSharedLib.tableName(_izContext, flowSchema.flowTag + "Main"),
-          {
-            taskKey: requestParams.identifiers,
-            connectionId: requestParams.connectionId
-          }
-        )
-        if (!getStoredCache) {
+        const resultStoredCache = resultStoredCache(getFlowSchema.objType.objectType)
+
+        if (!resultStoredCache) {
           throw new NoRetryError(`storedCache not found ${requestParams.identifiers}`)
         }
-        return getStoredCache
+        await postToConnection({message:resultStoredCache }, connectionId)
+        return resultStoredCache
       case "none":
         // if statusType=none -> post message to client ---- not remove connectionId when post message, can remove connectionId when cannot use connectionId only(user disconnect)
         break
       default:
-        throw new NoRetryError(`statusType not found ${flowSchema.statusType}`)
+        throw new NoRetryError(`statusType not found ${getFlowSchema.statusType}`)
     }
 
   } catch (err) {

package/src/reStructure/TemplateData/flowSchema/createRecordByStatusType/dynamoDB/register.js

@@ -0,0 +1,82 @@
+/*
+Copyright (C) 2020 Sven Mason <http://izara.io>
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU Affero General Public License as
+published by the Free Software Foundation, either version 3 of the
+License, or (at your option) any later version.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU Affero General Public License for more details.
+
+You should have received a copy of the GNU Affero General Public License
+along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+'use strict';
+
+const path = require("path")
+
+const templatePath = path.join(__dirname, '../../../resourceYaml/dynamodb/template.ejs')
+const { SOURCE_PATH, SAVE_FILE_NAME, FUNCTION_NAME, defaultIamRolePerAction, DYNAMO_RESOURCE, RESOURCE_CLASSES, resourceNames, createIamRole, SOURCE_GENERATE_IAM_ROLE } = require('../../../../../MainLibs/src/Consts.js')
+
+/**
+ * create param of crateSouce for FindData And processLogical
+ *
+ * @param {Object} _izContext
+ * @param {String} srcPath
+ * @returns {Object[]}
+ */
+const createRegisterTable = (_izContext, srcPath) => {
+  let additionalResourcePermission = defaultIamRolePerAction();
+  const tableName = "RegisterRecords";
+  additionalResourcePermission.push(
+    createIamRole(
+      {
+        [RESOURCE_CLASSES.dynamoDbTable]: [
+          DYNAMO_RESOURCE.putItem,
+          DYNAMO_RESOURCE.getItem,
+          DYNAMO_RESOURCE.query,
+          DYNAMO_RESOURCE.deleteItem,
+          DYNAMO_RESOURCE.updateItem
+        ]
+      },
+      [
+        resourceNames(RESOURCE_CLASSES.dynamoDbTable, tableName)
+      ]
+    ),
+  );
+
+  return [{
+    templatePath: templatePath,
+    templateData: {
+      tableName: tableName,
+      resourceName: tableName,
+      roleName: SOURCE_GENERATE_IAM_ROLE.RegisterRole,
+      additionalResourcePermission,
+      attributes: [
+        {
+          keyType: "partitionKey",
+          AttributeName: "identifierTask",
+          AttributeType: "S"
+        },
+        {
+          keyType: "sortKey",
+          AttributeName: "connectionId",
+          AttributeType: "S"
+        }
+      ],
+    },
+    setting: {
+      initialData: 'Resources:\n',
+      savePath: path.join(srcPath, SOURCE_PATH.resourceYaml),
+      saveFileName: SAVE_FILE_NAME.dynamoDbYaml,
+      fileExtension: '.yml',
+      isAppend: true
+    }
+  }];
+};
+
+module.exports = createRegisterTable;

package/src/reStructure/TemplateData/flowSchema/createRecordByStatusType/sns-in/data.js

@@ -48,7 +48,7 @@ function createDataForSnsIn(_izContext, srcPath) {
   return [{
     templatePath: templatePath,
     templateData: {
-      queueName: FUNCTION_NAME.flowSchemaRegister + upperCase(HANDLER.hdrSqs),
+      queueName: upperCase(FUNCTION_NAME.flowSchemaRegister),
       firstLetterUpperCase: upperCase,
       roleName: SOURCE_GENERATE_IAM_ROLE.RegisterRole,
     },

package/src/reStructure/TemplateData/flowSchema/createRecordByStatusType/sns-in/template.ejs

@@ -9,4 +9,37 @@
             - <%- queueName %>DLQ
             - Arn
         maxReceiveCount: 3
-      VisibilityTimeout: 120
+      VisibilityTimeout: 120
+  ##==== [QueueDLQ]
+  <%- queueName %>DLQ:
+    Type: AWS::SQS::Queue
+    Properties:
+      QueueName: ${self:custom.iz_resourcePrefix}<%- queueName %>DLQ
+##==== [QueuePolicy]
+  <%- queueName %>Policy:
+    Type: AWS::SQS::QueuePolicy
+    Properties:
+      PolicyDocument:
+        Version: "2012-10-17"
+        Statement:
+          - Sid: "allow-sns-messages"
+            Effect: Allow
+            Principal: "*"
+            Resource:
+            # !GetAtt
+              Fn::GetAtt:
+                - <%- queueName %>
+                - Arn
+            Action: "SQS:SendMessage"
+      Queues:
+        - Ref: <%- queueName %>
+  #<#<%- firstLetterUpperCase(queueName) %>QueueSetting#>
+  #<#/<%- firstLetterUpperCase(queueName) %>QueueSetting#>
+
+
+      <%_ function firstLetterUpperCase(text){
+        return text.charAt(0).toUpperCase() + text.slice(1)
+      } _%>
+      <%_ function firstLetterLowerCase(str) {
+        return str.charAt(0).toLowerCase() + str.slice(1)
+      } _%>

package/src/reStructure/TemplateData/flowSchema/createRecordByStatusType/storedCacheTemplate.ejs

@@ -1,34 +1,11 @@
-// create storedCache
-function createExpiryTime (_izContext, expiryInterval) {
-  const currentTime = new Date();
-  const expiryTime = new Date(currentTime);
-  expiryTime.setTime(expiryTime.getTime() + expiryInterval);
-  return expiryTime.getTime();
-};
+let expiryTime = storedCacheSharedLib.createExpiryTime(_izContext, 31536000);
 
-await dynamodbSharedLib.putItem(_izContext,
-  await dynamodbSharedLib.tableName(_izContext, "<%- flowTag %>Main"),
-  {
-    <%- flowTag %>Id: hash({
-      //(<createStoredCacheId>)
-      //(</createStoredCacheId>)
-    }),
-    cacheExpiryTime: createExpiryTime(_izContext, 31536000000),
-    cacheStatus: "processing",
-    cacheUniqueRequestId: _izContext.uniqueRequestId
-  }
-)
+//(<createKeyValuesStoredCacheTable>)
+//(</createKeyValuesStoredCacheTable>)
 
-await dynamodbSharedLib.updateItem(_izContext,
-  await dynamodbSharedLib.tableName(_izContext, "WebSocketTask"),
-  {
-    taskKey: correlationId,
-    connectionId: connectionId,
-  },
-  {
-    identifiersTask: hash({
-      //(<updateStoredCacheId>)
-      //(</updateStoredCacheId>)
-    })
-  }
-  )
+let [ cacheMainStauts, cacheId, cacheMain ] = await storedCacheSharedLib.checkStoredCacheStatus(
+  _izContext,
+  "<%- tableName %>",
+  //(<keyValueStoredCacheTable>)
+  //(</keyValueStoredCacheTable>)
+)

package/src/reStructure/TemplateData/flowSchema/createRecordByStatusType/subscriptionOutAll/data.js

@@ -45,29 +45,39 @@ const { SOURCE_PATH, SAVE_FILE_NAME, FUNCTION_NAME, HANDLER } = require('../../.
  */
 
 function createDataForSubscriptionOutAll(_izContext, allLocalFlowSchemas, srcPath) {
-  console.log("allLocalFlowSchemas", allLocalFlowSchemas.records)
+  // console.log("allLocalFlowSchemas", allLocalFlowSchemas.records)
   const result = []
   for (const flowSchema of allLocalFlowSchemas.records) {
     let event = flowSchema.event;
-    if (!event.includes("eventBridge")) {
-      const flowStepOut = Object.keys(flowSchema.flowSteps)[1]
-      console.log("flowStepOut", flowStepOut)
-      result.push({
-        templatePath: templatePath,
-        templateData: {
-          queueName: flowStepOut,
-          endpoint: "registerComplete_",
-          firstLetterUpperCase: upperCase,
-        },
-        setting: {
-          initialData: "Resources:\n",
-          savePath: path.join(srcPath, SOURCE_PATH.resourceYaml),
-          saveFileName: SAVE_FILE_NAME.snsInSqsYaml,
-          fileExtension: ".yml",
-          isAppend: true
-        }
-      })
-    };
+    if (!event.includes("eventBridge") || !event.includes("s3")) {
+      let flowSteps = flowSchema.flowSteps;
+      if (!flowSteps) {
+        return result
+      }
+      let queueName;
+      if (flowSchema.outputTopic) {
+        let flowStepOut = Object.keys(flowSchema.flowSteps)[1].split("_")
+        queueName = flowStepOut[3] + flowStepOut[2]
+      }
+
+      if (queueName) {
+        result.push({
+          templatePath: templatePath,
+          templateData: {
+            queueName: queueName,
+            endpoint: "Register",
+            firstLetterUpperCase: upperCase,
+          },
+          setting: {
+            initialData: "Resources:\n",
+            savePath: path.join(srcPath, SOURCE_PATH.resourceYaml),
+            saveFileName: SAVE_FILE_NAME.snsInSqsYaml,
+            fileExtension: ".yml",
+            isAppend: true
+          }
+        })
+      };
+    }
   }
 
   return result

package/src/reStructure/TemplateData/flowSchema/createRecordByStatusType/subscriptionOutAll/template.ejs

@@ -1,10 +1,10 @@
 ##===== [Topic In]
-SubscriptionIn<%- queueName %>:
-  Type: AWS::SNS::Subscription
-  Properties:
-    TopicArn: !Ref In<%- queueName %>
-    Endpoint: "arn:aws:sqs:${self:custom.iz_region}:${self:custom.iz_accountId}:${self:custom.iz_resourcePrefix}<%- queueName %>HdrSqs"
-    Protocol: "sqs"
+  Subscription<%- queueName %>:
+    Type: AWS::SNS::Subscription
+    Properties:
+      TopicArn: !Ref <%- queueName %>
+      Endpoint: "arn:aws:sqs:${self:custom.iz_region}:${self:custom.iz_accountId}:${self:custom.iz_resourcePrefix}<%- endpoint %>"
+      Protocol: "sqs"
 <%_ function firstLetterUpperCase(text){
   return text.charAt(0).toUpperCase() + text.slice(1)
 } _%>

package/src/reStructure/TemplateData/flowSchema/createRecordByStatusType/wbs/functionYaml/data.js

@@ -19,7 +19,7 @@ along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
 const path = require('path');
 
-const { FLOW_SCHEMA_HOOK_STATE } = require('@izara_project/izara-core-library-service-schemas/src/Consts')
+const { FLOW_SCHEMA_HOOK_STATE, STORAGE_TYPES } = require('@izara_project/izara-core-library-service-schemas/src/Consts')
 
 const { firstLetterUpperCase: upperCase } = require("../../../../../../MainLibs/src/Utils");
 const {
@@ -37,6 +37,7 @@ const {
   SOURCE_GENERATE_IAM_ROLE,
   LAMBDA_RESOURCE
 } = require("../../../../../../MainLibs/src/Consts");
+const { getLocalObjectSchemas } = require('@izara_project/izara-core-library-service-schemas/src/GetObjectSchema');
 
 const templatePath = path.join(__dirname, "template.ejs");
 const hookTemplate = path.join(__dirname, "./hookTemplate.ejs");
@@ -47,12 +48,11 @@ const hookTemplate = path.join(__dirname, "./hookTemplate.ejs");
  * @param {Object} objectSchema
  * @return {{templatePath, templateData,setting}}
  */
-function data(_izContext, srcPath) {
+async function data(_izContext, allLocalFlowSchemas, srcPath) {
   const handlerType = upperCase("hdrWbs");
   const functionName = FUNCTION_NAME.flowSchemaRegister;
   let functionNameConfig = upperCase(functionName) + upperCase(shortNameHandler(handlerType));
   let additionalResourcePermission = defaultIamRolePerAction();
-  let route = "webSocket";
 
   additionalResourcePermission.push(
     createIamRole(
@@ -91,40 +91,40 @@ function data(_izContext, srcPath) {
     )
   )
 
-  // if (flowSchema?.hook) {
-  //   for (const [hookName, hookProperties] of Object.entries(flowSchema.hook)) {
-  //     if (hookName === FLOW_SCHEMA_HOOK_STATE.beforeCreate) {
-  //       for (const hookFunction of hookProperties.functionList) {
-  //         if ((hookFunction.event.hasOwnProperty("initialEvent") && (hookFunction.event.initialEvent === true))) {
-  //           if (hookFunction?.permission) {
-  //             for (const permission of hookFunction.permission) {
-  //               additionalResourcePermission.push(
-  //                 createIamRole(
-  //                   permission.action,
-  //                   permission.resource.map(resource =>
-  //                     externalResourceYaml(
-  //                       resource.resource,
-  //                       resource.resourceName,
-  //                       resource.serviceTag),
-  //                     permission.effect
-  //                   )
-  //                 )
-  //               )
-  //             }
-  //           }
-  //         }
-  //       }
-  //     }
-  //   }
-  // }
+  let tableStoredCacheList = new Set();
+  await Promise.all(allLocalFlowSchemas.records.map(async (flowSchema) => {
+    if (flowSchema.statusType === "storedCache") {
+      let objectSchema = await getLocalObjectSchemas(_izContext, [flowSchema.objType.objectType], path.join(srcPath, "./schemas")).then(res => res.records[0]);
+      for (const storageResource of Object.values(objectSchema.storageResources)) {
+        if (storageResource.storageType === STORAGE_TYPES.dynamoDB) {
+          tableStoredCacheList.add(storageResource.tableName);
+        }
+      }
+    }
+  }))
+  if (tableStoredCacheList.size > 0) {
+    let tableNames = [...tableStoredCacheList]
+    additionalResourcePermission.push(
+      createIamRole(
+        {
+          [RESOURCE_CLASSES.dynamoDbTable]: [
+            DYNAMO_RESOURCE.getItem,
+            // DYNAMO_RESOURCE.putItem,
+            // DYNAMO_RESOURCE.query,
+            // DYNAMO_RESOURCE.deleteItem,
+            // DYNAMO_RESOURCE.updateItem
+          ]
+        },
+        tableNames.map(tableName => resourceNames(RESOURCE_CLASSES.dynamoDbTable, tableName))
+      )
+    )
+  }
 
-  // console.log("additionalResourcePermission in createObjectS3", JSON.stringify(additionalResourcePermission))
   return [{
     templatePath: templatePath,
     templateData: {
-      resourceLocation: SOURCE_PATH.resourceLocationFlowSchemaRegister,
+      resourceLocation: path.join(SOURCE_PATH.resourceLocationFlowSchemaRegister, "source/"),
       additionalResourcePermission,
-      route,
       functionName,
       handlerType,
       roleName: SOURCE_GENERATE_IAM_ROLE.RegisterRole,

package/src/reStructure/TemplateData/flowSchema/createRecordByStatusType/wbs/functionYaml/template.ejs

@@ -4,12 +4,12 @@
   name: ${self:custom.iz_resourcePrefix}<%- firstLetterUpperCase(functionName) %><%- firstLetterUpperCase(handlerType) %>
   events:
       - websocket:
-        route: <%- route %>
-        # authorizer: ${self:custom.iz_authorizerAppLevel}
-        # identifierResource: 'route.request.header.Auth'
+          route: <%- functionName %>
+          # authorizer: ${self:custom.iz_authorizerAppLevel}
+          # identifierResource: 'route.request.header.Auth'
   role: <%- roleName %>Role
-    #<#<%- functionName %><%- handlerType %>IamRole#>
-    #<#/<%- functionName %><%- handlerType %>IamRole#>
+  #<#<%- functionName %><%- handlerType %>IamRole#>
+  #<#/<%- functionName %><%- handlerType %>IamRole#>
 <%_ function firstLetterUpperCase(text){
   return text.charAt(0).toUpperCase() + text.slice(1)
 } _%>

package/src/reStructure/TemplateData/flowSchema/createRecordByStatusType/wbs/handler/template.ejs

@@ -21,11 +21,10 @@ const { middlewareHandler, consts } = require("@izara_project/izara-middleware")
 // const AWS = require('aws-sdk');
 // const api = new AWS.ApiGatewayManagementApi({ endpoint: process.env.iz_webSocketEndpoint });
 const dynamodbSharedLib = require('@izara_project/izara-core-library-dynamodb');
-const callingFlowSharedLib = require('@izara_project/izara-core-calling-flow');
-const hash = require('object-hash');
+const callingFlowSharedLib = require('@izara_project/izara-core-library-calling-flow');
+const hash = require('@izara_project/izara-shared-core').objectHash;
 // const coreConsts = require("@izara_project/izara-core-library-core/src/Consts");
-const consts = require('@izara_project/izara-middleware/src/MiddlewareCore/Consts');
-const { postToConnection } = require("../../libs/source/ImportDataLib");
+const { postToConnection } = require("../../../libs/source/GenerateCodeLibs");
 const { <%- functionName%>Main } = require("./<%- firstLetterUpperCase(functionName) %>_Main");
 
 module.exports.main = middlewareHandler.wrap(async (event, context, callback) => {
@@ -58,13 +57,17 @@ module.exports.main = middlewareHandler.wrap(async (event, context, callback) =>
 
           let callingFlow = eventParams?.callingFlow ? eventParams.callingFlow : {};
 
-          await <%- functionName%>Main(
+          const result = await registerMain(
             event._izContext,
             eventParams,
             callingFlowSharedLib.addCallingFlowToPassOnProperties(callingFlow)
             //(<additionalParams>)
             //(</additionalParams>)
           )
+          await postToConnection(
+            { message: result },
+            connectionId
+          )
           break
       }
     } else {
@@ -72,15 +75,11 @@ module.exports.main = middlewareHandler.wrap(async (event, context, callback) =>
       // await postMessage({ message: event.message }, connId);
       await postToConnection({ message: event.message }, connId);
     }
-
-    return (izara.response.webSocketSuccess());
-
   } catch (err) {
     const connId = event.requestContext.connectionId
     event._izContext.logger.error('Error, WebSocket: ', err);
     // await sendErrorMessage({ message: err.message }, connId);
     await postToConnection({ message: err.message }, connId);
-    return (izara.response.failure(err));
   }
 });
 <%_ const join = require('path').join _%>