aboutsummaryrefslogtreecommitdiffhomepage
path: root/server/initializers/migrator.js
blob: 233ee2bddf294e2be4a21c86125a31db13db44cf (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
'use strict'

const eachSeries = require('async/eachSeries')
const fs = require('fs')
const path = require('path')

const constants = require('./constants')
const db = require('./database')
const logger = require('../helpers/logger')

const migrator = {
  migrate: migrate
}

function migrate (callback) {
  db.Application.loadMigrationVersion(function (err, actualVersion) {
    if (err) return callback(err)

    // If there are a new migration scripts
    if (actualVersion < constants.LAST_MIGRATION_VERSION) {
      logger.info('Begin migrations.')

      getMigrationScripts(function (err, migrationScripts) {
        if (err) return callback(err)

        eachSeries(migrationScripts, function (entity, callbackEach) {
          executeMigration(actualVersion, entity, callbackEach)
        }, function (err) {
          if (err) return callback(err)

          logger.info('Migrations finished. New migration version schema: %s', constants.LAST_MIGRATION_VERSION)
          return callback(null)
        })
      })
    } else {
      return callback(null)
    }
  })
}

// ---------------------------------------------------------------------------

module.exports = migrator

// ---------------------------------------------------------------------------

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 = parseInt(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)

    const options = {
      transaction: t,
      queryInterface: db.sequelize.getQueryInterface(),
      Sequelize: db.Sequelize
    }
    migrationScript.up(options, 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)
        }

        t.commit().asCallback(callback)
      })
    })
  })
}