feat: export columns method

taozhi8833998 · taozhi8833998 · commit 7027ff47648c · 2024-10-25T09:21:53.000+08:00
diff --git a/src/column.js b/src/column.js
@@ -213,4 +213,5 @@ export {
   columnOrderToSQL,
   columnReferenceDefinitionToSQL,
   fullTextSearchToSQL,
+  getDual,
 }
diff --git a/src/parser.js b/src/parser.js
@@ -1,3 +1,4 @@
+import { columnToSQL, getDual } from './column'
 import { exprToSQL } from './expr'
 import parsers from './parser.all'
 import astToSQL from './sql'
@@ -19,6 +20,13 @@ class Parser {
     return exprToSQL(expr)
   }
 
+  columnsToSQL(columns, tables, opt = DEFAULT_OPT) {
+    setParserOpt(opt)
+    if (!columns || columns === '*') return []
+    const isDual = getDual(tables)
+    return columns.map(col => columnToSQL(col, isDual))
+  }
+
   parse(sql, opt = DEFAULT_OPT) {
     const { database = (PARSER_NAME || 'mysql') } = opt
     setParserOpt(opt)
diff --git a/test/ast.spec.js b/test/ast.spec.js
@@ -1268,8 +1268,30 @@ describe('AST', () => {
             const sql = parser.exprToSQL(ast.where);
             expect(sql).to.equal('`id` = 1');
         });
+        
+        it('should be able to get columns from ast', () => {
+            const ast = parser.astify(`SELECT
+                campaign.id,
+                ad_group.id,
+                'http://' + ad_group_ad.ad.final_urls + '?tm=123' as url,
+                ad_group_ad.resource_name,
+                ad_group_ad.policy_summary.policy_topic_entries as policy_topic_entries,
+                ad_group_ad.policy_summary.policy_topic_entries:topic as topic,
+                ad_group_ad.policy_summary.policy_topic_entries:topic + '!' as topic2
+                FROM ad_group_ad`);
+            const columns = parser.columnsToSQL(ast.columns, ast.from);
+            expect(columns).to.be.eql([
+                '`campaign`.`id`',
+                '`ad_group`.`id`',
+                "'http://' + `ad_group_ad`.`ad`.`final_urls` + '?tm=123' AS `url`",
+                '`ad_group_ad`.`resource_name`',
+                '`ad_group_ad`.`policy_summary`.`policy_topic_entries` AS `policy_topic_entries`',
+                '`ad_group_ad`.`policy_summary`.`policy_topic_entries:topic` AS `topic`',
+                "`ad_group_ad`.`policy_summary`.`policy_topic_entries:topic` + '!' AS `topic2`"
+            ])
+        });
     })
-
+    
     describe('jsonb operator ast order', () => {
       it('should parse jsonb operator ast order correct', () => {
         const sql = `SELECT company.name

Original file line number	Diff line number	Diff line change
`@@ -213,4 +213,5 @@ export {`
`213`	`213`	`columnOrderToSQL,`
`214`	`214`	`columnReferenceDefinitionToSQL,`
`215`	`215`	`fullTextSearchToSQL,`
	`216`	`+ getDual,`
`216`	`217`	`}`