fix: sqlify to origin table name in bigquery

Author: taozhi8833998

pegjs/bigquery.pegjs

@@ -195,6 +195,19 @@
     columnList.clear()
     columns.forEach(col => columnList.add(col))
   }
+  
+  function getSurroundFromLiteralType(literal) {
+    switch (literal.type) {
+      case 'double_quote_string':
+        return '"'
+      case 'single_quote_string':
+        return "'"
+      case 'backticks_quote_string':
+        return '`'
+      default:
+        return ''
+    }
+  }
 
   const cmpPrefixMap = {
     '+': true,
@@ -1926,21 +1939,23 @@ join_op
   / k:KW_INNER? __ KW_JOIN { return k ? `${k[0].toUpperCase()} JOIN` : 'JOIN'; }
 
 table_name
-  = db:ident_without_kw schema:(__ DOT __ ident_without_kw) tail:(__ DOT __ ident_without_kw) {
-      const obj = { db: null, table: db };
+  = db:ident_without_kw_type schema:(__ DOT __ ident_without_kw_type) tail:(__ DOT __ ident_without_kw_type) {
+      const obj = { db: null, table: db.value };
       if (tail !== null) {
-        obj.db = db;
-        obj.catalog = db;
-        obj.schema = schema[3];
-        obj.table = tail[3];
+        obj.db = db.value;
+        obj.catalog = db.value;
+        obj.schema = schema[3].value;
+        obj.table = tail[3].value;
+        obj.surround = { table: getSurroundFromLiteralType(tail[3]), db: getSurroundFromLiteralType(db), schema: getSurroundFromLiteralType(schema[3]) };
       }
       return obj;
     }
-  / dt:ident_without_kw tail:(__ DOT __ ident_without_kw)? {
-      const obj = { db: null, table: dt };
+  / dt:ident_without_kw_type tail:(__ DOT __ ident_without_kw_type)? {
+      const obj = { db: null, table: dt.value, surround: { table: getSurroundFromLiteralType(dt) } };
       if (tail !== null) {
-        obj.db = dt;
-        obj.table = tail[3];
+        obj.db = dt.value;
+        obj.table = tail[3].value;
+        obj.surround = { table: getSurroundFromLiteralType(tail[3]), db: getSurroundFromLiteralType(dt) };
       }
       return obj;
     }

src/insert.js

@@ -19,7 +19,7 @@ function partitionToSQL(partition) {
   if (!partition) return ''
   const partitionArr = ['PARTITION', '(']
   if (Array.isArray(partition)) {
-    partitionArr.push(partition.map(identifierToSql).join(', '))
+    partitionArr.push(partition.map(partitionItem => identifierToSql(partitionItem)).join(', '))
   } else {
     const { value } = partition
     partitionArr.push(value.map(exprToSQL).join(', '))

src/tables.js

@@ -56,7 +56,7 @@ function tableHintToSQL(tableHintExpr) {
       result.push(toUpper(keyword), '=', exprToSQL(expr))
       break
     case 'index':
-      result.push(toUpper(prefix), toUpper(keyword), parentheses ? `(${expr.map(identifierToSql).join(', ')})` : `= ${identifierToSql(expr)}`)
+      result.push(toUpper(prefix), toUpper(keyword), parentheses ? `(${expr.map(indexItem => identifierToSql(indexItem)).join(', ')})` : `= ${identifierToSql(expr)}`)
       break
     default:
       result.push(exprToSQL(expr))
@@ -114,11 +114,11 @@ function generateVirtualTable(stmt) {
 
 function tableToSQL(tableInfo) {
   if (toUpper(tableInfo.type) === 'UNNEST') return unnestToSQL(tableInfo)
-  const { table, db, as, expr, operator, prefix: prefixStr, schema, server, suffix, tablesample, temporal_table, table_hint } = tableInfo
-  const serverName = identifierToSql(server)
-  const database = identifierToSql(db)
-  const schemaStr = identifierToSql(schema)
-  let tableName = table && identifierToSql(table)
+  const { table, db, as, expr, operator, prefix: prefixStr, schema, server, suffix, tablesample, temporal_table, table_hint, surround = {} } = tableInfo
+  const serverName = identifierToSql(server, false, surround.server)
+  const database = identifierToSql(db, false, surround.db)
+  const schemaStr = identifierToSql(schema, false, surround.schema)
+  let tableName = table && identifierToSql(table, false, surround.table)
   if (expr) {
     const exprType = expr.type
     switch (exprType) {

src/util.js

@@ -151,11 +151,12 @@ function columnIdentifierToSql(ident) {
   }
 }
 
-function identifierToSql(ident, isDual) {
-  const { database } = getParserOpt()
+function identifierToSql(ident, isDual, surround) {
   if (isDual === true) return `'${ident}'`
   if (!ident) return
   if (ident === '*') return ident
+  if (surround) return `${surround}${ident}${surround}`
+  const { database } = getParserOpt()
   switch (database && database.toLowerCase()) {
     case 'mysql':
     case 'mariadb':

test/bigquery.spec.js

@@ -891,6 +891,13 @@ describe('BigQuery', () => {
         "SELECT SAFE_CAST(SPLIT(t1.CREATED_BY, 'a')[0] AS STRING) AS FIRST_SPLIT_ARG FROM table1 AS t1 WHERE CREATED_BY = 'node-sql-parser' AND FIRST_SPLIT_ARG = @FIRST_SPLIT_ARG LIMIT 1"
       ]
     },
+    {
+      title: 'quoted table name',
+      sql: [
+        'SELECT COUNT(*) FROM `bigquery-public-data.blah.events_*`',
+        'SELECT COUNT(*) FROM `bigquery-public-data.blah.events_*`'
+      ]
+    }
   ]
 
   SQL_LIST.forEach(sqlInfo => {