Merge branch 'master' of https://github.com/complex-joins/carvis-api

JasonArkens17 · Aug 13, 2016 · 2dd3a3c · 2dd3a3c
2 parents 4f40716 + 8fa1f67
commit 2dd3a3c
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/src/db/stork/Model.js b/src/db/stork/Model.js
@@ -31,13 +31,19 @@ class Model {
     });
   }
 
-  // TODO
   updateOrCreate(obj) {
-
+    return this.db.find(obj)
+    .then((foundObj) => {
+      if (!foundObj) {
+        return this.create(obj);
+      } else {
+        return update(foundObj, obj);
+      }
+    });
   }
 
   create(obj) {
-    return this.db.insert(obj).into(this.table).returning(...Object.keys(obj));
+    return this.db.insert(obj).into(this.table).returning(...Object.keys(obj), 'id');
   }
 
   save(obj) {
@@ -48,7 +54,7 @@ class Model {
     return this.db(this.table)
       .update(updateObj, [...updateObj])
       .where(criteriaObj)
-      .returning(...Object.keys(updateObj));
+      .returning(...Object.keys(updateObj), 'id');
   }
 
   remove(obj) {