added save method

LICENSE.md

@@ -1,6 +1,6 @@
 The MIT License (MIT)
 
-Copyright (c) 2014 Brian Detering
+Copyright (c) 2015 Brian Detering
 
 Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
 

README.md

@@ -146,6 +146,17 @@ sql.insert('fruit', [
 sql.replace('fruit', { uniqueKey: 5, name: 'plum', isRipe: false, color: 'brown' });
 ```
 
+###save(table, insertObject, \*callback)
+Inserts a new row if no duplicate unique or primary keys
+are found, else it updates that row.
+```sql
+INSERT INTO fruit (uniqueKey, isRipe) VALUES (5, 0)
+ON DUPLICATE KEY UPDATE uniqueKey=5, isRipe=0
+```
+```javascript
+sql.save('fruit', { uniqueKey: 5, isRipe: false });
+```
+
 ###update(table, setValues, \*whereEqualsObject, \*callback)
 ```javascript
 sql.update('fruit', { isRipe: false }, { name: 'grape' })

mysql-wrap.js

@@ -234,6 +234,30 @@ var createMySQLWrap = function (connection, options) {
         );
     };
 
+    self.save = function (table, rowRaw, callback) {
+        var prepareSaveRows = function () {
+            var insertRow = prepareInsertRows(rowRaw);
+            var setValues = [];
+            var setSQL = _.map(rowRaw, function (val, key) {
+                setValues.push(key, val);
+                return '?? = ?';
+            }).join(', ');
+
+            return {
+                sql: insertRow.sql + ' ON DUPLICATE KEY UPDATE ' + setSQL,
+                values: insertRow.values.concat(setValues)
+            };
+        };
+
+        var row = prepareSaveRows();
+
+        return self.query(
+            'INSERT INTO ?? ' + row.sql,
+            [table].concat(row.values),
+            callback
+        );
+    };
+
     self.update = function (table, setData, whereEquals, callback) {
         var prepareSetRows = function (setData) {
             var values = [];

package.json

@@ -1,6 +1,6 @@
 {
   "name": "mysql-wrap",
-  "version": "0.5.2",
+  "version": "0.5.3",
   "description": "Mysql interface and helper functions, wrapping node-mysql",
   "main": "mysql-wrap.js",
   "scripts": {

test/shared_tests.js

@@ -285,6 +285,48 @@ module.exports = function (connection, createNodeMySQL) {
         .done();
     };
 
+    self.save = function (test) {
+        test.expect(3);
+        var self = this;
+        self.sql.save('table', { unique: 'd', field: 'baz' })
+        .then(function (res) {
+            test.strictEqual(res.affectedRows, 1, 'returns affectedRows');
+            test.strictEqual(res.insertId, 4, 'returns insert id');
+            connection.query(
+                'SELECT * FROM `table` WHERE `id` = 4',
+                function (err, res) {
+                    test.deepEqual(res, [
+                        { id: 4, unique: 'd', field: 'baz' }
+                    ], 'inserts into database');
+                    test.done();
+                }
+            );
+        })
+        .done();
+    };
+
+    self.save_updates_row_with_same_unique_key = function (test) {
+        test.expect(1);
+        var self = this;
+        self.sql.replace('table', { unique: 'd', field: 'baz' })
+        .then(function () {
+            return self.sql.replace('table', { unique: 'd', field: 'bar' });
+        })
+        .then(function () {
+            connection.query(
+                'SELECT * FROM `table` WHERE `unique` = "d"',
+                function (err, res) {
+                    test.deepEqual(res, [
+                        { id: 5, unique: 'd', field: 'bar' }
+                    ], 'replaces existing row');
+                    test.done();
+                }
+            );
+        })
+        .done();
+    };
+
+
     self.update = function (test) {
         test.expect(1);
         var self = this;