PeerTube/server/initializers/migrator.js

99 lines
2.6 KiB
JavaScript
Raw Normal View History

'use strict'
const eachSeries = require('async/eachSeries')
2016-12-25 09:44:57 +01:00
const fs = require('fs')
const path = require('path')
const constants = require('./constants')
2016-12-11 21:50:51 +01:00
const db = require('./database')
const logger = require('../helpers/logger')
const migrator = {
migrate: migrate
}
function migrate (callback) {
2016-12-25 09:44:57 +01:00
db.Application.loadMigrationVersion(function (err, actualVersion) {
if (err) return callback(err)
2016-12-25 09:44:57 +01:00
// If there are a new migration scripts
if (actualVersion < constants.LAST_MIGRATION_VERSION) {
logger.info('Begin migrations.')
2016-12-25 09:44:57 +01:00
getMigrationScripts(function (err, migrationScripts) {
if (err) return callback(err)
2016-12-25 09:44:57 +01:00
eachSeries(migrationScripts, function (entity, callbackEach) {
executeMigration(actualVersion, entity, callbackEach)
}, function (err) {
if (err) return callback(err)
2016-12-25 09:44:57 +01:00
logger.info('Migrations finished. New migration version schema: %s', constants.LAST_MIGRATION_VERSION)
return callback(null)
})
})
} else {
return callback(null)
}
})
}
// ---------------------------------------------------------------------------
module.exports = migrator
2016-12-25 09:44:57 +01:00
// ---------------------------------------------------------------------------
function getMigrationScripts (callback) {
fs.readdir(path.join(__dirname, 'migrations'), function (err, files) {
if (err) return callback(err)
const filesToMigrate = []
files.forEach(function (file) {
// Filename is something like 'version-blabla.js'
const version = file.split('-')[0]
filesToMigrate.push({
version,
script: file
})
})
return callback(err, filesToMigrate)
})
}
function executeMigration (actualVersion, entity, callback) {
const versionScript = entity.version
// Do not execute old migration scripts
if (versionScript <= actualVersion) return callback(null)
// Load the migration module and run it
const migrationScriptName = entity.script
logger.info('Executing %s migration script.', migrationScriptName)
const migrationScript = require(path.join(__dirname, 'migrations', migrationScriptName))
db.sequelize.transaction().asCallback(function (err, t) {
if (err) return callback(err)
migrationScript.up({ transaction: t }, function (err) {
if (err) {
t.rollback()
return callback(err)
}
// Update the new migration version
db.Application.updateMigrationVersion(versionScript, t, function (err) {
if (err) {
t.rollback()
return callback(err)
}
2017-01-11 16:22:50 +01:00
t.commit().asCallback(callback)
2016-12-25 09:44:57 +01:00
})
})
})
}