Merge pull request #172 from terencenero007/v1

Added `convertEmptyValues` option to plugin which allows to save NULL types

Author: rehrumesh

README.md

@@ -53,6 +53,7 @@ All CLI options are optional:
 --optimizeDbBeforeStartup -o  Optimizes the underlying database tables before starting up DynamoDB on your computer. You must also specify -dbPath when you use this parameter.
 --migrate                 -m  After starting DynamoDB local, create DynamoDB tables from the Serverless configuration.
 --seed                    -s  After starting and migrating dynamodb local, injects seed data into your tables. The --seed option determines which data categories to onload.
+--convertEmptyValues      -e  Set to true if you would like the document client to convert empty values (0-length strings, binary buffers, and sets) to be converted to NULL types when persisting to DynamoDB.
 ```
 
 All the above options can be added to serverless.yml to set default configuration: e.g.
@@ -68,6 +69,7 @@ custom:
       inMemory: true
       migrate: true
       seed: true
+      convertEmptyValues: true
     # Uncomment only if you already have a DynamoDB running locally
     # noStart: true
 ```

index.js

@@ -68,6 +68,10 @@ class ServerlessDynamodbLocal {
                             seed: {
                                 shortcut: "s",
                                 usage: "After starting and migrating dynamodb local, injects seed data into your tables. The --seed option determines which data categories to onload.",
+                            },
+                            convertEmptyValues: {
+                                shortcut: "e",
+                                usage: "Set to true if you would like the document client to convert empty values (0-length strings, binary buffers, and sets) to be converted to NULL types when persisting to DynamoDB.",
                             }
                         }
                     },
@@ -135,13 +139,15 @@ class ServerlessDynamodbLocal {
             }
             dynamoOptions = {
                 region: options.region,
+                convertEmptyValues: options && options.convertEmptyValues ? options.convertEmptyValues : false,
             };
         } else {
             dynamoOptions = {
                 endpoint: `http://${this.host}:${this.port}`,
                 region: "localhost",
                 accessKeyId: "MOCK_ACCESS_KEY_ID",
-                secretAccessKey: "MOCK_SECRET_ACCESS_KEY"
+                secretAccessKey: "MOCK_SECRET_ACCESS_KEY",
+                convertEmptyValues: options && options.convertEmptyValues ? options.convertEmptyValues : false,
             };
         }
 

src/seeder.js

@@ -96,7 +96,8 @@ function fileExists(fileName) {
  */
 function unmarshalBuffer(json) {
   _.forEach(json, function(value, key) {
-    if (value.type==="Buffer") {
+    // Null check to prevent creation of Buffer when value is null
+    if (value !== null && value.type==="Buffer") {
       json[key]= new Buffer(value.data);
     }
   });