mirror of
https://github.com/apache/datafusion-sqlparser-rs.git
synced 2025-07-07 17:04:59 +00:00
113 lines
3.8 KiB
Rust
113 lines
3.8 KiB
Rust
// Licensed to the Apache Software Foundation (ASF) under one
|
|
// or more contributor license agreements. See the NOTICE file
|
|
// distributed with this work for additional information
|
|
// regarding copyright ownership. The ASF licenses this file
|
|
// to you under the Apache License, Version 2.0 (the
|
|
// "License"); you may not use this file except in compliance
|
|
// with the License. You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing,
|
|
// software distributed under the License is distributed on an
|
|
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
// KIND, either express or implied. See the License for the
|
|
// specific language governing permissions and limitations
|
|
// under the License.
|
|
|
|
#![warn(clippy::all)]
|
|
|
|
//! A small command-line app to run the parser.
|
|
//! Run with `cargo run --example cli`
|
|
|
|
use std::fs;
|
|
use std::io::{stdin, Read};
|
|
|
|
use simple_logger::SimpleLogger;
|
|
use sqlparser::dialect::*;
|
|
use sqlparser::parser::Parser;
|
|
|
|
fn main() {
|
|
SimpleLogger::new().init().unwrap();
|
|
|
|
let filename = std::env::args().nth(1).expect(
|
|
r#"
|
|
No arguments provided!
|
|
|
|
Usage:
|
|
$ cargo run --example cli FILENAME.sql [--dialectname]
|
|
|
|
To print the parse results as JSON:
|
|
$ cargo run --feature json_example --example cli FILENAME.sql [--dialectname]
|
|
|
|
To read from stdin instead of a file:
|
|
$ cargo run --example cli - [--dialectname]
|
|
|
|
"#,
|
|
);
|
|
|
|
let dialect: Box<dyn Dialect> = match std::env::args().nth(2).unwrap_or_default().as_ref() {
|
|
"--ansi" => Box::new(AnsiDialect {}),
|
|
"--bigquery" => Box::new(BigQueryDialect {}),
|
|
"--postgres" => Box::new(PostgreSqlDialect {}),
|
|
"--ms" => Box::new(MsSqlDialect {}),
|
|
"--mysql" => Box::new(MySqlDialect {}),
|
|
"--snowflake" => Box::new(SnowflakeDialect {}),
|
|
"--hive" => Box::new(HiveDialect {}),
|
|
"--redshift" => Box::new(RedshiftSqlDialect {}),
|
|
"--clickhouse" => Box::new(ClickHouseDialect {}),
|
|
"--duckdb" => Box::new(DuckDbDialect {}),
|
|
"--sqlite" => Box::new(SQLiteDialect {}),
|
|
"--generic" | "" => Box::new(GenericDialect {}),
|
|
s => panic!("Unexpected parameter: {s}"),
|
|
};
|
|
|
|
let contents = if filename == "-" {
|
|
println!("Parsing from stdin using {dialect:?}");
|
|
let mut buf = Vec::new();
|
|
stdin()
|
|
.read_to_end(&mut buf)
|
|
.expect("failed to read from stdin");
|
|
String::from_utf8(buf).expect("stdin content wasn't valid utf8")
|
|
} else {
|
|
println!("Parsing from file '{}' using {:?}", &filename, dialect);
|
|
fs::read_to_string(&filename)
|
|
.unwrap_or_else(|_| panic!("Unable to read the file {}", &filename))
|
|
};
|
|
let without_bom = if contents.chars().next().unwrap() as u64 != 0xfeff {
|
|
contents.as_str()
|
|
} else {
|
|
let mut chars = contents.chars();
|
|
chars.next();
|
|
chars.as_str()
|
|
};
|
|
let parse_result = Parser::parse_sql(&*dialect, without_bom);
|
|
match parse_result {
|
|
Ok(statements) => {
|
|
println!(
|
|
"Round-trip:\n'{}'",
|
|
statements
|
|
.iter()
|
|
.map(std::string::ToString::to_string)
|
|
.collect::<Vec<_>>()
|
|
.join("\n")
|
|
);
|
|
|
|
if cfg!(feature = "json_example") {
|
|
#[cfg(feature = "json_example")]
|
|
{
|
|
let serialized = serde_json::to_string_pretty(&statements).unwrap();
|
|
println!("Serialized as JSON:\n{serialized}");
|
|
}
|
|
} else {
|
|
println!("Parse results:\n{statements:#?}");
|
|
}
|
|
|
|
std::process::exit(0);
|
|
}
|
|
Err(e) => {
|
|
println!("Error during parsing: {e:?}");
|
|
std::process::exit(1);
|
|
}
|
|
}
|
|
}
|