mirror of
https://github.com/apache/datafusion-sqlparser-rs.git
synced 2025-07-08 01:15:00 +00:00
ac555d7e86
The rationale here is the same as the last commit: since this crate exclusively parses SQL, there's no need to restate that in every type name. (The prefix seems to be an artifact of this crate's history as a submodule of Datafusion, where it was useful to explicitly call out which types were related to SQL parsing.) This commit has the additional benefit of making all type names consistent; over type we'd added some types which were not prefixed with "SQL".
69 lines
2.3 KiB
Rust
69 lines
2.3 KiB
Rust
// Licensed 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)]
|
|
|
|
use simple_logger;
|
|
|
|
///! A small command-line app to run the parser.
|
|
/// Run with `cargo run --example cli`
|
|
use std::fs;
|
|
|
|
use sqlparser::dialect::*;
|
|
use sqlparser::parser::Parser;
|
|
|
|
fn main() {
|
|
simple_logger::init().unwrap();
|
|
|
|
let filename = std::env::args().nth(1).expect(
|
|
"No arguments provided!\n\n\
|
|
Usage: cargo run --example cli FILENAME.sql [--dialectname]",
|
|
);
|
|
|
|
let dialect: Box<dyn Dialect> = match std::env::args().nth(2).unwrap_or_default().as_ref() {
|
|
"--ansi" => Box::new(AnsiDialect {}),
|
|
"--postgres" => Box::new(PostgreSqlDialect {}),
|
|
"--ms" => Box::new(MsSqlDialect {}),
|
|
"--generic" | "" => Box::new(GenericDialect {}),
|
|
s => panic!("Unexpected parameter: {}", s),
|
|
};
|
|
|
|
println!("Parsing from file '{}' using {:?}", &filename, dialect);
|
|
let contents = fs::read_to_string(&filename)
|
|
.unwrap_or_else(|_| panic!("Unable to read the file {}", &filename));
|
|
let without_bom = if contents.chars().nth(0).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.to_owned());
|
|
match parse_result {
|
|
Ok(statements) => {
|
|
println!(
|
|
"Round-trip:\n'{}'",
|
|
statements
|
|
.iter()
|
|
.map(std::string::ToString::to_string)
|
|
.collect::<Vec<_>>()
|
|
.join("\n")
|
|
);
|
|
println!("Parse results:\n{:#?}", statements);
|
|
std::process::exit(0);
|
|
}
|
|
Err(e) => {
|
|
println!("Error during parsing: {:?}", e);
|
|
std::process::exit(1);
|
|
}
|
|
}
|
|
}
|