const exec = require('child_process').exec; const log = require('fancy-log'); const dataSources = require('../loopback/server/datasources.json'); module.exports = class Docker { constructor(name) { Object.assign(this, { id: name, name, isRandom: name == null, dbConf: Object.assign({}, dataSources.vn) }); } /** * Builds the database image and runs a container. It only rebuilds the * image when fixtures have been modified or when the day on which the * image was built is different to today. Some workarounds have been used * to avoid a bug with OverlayFS driver on MacOS. * * @param {Boolean} ci continuous integration environment argument * @param {String} networkName Name of the container network */ async run(ci, networkName = 'jenkins') { let d = new Date(); let pad = v => v < 10 ? '0' + v : v; let stamp = `${d.getFullYear()}-${pad(d.getMonth() + 1)}-${pad(d.getDate())}`; log('Building container image...'); await this.execP(`docker build --build-arg STAMP=${stamp} -t salix-db ./db`); log('Image built.'); let dockerArgs; if (this.isRandom) dockerArgs = '-p 3306'; else { try { await this.rm(); } catch (e) {} dockerArgs = `--name ${this.name} -p 3306:${this.dbConf.port}`; } let runChown = process.platform != 'linux'; let network = ''; if (ci) network = `--network="${networkName}"`; log('Starting container...'); const container = await this.execP(` docker run \ ${network} \ --env RUN_CHOWN=${runChown} \ -d ${dockerArgs} salix-db `); this.id = container.stdout.trim(); try { if (this.isRandom) { let inspect = await this.execP(`docker inspect -f "{{json .NetworkSettings}}" ${this.id}`); let netSettings = JSON.parse(inspect.stdout); if (ci) { this.dbConf.host = netSettings.Networks[networkName].IPAddress; this.dbConf.port = 3306; } else this.dbConf.port = netSettings.Ports['3306/tcp'][0]['HostPort']; } await this.wait(); } catch (err) { if (this.isRandom) await this.rm(); throw err; } } /** * Does the minium effort to start the database container, if it doesn't exists * calls the 'docker' task, if it is started does nothing. Keep in mind that when * you do not rebuild the docker you may be using an outdated version of it. * See the 'docker' task for more info. */ async start() { let state; try { let result = await this.execP(`docker inspect -f "{{json .State}}" ${this.id}`); state = JSON.parse(result.stdout); } catch (err) { return await this.run(); } switch (state.Status) { case 'running': return; case 'exited': await this.execP(`docker start ${this.id}`); await this.wait(); return; default: throw new Error(`Unknown docker status: ${state.Status}`); } } waitForHealthy() { return new Promise((resolve, reject) => { let interval = 100; let elapsedTime = 0; let maxInterval = 4 * 60 * 1000; log('Waiting for MySQL init process...'); async function checker() { elapsedTime += interval; let status; try { let result = await this.execP(`docker inspect -f "{{.State.Health.Status}}" ${this.id}`); status = result.stdout.trimEnd(); } catch (err) { return reject(new Error(err.message)); } if (status == 'unhealthy') return reject(new Error('Docker exited, please see the docker logs for more info')); if (status == 'healthy') { log('MySQL process ready.'); return resolve(); } if (elapsedTime >= maxInterval) reject(new Error(`MySQL not initialized whithin ${elapsedTime / 1000} secs`)); else setTimeout(bindedChecker, interval); } let bindedChecker = checker.bind(this); bindedChecker(); }); } wait() { return new Promise((resolve, reject) => { const mysql = require('mysql2'); let interval = 100; let elapsedTime = 0; let maxInterval = 4 * 60 * 1000; let myConf = { user: this.dbConf.username, password: this.dbConf.password, host: this.dbConf.host, port: this.dbConf.port, connectTimeout: maxInterval }; log('Waiting for MySQL init process...'); async function checker() { elapsedTime += interval; let state; try { let result = await this.execP(`docker inspect -f "{{json .State}}" ${this.id}`); state = JSON.parse(result.stdout); } catch (err) { return reject(new Error(err.message)); } if (state.Status === 'exited') return reject(new Error('Docker exited, please see the docker logs for more info')); let conn = mysql.createConnection(myConf); conn.on('error', () => {}); conn.connect(err => { conn.destroy(); if (!err) { log('MySQL process ready.'); return resolve(); } if (elapsedTime >= maxInterval) reject(new Error(`MySQL not initialized whithin ${elapsedTime / 1000} secs`)); else setTimeout(bindedChecker, interval); }); } let bindedChecker = checker.bind(this); bindedChecker(); }); } rm() { return this.execP(`docker stop ${this.id} && docker rm -v ${this.id}`); } /** * Promisified version of exec(). * * @param {String} command The exec command * @return {Promise} The promise */ execP(command) { return new Promise((resolve, reject) => { exec(command, (err, stdout, stderr) => { if (err) reject(err); else { resolve({ stdout: stdout, stderr: stderr }); } }); }); } };