primaryKey name / type (#1)

README.md

@@ -28,6 +28,8 @@ A 4th parameter is a configuration object:
 * `delay: number` adjust the delay in millisecond between two scan
 * `dyCapacityUnitLimit: number` defined a dynamodb capacity unit limit to stop scan if it's reached
 * `progressCallbackInterval: number` interval in millisecond for calling progress callback 
+* `primaryKeyName: string` the name of your primary key (default: 'uuid')
+* `primaryKeyType: string` the dynamoDB type of your primary key (default: 'S')
 
 then you can call the `run` method to start the scan stream to kinesis.
 it returns a promise where your callback will call at the end of a full scan.

package.json

@@ -1,38 +1,42 @@
 {
-  "name": "dyki",
-  "version": "1.0.0",
-  "description": "Scan DynamoDB items to dispatch into Kinesis Stream",
-  "keywords": [
-      "aws",
-      "serverless",
-      "amazon web service",
-      "dynamodb",
-      "kinesis",
-      "stream",
-      "elastic"
-  ],
-  "repository": {
-      "type": "git",
-      "url": "https://github.com/pierozi/dyki"
-  },
-  "bugs": "https://github.com/pierozi/dyki/issues",
-  "maintainers": [{
-	  "name": "Pierre Tomasina",
-      "web": "http://continuous.lu"
-  }],
-  "contributors": [{
-	  "name": "Pierre Tomasina",
-      "web": "http://continuous.lu"
-  }],
-  "main": "src/main.js",
-  "scripts": {
-    "test": "node tests/scratch.js"
-  },
-  "author": "Pierre Tomasina <pierre.tomasina@continuous.lu> (continuous.lu)",
-  "license": "MIT",
-  "dependencies": {
-    "aws-sdk": "^2.84.0",
-    "base-64": "^0.1.0",
-    "when": "^3.7.8"
-  }
+    "name": "dyki",
+    "version": "1.0.0",
+    "description": "Scan DynamoDB items to dispatch into Kinesis Stream",
+    "keywords": [
+        "aws",
+        "serverless",
+        "amazon web service",
+        "dynamodb",
+        "kinesis",
+        "stream",
+        "elastic"
+    ],
+    "repository": {
+        "type": "git",
+        "url": "https://github.com/pierozi/dyki"
+    },
+    "bugs": "https://github.com/pierozi/dyki/issues",
+    "maintainers": [
+        {
+            "name": "Pierre Tomasina",
+            "web": "http://continuous.lu"
+        }
+    ],
+    "contributors": [
+        {
+            "name": "Pierre Tomasina",
+            "web": "http://continuous.lu"
+        }
+    ],
+    "main": "src/main.js",
+    "scripts": {
+        "test": "node tests/scratch.js"
+    },
+    "author": "Pierre Tomasina <pierre.tomasina@continuous.lu> (continuous.lu)",
+    "license": "MIT",
+    "dependencies": {
+        "aws-sdk": "^2.697.0",
+        "base-64": "^0.1.0",
+        "when": "^3.7.8"
+    }
 }

src/main.js

@@ -9,6 +9,8 @@ DyKi.Config = {
     dyCapacityUnitLimit: 5,
     delay: 1000,
     progressCallbackInterval: 1000,
+    primaryKeyName: 'uuid',
+    primaryKeyType: 'S',
 };
 
 DyKi.Error = class extends Error {};
@@ -68,7 +70,7 @@ DyKi.Client = class {
         }
 
         clearInterval(progressInterval);
-        
+
         if (this.EOF) {
             return when(this.info());
         }
@@ -135,7 +137,7 @@ DyKi.Client = class {
 
         const params = {
             Data: JSON.stringify(record),
-            PartitionKey: item.uuid['S'],
+            PartitionKey: item[this.config.primaryKeyName][this.config.primaryKeyType],
             StreamName: this.streamName,
         };