Adding optional migration callback to make migrations more general in…

… usage.
xavierlacot · Jul 26, 2011 · 1bf1bb0 · 1bf1bb0
1 parent bf1e55a
commit 1bf1bb0
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/joli.js b/joli.js
@@ -398,7 +398,7 @@ joli.Models.prototype = {
     });
   },
 
-  migrate: function(version) {
+  migrate: function(version, migrationCallback) {
     // create migration table
     var query = 'CREATE TABLE IF NOT EXISTS ' + this.migration.table + ' (version)';
     joli.connection.execute(query);
@@ -408,6 +408,14 @@ joli.Models.prototype = {
         var query = 'DROP TABLE IF EXISTS ' + modelName;
         joli.connection.execute(query);
       });
+
+      // optional migration callback
+      if(migrationCallback && typeof(migrationCallback) == "function") {
+      	migrationCallback({
+      		table:this.migration.table,
+      		newVersion:version
+      	});
+      }
 
       // insert migration
       this.migration.setVersion(version);