/* eslint-disable no-unused-expressions */ require('module-alias/register') const pg = require('pg') const config = require('@config/index') const path = require('path') const fs = require('fs') const dbVersion = require('./dbVersion.json') const versions = dbVersion.versions const schemaPath = path.resolve(__dirname, '../schema') if (!Array.isArray(versions) || versions.length === 0) { throw new Error('Schema version empty') } const client = new pg.Client({ host: config.database.host, port: config.database.port, user: config.database.user, password: config.database.pass, database: config.database.dbname }) // auto start function !(async function () { let flag = false await client.connect() await client.query(`select now()`) console.log('Database Connected') if (process.env['NODE_ENV'] === 'test') { if ('test' in dbVersion && Array.isArray(dbVersion.test)) { for (let i in dbVersion.test) { let qstr = fs.readFileSync(path.resolve(schemaPath, dbVersion.test[i].file)).toString() await client.query(qstr) } } } let version = -1 let checkTable = await client.query(`select exists(select 1 from "information_schema"."tables" where "table_schema" = $1 and "table_name" = $2) as exists`, ['public', 'version_ctrl']) if (checkTable.rows.length > 0 && checkTable.rows[0].exists === true) { let checkVersion = await client.query(`select max(version) as version from "public"."version_ctrl"`) if (checkVersion.rows.length > 0 && 'version' in checkVersion.rows[0] && isFinite(checkVersion.rows[0].version)) { version = checkVersion.rows[0].version } } let runVer = versions.filter(t => t.version > version).sort((a, b) => a.version - b.version) if (runVer.length === 0) return await client.query('begin') try { // write table query for (let i in runVer) { let qstr = fs.readFileSync(path.resolve(schemaPath, runVer[i].file)).toString() await client.query(qstr) let query = `insert into "public"."version_ctrl" ("version", "ctime", "querystr") values ($1, now(), $2)` let param = [runVer[i].version, qstr] await client.query(query, param) } await client.query('commit') } catch (err) { flag = true console.log(err) await client.query('rollback') } await client.end() if (flag) throw new Error('Not Finish') })().then(() => { console.log('Finish') process.exit(0) }).catch(err => { console.error(err) process.exit(1) })