Merge pull request #5 from brandsExclusive/upsert

Upsert

lib/createGeneric.js

@@ -12,12 +12,13 @@ function createGeneric(table, settable_fields, object) {
     return "$" + counter;
   });
 
-  let query = `INSERT INTO ${table} (`
-    + set_fields.join(",")
-    + ") VALUES ("
-    + placeholders
-    + ") RETURNING *;";
-
+  let query = `
+    INSERT INTO ${table} (
+      ${set_fields.join(",")}
+    ) VALUES (
+      ${placeholders}
+    ) RETURNING *;
+  `
   return db.getFirstRow(query, values);
 }
-module.exports = createGeneric;
+module.exports = createGeneric;

lib/index.js

@@ -50,4 +50,5 @@ module.exports.getCount = (text, values) => {
 };
 
 module.exports.createGeneric = require('./createGeneric');
+module.exports.upsert = require('./upsert');
 module.exports.queryBuilder = require('./queryBuilder');

lib/upsert.js

@@ -0,0 +1,36 @@
+const db = require('./index.js');
+module.exports = function (object, table, settable_fields, upsert_index) {
+  if(!(upsert_index in object)){
+    throw new Error(upsert_index+" is required for upsert.")
+  }
+  let set_fields = settable_fields.filter(function(field){
+    return Object.keys(object).includes(field);
+  });
+
+  let values = set_fields.map(function(field){
+    return object[field];
+  });
+  let counter = 0;
+  let placeholders = set_fields.map(function() {
+    counter = counter + 1;
+    return "$" + counter;
+  });
+
+  let query = `
+    INSERT INTO ${table} (
+      ${set_fields.join(",")}
+    ) VALUES (
+      ${placeholders}
+    )
+    ON CONFLICT (${upsert_index})
+    DO
+      UPDATE SET (
+        ${set_fields.join(",")}
+      ) = (
+        ${placeholders}
+      )
+      WHERE ${table}.${upsert_index} = '${object[upsert_index]}'
+      RETURNING *;
+  `
+  return db.getFirstRow(query, values);
+}