-
Notifications
You must be signed in to change notification settings - Fork 2.1k
/
table-creator.js
62 lines (58 loc) · 1.69 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
import {
getTable,
getLockTableName,
getLockTableNameWithSchema,
getTableName,
} from './table-resolver';
export function ensureTable(tableName, schemaName, trxOrKnex) {
const lockTable = getLockTableName(tableName);
const lockTableWithSchema = getLockTableNameWithSchema(tableName, schemaName);
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 &&
trxOrKnex.into(lockTableWithSchema).insert({ is_locked: 0 })
);
});
}
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');
}
);
}
//Get schema-aware schema builder for a given schema nam
export function getSchemaBuilder(trxOrKnex, schemaName) {
return schemaName
? trxOrKnex.schema.withSchema(schemaName)
: trxOrKnex.schema;
}