Merge pull request #35 from topcoder-platform/feature/shapeup3

Feature/shapeup3 - adding `key` feature

Author: nkumar-topcoder

.circleci/config.yml

@@ -77,7 +77,7 @@ workflows:
           context : org-global
           filters:
             branches:
-              only: ['dev']
+              only: ['dev', 'feature/shapeup3']
       - "build-prod":
           context : org-global
           filters:

api/swagger.yaml

@@ -979,6 +979,10 @@ definitions:
         type: "string"
         example: "application/json"
         description: "Mime-type for 'payload'.\n"
+      key:
+        type: "string"
+        example: "abc123"
+        description: "optional field to assure the one partition for 'payload'.\n"
       payload:
         type: "object"
         example:

common/helper.js

@@ -133,6 +133,7 @@ function validateEventPayload (event) {
       originator: Joi.string().required(),
       timestamp: Joi.string().required(),
       'mime-type': Joi.string().required(),
+      key: Joi.string(),
       payload: Joi.any()
     })
   })

service/MessageBusService.js

@@ -30,12 +30,17 @@ async function postEvent (event) {
     helper.validateEventPayload(event)
 
     // Post new structure
-    const result = await producer.send({
+    const message = {
+      value: JSON.stringify(event)
+    }
+    if (event.key) {
+      _.merge(message, { key: event.key })
+    }
+    const kafkaPayload = {
       topic: event.topic,
-      message: {
-        value: JSON.stringify(event)
-      }
-    })
+      message
+    }
+    const result = await producer.send(kafkaPayload)
     // Check if there is any error
     const error = _.get(result, '[0].error')
     if (error) {