2018-07-03 11:33:53 +08:00
# dt-sql-parser
2018-07-02 18:01:01 +08:00
2021-01-04 14:44:26 +08:00
[![NPM version][npm-image]][npm-url] [![NPM downloads][download-img]][download-url]
2020-05-12 09:39:49 +08:00
2020-12-17 11:17:14 +08:00
English | [简体中文 ](./README-zh_CN.md )
2020-12-15 15:04:46 +08:00
2020-12-17 16:46:42 +08:00
[npm-image]: https://img.shields.io/npm/v/dt-sql-parser.svg?style=flat-square
[npm-url]: https://www.npmjs.com/package/dt-sql-parser
2020-12-15 15:04:46 +08:00
2021-01-04 14:44:26 +08:00
[download-img]: https://img.shields.io/npm/dm/dt-sql-parser.svg?style=flat
[download-url]: https://www.npmjs.com/package/dt-sql-parser
2020-12-17 17:15:46 +08:00
dt-sql-parser is a **SQL Parser** project built with [ANTLR4 ](https://github.com/antlr/antlr4 ), and it's mainly for the **BigData** domain. The [ANTLR4 ](https://github.com/antlr/antlr4 ) generated the basic Parser, Visitor, and Listener, so it's easy to complete the **syntax validation** , **tokenizer** , **traverse** the AST, and so on features.
2020-12-15 15:04:46 +08:00
2020-12-17 17:15:46 +08:00
Besides, it' provides some helper methods, like **split** SQL, and filter the `--` and `/**/` types of comments in SQL.
2018-07-02 18:01:01 +08:00
2020-12-17 11:17:14 +08:00
Supported SQL:
- MySQL
- Flink SQL
- Spark SQL
- Hive SQL
- PL/SQL
2018-07-05 11:29:39 +08:00
2020-12-17 17:21:40 +08:00
>Tips: This project is the default for Javascript language, also you can try to compile it to other languages if you need.
2020-12-15 15:24:37 +08:00
## Installation
2018-07-05 11:29:39 +08:00
2020-12-17 16:46:42 +08:00
```bash
2020-12-15 15:04:46 +08:00
// use npm
npm i dt-sql-parser --save
// use yarn
yarn add dt-sql-parser
```
2020-12-15 15:24:37 +08:00
## Usage
2018-07-05 11:29:39 +08:00
2020-12-17 16:46:42 +08:00
### Syntax Validation
2020-08-28 13:29:47 +08:00
2020-12-17 17:15:46 +08:00
First, we need to import the **Parser** object from `dt-sql-parser` , the different language needs
different Parser, so if you need to handle the **Flink SQL** , you can import the **FlinkSQL Parser** .
2020-12-17 16:46:42 +08:00
2020-12-17 17:15:46 +08:00
The below is a **GenericSQL Parser** example:
2018-08-16 19:30:22 +08:00
2020-12-15 15:04:46 +08:00
```javascript
2020-12-17 16:46:42 +08:00
import { GenericSQL } from 'dt-sql-parser';
2020-12-15 15:04:46 +08:00
2020-12-17 16:46:42 +08:00
const parser = new GenericSQL();
const correctSql = 'select id,name from user1;';
const errors = parser.validate(correctSql);
console.log(errors);
```
2018-08-16 19:30:22 +08:00
2020-12-17 17:18:45 +08:00
Output:
2020-12-17 16:46:42 +08:00
```javascript
2018-08-16 19:30:22 +08:00
/*
2020-12-17 16:46:42 +08:00
[]
2018-08-16 19:30:22 +08:00
*/
```
2020-12-17 17:18:45 +08:00
Validate failed:
2018-10-08 17:58:36 +08:00
2020-12-15 15:04:46 +08:00
```javascript
2020-12-17 16:46:42 +08:00
const incorrectSql = 'selec id,name from user1;'
const errors = parser.validate(incorrectSql);
console.log(errors);
```
2020-08-28 13:29:47 +08:00
2020-12-17 17:18:45 +08:00
Output:
2018-10-08 17:58:36 +08:00
2020-12-17 16:46:42 +08:00
```javascript
2020-12-15 15:04:46 +08:00
/*
2020-12-17 16:46:42 +08:00
[
{
endCol: 5,
endLine: 1,
startCol: 0,
startLine: 1,
message: "mismatched input 'SELEC' expecting {< EOF > , 'ALTER', 'ANALYZE', 'CALL', 'CHANGE', 'CHECK', 'CREATE', 'DELETE', 'DESC', 'DESCRIBE', 'DROP', 'EXPLAIN', 'GET', 'GRANT', 'INSERT', 'KILL', 'LOAD', 'LOCK', 'OPTIMIZE', 'PURGE', 'RELEASE', 'RENAME', 'REPLACE', 'RESIGNAL', 'REVOKE', 'SELECT', 'SET', 'SHOW', 'SIGNAL', 'UNLOCK', 'UPDATE', 'USE', 'BEGIN', 'BINLOG', 'CACHE', 'CHECKSUM', 'COMMIT', 'DEALLOCATE', 'DO', 'FLUSH', 'HANDLER', 'HELP', 'INSTALL', 'PREPARE', 'REPAIR', 'RESET', 'ROLLBACK', 'SAVEPOINT', 'START', 'STOP', 'TRUNCATE', 'UNINSTALL', 'XA', 'EXECUTE', 'SHUTDOWN', '--', '(', ';'}"
}
]
2020-12-15 15:04:46 +08:00
*/
```
2020-08-28 13:29:47 +08:00
2020-12-17 17:15:46 +08:00
We instanced a Parser object, and use the **validate** method to check the SQL syntax, if failed
returns an array object includes **error** message.
2020-12-17 16:46:42 +08:00
### Tokenizer
2018-10-08 17:58:36 +08:00
2020-12-17 17:32:40 +08:00
Get all **tokens** by the Parser:
2018-10-08 17:58:36 +08:00
2020-12-15 15:04:46 +08:00
```javascript
import { GenericSQL } from 'dt-sql-parser';
2020-08-28 13:29:47 +08:00
2020-12-15 15:04:46 +08:00
const parser = new GenericSQL()
const sql = 'select id,name,sex from user1;'
const tokens = parser.getAllTokens(sql)
console.log(tokens)
/*
[
2020-12-17 10:12:05 +08:00
{
channel: 0
2020-12-15 15:04:46 +08:00
column: 0
line: 1
source: [SqlLexer, InputStream]
start: 0
stop: 5
tokenIndex: -1
type: 137
_text: null
text: "SELECT"
2020-12-17 10:12:05 +08:00
},
...
2020-12-15 15:04:46 +08:00
]
*/
```
2018-10-08 18:02:26 +08:00
2020-12-15 15:04:46 +08:00
### Visitor
2020-12-17 16:46:42 +08:00
Traverse the tree node by the Visitor:
2018-10-08 17:58:36 +08:00
2020-12-15 15:04:46 +08:00
```javascript
import { GenericSQL, SqlParserVisitor } from 'dt-sql-parser';
const parser = new GenericSQL()
const sql = `select id,name from user1;`
// parseTree
const tree = parser.parse(sql)
class MyVisitor extends SqlParserVisitor {
2020-12-15 15:24:37 +08:00
// overwrite visitTableName
2020-12-15 15:04:46 +08:00
visitTableName(ctx) {
let tableName = ctx.getText().toLowerCase()
console.log('TableName', tableName)
}
2020-12-15 15:24:37 +08:00
// overwrite visitSelectElements
2020-12-15 15:04:46 +08:00
visitSelectElements(ctx) {
let selectElements = ctx.getText().toLowerCase()
console.log('SelectElements', selectElements)
}
}
const visitor = new MyVisitor()
visitor.visit(tree)
/*
SelectElements id,name
TableName user1
*/
```
2020-12-17 16:46:42 +08:00
> Tips: The node's method name can be found in the Visitor file under the corresponding SQL directory
2020-12-15 15:04:46 +08:00
### Listener
2020-12-17 17:15:46 +08:00
Access the specified node in the AST by the Listener
2020-12-15 15:04:46 +08:00
```javascript
import { GenericSQL, SqlParserListener } from 'dt-sql-parser';
const parser = new GenericSQL();
const sql = 'select id,name from user1;'
// parseTree
const tree = parser.parse(sql)
class MyListener extends SqlParserListener {
enterTableName(ctx) {
let tableName = ctx.getText().toLowerCase()
console.log('TableName', tableName)
}
enterSelectElements(ctx) {
let selectElements = ctx.getText().toLowerCase()
log('SelectElements', selectElements)
}
}
const listenTableName = new MyListener();
parser.listen(listenTableName, tree);
/*
SelectElements id,name
TableName user1
*/
```
2018-10-08 17:58:36 +08:00
2020-12-17 16:46:42 +08:00
> Tips: The node's method name can be found in the Listener file under the corresponding SQL directory
### Clean
2020-12-17 17:15:46 +08:00
Clear the **comments** and **spaces** before and after
2020-12-17 16:46:42 +08:00
```javascript
import { cleanSql } from 'dt-sql-parser';
const sql = `-- comment comment
select id,name from user1; `
const cleanedSql = cleanSql(sql)
console.log(cleanedSql)
/*
select id,name from user1;
*/
```
### Split SQL
2020-12-17 17:15:46 +08:00
When the SQL text is very big, you can think about to split it by `;` , and handle it by each line.
2020-12-17 16:46:42 +08:00
```javascript
import { splitSql } from 'dt-sql-parser';
const sql = `select id,name from user1;
select id,name from user2;`
const sqlList = splitSql(sql)
console.log(sqlList)
/*
["select id,name from user1;", "\nselect id,name from user2;"]
*/
```
### Other API
2020-08-28 13:29:47 +08:00
2020-12-17 16:46:42 +08:00
- parserTreeToString(input: string)
2019-09-25 15:57:25 +08:00
2020-12-17 16:46:42 +08:00
Parse the input and convert the AST to a `List-like` tree string.
2018-10-08 17:58:36 +08:00
2020-12-15 15:24:37 +08:00
## Roadmap
2018-11-30 13:40:48 +08:00
2020-12-15 15:04:46 +08:00
- Auto-complete
2020-12-30 16:36:54 +08:00
- Code formatting
2018-11-30 13:40:48 +08:00
2020-12-15 15:24:37 +08:00
## License
2018-11-30 13:40:48 +08:00
2020-12-17 16:46:42 +08:00
[MIT ](./LICENSE )