guillem/dr
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases 23 Wiki Activity

Schemas can be generated reading csv from stdin
All checks were successful
ci/woodpecker/push/woodpecker Pipeline was successful
Details

Browse source
This commit is contained in:
Guillem Borrell 2023-01-17 18:07:44 +00:00
parent 99d58ff9c3
commit d534bdef8d
1 changed files with 17 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
src/main.rs
View file

@ -37,6 +37,12 @@ fn main() {
.subcommand( .subcommand(
Command::new("schema") Command::new("schema")
.about("Several table schema related utilities") .about("Several table schema related utilities")
.arg(
arg!(-i --stdin ... "Read from stdin")
.required(false)
.action(ArgAction::SetTrue),
)
.arg(arg!(-d --delimiter <String> "Column delimiter. Assume ,").required(false))
.arg(arg!(-n --name <String> "Table name").required(false)) .arg(arg!(-n --name <String> "Table name").required(false))
.arg(arg!(-l --strlen <String> "Default length for string columns").required(false)) .arg(arg!(-l --strlen <String> "Default length for string columns").required(false))
.arg( .arg(
@ -219,8 +225,18 @@ fn main() {
eprintln!("Could now write to parquet"); eprintln!("Could now write to parquet");
} }
} else if let Some(_matches) = matches.subcommand_matches("schema") { } else if let Some(_matches) = matches.subcommand_matches("schema") {
let delimiter = match _matches.get_one::<String>("delimiter") {
Some(delimiter) => delimiter.as_bytes()[0],
None => b',',
};
let ldf = if _matches.get_flag("stdin") {
io::load_csv_from_stdin(delimiter)
} else {
io::read_ipc()
};
if _matches.get_flag("summary") { if _matches.get_flag("summary") {
schema::print_schema(io::read_ipc()); schema::print_schema(ldf);
} else if _matches.get_flag("postgresql") { } else if _matches.get_flag("postgresql") {
let name = _matches let name = _matches
.get_one::<String>("name") .get_one::<String>("name")