feat(common): add databaseExists method

mateonunez · Mar 1, 2023 · a25936e · a25936e
1 parent a59b9b8
commit a25936e
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/packages/drill/lib/queries/common.js b/packages/drill/lib/queries/common.js
@@ -6,6 +6,7 @@ export default async function (logger, db, sql) {
   return {
     selectData: (tableName, columns, where) => selectData(logger, db, sql, tableName, columns, where),
     [symbols.privateMethods]: {
+      databaseExists: (database) => databaseExists(logger, db, sql, database),
       createDatabase: (database, options) => createDatabase(logger, db, sql, database, options),
       dropDatabase: (database, options) => dropDatabase(logger, db, sql, database),
       createTable: (tableName, columns, options) => createTable(logger, db, sql, tableName, columns, options),
@@ -17,6 +18,16 @@ export default async function (logger, db, sql) {
   }
 }
 
+async function databaseExists (logger, db, sql, database) {
+  if (logger) logger.info(`Checking if database "${database}" exists.`)
+  const response = await db.query(sql`
+    SELECT SCHEMA_NAME
+    FROM INFORMATION_SCHEMA.SCHEMATA
+    WHERE SCHEMA_NAME = ${database}
+  `)
+  return response.length > 0
+}
+
 async function selectData (logger, db, sql, tableName, columns = '*', where = {}) {
   if (logger) logger.info(`Selecting data from "${tableName}".`)
   const columnDefinitions = []