/
table-creator.js
77 lines (71 loc) · 2 KB
/
table-creator.js
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
const {
getTable,
getLockTableName,
getLockTableNameWithSchema,
getTableName,
} = require('./table-resolver');
function ensureTable(tableName, schemaName, trxOrKnex) {
const lockTable = getLockTableName(tableName);
return getSchemaBuilder(trxOrKnex, schemaName)
.hasTable(tableName)
.then((exists) => {
return !exists && _createMigrationTable(tableName, schemaName, trxOrKnex);
})
.then(() => {
return getSchemaBuilder(trxOrKnex, schemaName).hasTable(lockTable);
})
.then((exists) => {
return (
!exists && _createMigrationLockTable(lockTable, schemaName, trxOrKnex)
);
})
.then(() => {
return getTable(trxOrKnex, lockTable, schemaName).select('*');
})
.then((data) => {
return (
!data.length && _insertLockRowIfNeeded(tableName, schemaName, trxOrKnex)
);
});
}
function _createMigrationTable(tableName, schemaName, trxOrKnex) {
return getSchemaBuilder(trxOrKnex, schemaName).createTable(
getTableName(tableName),
function (t) {
t.increments();
t.string('name');
t.integer('batch');
t.timestamp('migration_time');
}
);
}
function _createMigrationLockTable(tableName, schemaName, trxOrKnex) {
return getSchemaBuilder(trxOrKnex, schemaName).createTable(
tableName,
function (t) {
t.increments('index').primary();
t.integer('is_locked');
}
);
}
function _insertLockRowIfNeeded(tableName, schemaName, trxOrKnex) {
const lockTableWithSchema = getLockTableNameWithSchema(tableName, schemaName);
return trxOrKnex
.select('*')
.from(lockTableWithSchema)
.then((data) => {
return !data.length
? trxOrKnex.from(lockTableWithSchema).insert({ is_locked: 0 })
: null;
});
}
//Get schema-aware schema builder for a given schema nam
function getSchemaBuilder(trxOrKnex, schemaName) {
return schemaName
? trxOrKnex.schema.withSchema(schemaName)
: trxOrKnex.schema;
}
module.exports = {
ensureTable,
getSchemaBuilder,
};