feat: update mysql db helper

Soontao · Sep 1, 2020 · c43f7b3 · c43f7b3
1 parent 13066dd
commit c43f7b3
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/type/db_helper.ts b/src/type/db_helper.ts
@@ -67,14 +67,16 @@ export class BaseDBHelper implements DBHelper {
       colNameMapper = (v) => v;
     }
 
-    const fullTableName = buildNameWithQuote(schema, tableName);
+    const fullTableName = buildNameWithQuote(`\`${schema}\``, `\`${tableName}\``);
 
     const { sqlQuery, count, where, selectedFields } = transformQueryAst(
       query,
-      (col) => buildNameWithQuote(schema, tableName, colNameMapper(col))
+      (col) => buildNameWithQuote(`\`${schema}\``, `\`${tableName}\``, colNameMapper(col))
     );
 
-    const queryStatement = `SELECT ${isEmpty(selectedFields) ? '*' : selectedFields.join(', ')} FROM ${fullTableName} ${sqlQuery};`;
+    const sSelects = isEmpty(selectedFields) ? '*' : selectedFields.map((f) => `\`${f}\``).join(', ');
+
+    const queryStatement = `SELECT ${sSelects} FROM ${fullTableName} ${sqlQuery};`;
     let countStatement = undefined;
 
     if (count) {