MySQL support

Author: MIYASHITA, Akihiro

README.md

@@ -15,6 +15,10 @@ This software is released under the [MIT License](LICENSE).
 
 ## Release Note
 
+### Version 2.6.0
+
+* [Feat] MySQL support
+
 ### Version 2.5.0
 
 * [Feat] Allow multiple checks in a single testcase (Runner)

lib/connection.js

@@ -5,6 +5,7 @@ const knex = require('knex');
 const path = require('path');
 const util = require('util');
 const _ = require('./i18n').text;
+const exec = util.promisify(cp.exec);
 
 const NULL_MARKER = '__null__';
 const SQL_ERROR = {
@@ -57,6 +58,12 @@ function typeOf(x) {
 	return Object.prototype.toString.call(x).slice(8, -1).toLowerCase();
 }
 
+async function sleep(ms) {
+	return new Promise((res, rej) => {
+		setTimeout(() => res(), ms);
+	});
+}
+
 class SQLiteCasette {
 	async newConnection(options) {
 		const dbfile = (options || {}).file || 'db.sqlite';
@@ -131,13 +138,6 @@ class SQLiteCasette {
 
 class PostgreSQLCasette {
 	async newConnection(options) {
-		const exec = util.promisify(cp.exec);
-		async function sleep(ms) {
-			return new Promise((res, rej) => {
-				setTimeout(() => res(), ms);
-			});
-		}
-
 		async function startServer() {
 			return new Promise((_res, _rej) => {
 				let done = false;
@@ -262,8 +262,130 @@ class PostgreSQLCasette {
 	}
 }
 
+class MySQLCassette {
+	async newConnection(options) {
+		async function startServer() {
+			return new Promise((_res, _rej) => {
+				let done = false;
+				const res = (...args) => {
+					if (!done) {
+						done = true;
+						_res(...args);
+					}
+				}
+				const rej = (...args) => {
+					if (!done) {
+						done = true;
+						_rej(...args);
+					}
+				}
+				sleep(10000).then(() => rej(new Error('Timeout database connection')));
+				const p = cp.spawn('mysqld_safe', [`--datadir=${process.env.MYSQL_DATA}`], {
+					detached: true,
+				});
+				p.stdout.on('data', data => {
+					const message = data.toString();
+					if (options.verbose) {
+						console.log(message);
+					}
+					if (message.includes('database system is ready to accept connections')) {
+						res();
+					}
+				});
+				p.stderr.on('data', data => {
+					const message = data.toString();
+					if (options.verbose) {
+						console.error(message);
+					}
+					if (message.includes('database system is ready to accept connections')) {
+						res();
+					}
+				});
+			});
+		}
+
+		if (!fs.existsSync('/run/mysqld/mysqld.sock')) {
+			for (let retries = 5; retries > 0; retries--) {
+				try {
+					await startServer();
+					break;
+				} catch (e) {
+					if (options.verbose) {
+						console.error(e);
+					}
+				}
+				await sleep(1000);
+			}
+		}
+		for (let retries = 20; retries > 0; retries--) {
+			try {
+				await exec("mysql -e 'CREATE DATABASE IF NOT EXISTS track'");
+				const conn = knex({
+					client: 'mysql',
+					connection: {
+						host: '127.0.0.1',
+						port: 3306,
+						user: 'root',
+						database: 'track',
+					},
+				});
+				await conn.raw("SELECT 1");
+				if (options && options.clean) {
+					try {
+						let tables = await conn.raw("SHOW TABLES");
+						for (let table of tables.rows) {
+							await conn.raw(`DROP TABLE IF EXISTS ${table.Tables_in_track} CASCADE`);
+						}
+					} catch (e2) { console.error(e2); }
+				}
+				return conn;
+			} catch (e) {
+				await sleep(500);
+			}
+		}
+		throw Error("Failed to start MySQL server");
+	}
+	explainSql(s) {
+		// return `EXPLAIN ${s}`;
+	}
+	tableSchemaSql() {
+		return `
+			SELECT
+				ordinal_position AS "order",
+				column_name AS name,
+				data_type AS raw_type
+			FROM information_schema.columns
+			WHERE
+				table_schema = 'track' AND
+				table_name = ?
+			ORDER BY ordinal_position
+		`;
+	}
+	updateAutoIncrementSql(table, column) {
+		return `ALTER TABLE ${table} AUTO_INCREMENT = ? + 1`;
+	}
+	lastValueSql(table, idCol) {
+		return `SELECT * FROM ${table} WHERE ${idCol} = LAST_INSERT_ID()`;
+	}
+	async listFk(table, conn) {
+		return await conn.query(`
+			SELECT
+				kcu.column_name AS column,
+				ccu.table_name AS foreign_table,
+				ccu.column_name AS foreign_column
+			FROM information_schema.table_constraints AS tc
+			INNER JOIN information_schema.key_column_usage AS kcu ON tc.constraint_name = kcu.constraint_name
+			INNER JOIN information_schema.constraint_column_usage AS ccu ON ccu.constraint_name = tc.constraint_name
+			WHERE tc.constraint_type = 'FOREIGN KEY' AND tc.table_name = '${table}'
+		`);
+	}
+	errorOf(e) {
+		return [SQL_ERROR.UNKNOWN];
+	}
+}
+
 /**
- * Handles SQLite connection
+ * Handles connection
  */
 class Connection {
 	static timeout(client, extra) {
@@ -284,6 +406,10 @@ class Connection {
 			case "postgresql":
 				casette = new PostgreSQLCasette(options);
 				break;
+			case "my":
+			case "mysql":
+				casette = new MySQLCassette(options);
+				break;	
 			default:
 				casette = new SQLiteCasette(options);
 				break;

package-lock.json

@@ -1,6 +1,6 @@
 {
   "name": "track-db-test-library",
-  "version": "2.5.0",
+  "version": "2.6.0-rc1",
   "description": "Test utility for Track database challenges",
   "main": "index.js",
   "scripts": {