/
main.rs
105 lines (90 loc) · 2.88 KB
/
main.rs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
use clap::{crate_version, App, Arg};
use rustyline::Editor;
use rdbc::{Connection, DataType, Result};
use rdbc_mysql::MySQLDriver;
use rdbc_postgres::PostgresDriver;
use rdbc_sqlite::SqliteDriver;
fn main() -> Result<()> {
let matches = App::new("rdbc-cli")
.version(crate_version!())
.about("Rust DataBase Connectivity CLI")
.arg(
Arg::with_name("driver")
.help("RDBC driver name")
.short("d")
.long("driver")
.takes_value(true),
)
.arg(
Arg::with_name("connection-url")
.help("The database connection URL")
.short("c")
.long("connection-url")
.takes_value(true),
)
.get_matches();
let driver = matches.value_of("driver").unwrap();
let url = matches.value_of("connection-url").unwrap();
println!("Connecting to {} driver with url: {}", driver, url);
let driver: Box<dyn rdbc::Driver> = match driver {
"mysql" => Box::new(MySQLDriver::new()),
"postgres" => Box::new(PostgresDriver::new()),
"sqlite" => Box::new(SqliteDriver::new()),
_ => panic!("Invalid driver"),
};
let mut conn = driver.connect(url).unwrap();
let mut rl = Editor::<()>::new();
rl.load_history(".history").ok();
let mut query = "".to_owned();
loop {
let readline = rl.readline("> ");
match readline {
Ok(ref line) if line.trim_end().ends_with(';') => {
query.push_str(line.trim_end());
rl.add_history_entry(query.clone());
match execute(&mut *conn, &query) {
Ok(_) => {}
Err(e) => println!("Error: {:?}", e),
}
query = "".to_owned();
}
Ok(ref line) => {
query.push_str(line);
query.push_str(" ");
}
Err(_) => {
break;
}
}
}
rl.save_history(".history").ok();
Ok(())
}
fn execute(conn: &mut dyn Connection, sql: &str) -> Result<()> {
println!("Executing {}", sql);
let mut stmt = conn.create(sql)?;
let mut rs = stmt.execute_query(&vec![])?;
let meta = rs.meta_data()?;
for i in 0..meta.num_columns() {
if i > 0 {
print!("\t");
}
print!("{}", meta.column_name(i));
}
println!();
while rs.next() {
for i in 0..meta.num_columns() {
if i > 0 {
print!("\t");
}
match meta.column_type(i) {
DataType::Utf8 => print!("{:?}", rs.get_string(i)),
DataType::Integer => print!("{:?}", rs.get_i32(i)),
// TODO other types
_ => print!("{:?}", rs.get_string(i)),
}
}
println!();
}
Ok(())
}