destroy worker before ending

giblet22 · Apr 18, 2016 · ba287d9 · ba287d9
1 parent 1411939
commit ba287d9
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 7 deletions.
diff --git a/examples/kinesis-consumer/index.js b/examples/kinesis-consumer/index.js
@@ -21,10 +21,14 @@ exports.handler = function (event, context) {
     console.log('PAYLOAD:', payload);
     return worker.addToDB(JSON.parse(payload));
   }).then(function (result) {
-      console.log('SUCCESS:', result);
-      return context.succeed('Success');
+      return worker.destroy(function () {
+        console.log('SUCCESS:', result);
+        return context.succeed('Success');
+      });
   }).catch(function (err) {
-    console.log('FAILURE:', err);
-    return context.fail(err);
+      return worker.destroy(function () {
+        console.log('FAILURE:', err);
+        return context.fail(err);
+      });
   });
 };
diff --git a/index.js b/index.js
@@ -14,9 +14,9 @@ var Worker = function (loggingFn) {
   this.logger = loggingFn;
 };
 
-//Worker.prototype.destroy = function () {
-//  return this.bookshelf.knex.destroy();
-//};
+Worker.prototype.destroy = function () {
+  return this.bookshelf.knex.destroy();
+};
 
 Worker.prototype.addToDB = function (changeset) {
   this.changeset = changeset;