(IMPROVE) Implement optimistic locking for updatePartialSql

myteksi · Jan 12, 2015 · 5f566e6 · 5f566e6
1 parent 0d55632
commit 5f566e6
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/lib/driver.js b/lib/driver.js
@@ -122,16 +122,25 @@ var Driver = Class.extend({
       }
     }
 
+    if (model.columns["lockVersion"]) {
+      columnNamesSql.push(this.escapeColumnName('lock_version') + ' = lock_version + 1');
+    }
+
+    if (model.columns["updatedAt"]) {
+      columnNamesSql.push(this.escapeColumnName('updated_at') + ' = NOW()');
+    }
+
     if (columnNamesSql.length === 0) {
       throw new Error("No columns to update.");
     }
 
     var idColumndName = model.getIdPropertyName();
     values.push(id);
 
-    var sql = util.format('UPDATE %s SET %s WHERE %s = %s', model.tableName, columnNamesSql.join(','), idColumndName, this.getValuesSubstitutionString(valueSubstitutionIndex++));
+    var defaultUpdateQuery = 'UPDATE %s SET %s WHERE %s = %s';
+    var sql = util.format(defaultUpdateQuery, model.tableName, columnNamesSql.join(','), idColumndName, this.getValuesSubstitutionString(valueSubstitutionIndex++));
     var results = { sql: sql, values: values };
-    //console.log(results);
+
     return results;
   },