mirrors/datafusion-sqlparse
1
0
Fork 0
mirror of https://github.com/apache/datafusion-sqlparser-rs.git synced 2025-08-04 06:18:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Introduce comma_separated_string() to reduce boilerplate in ToString impls

Browse source
This commit is contained in:
Nickolay Ponomarev 2019-04-20 19:54:12 +03:00
parent ef76baa7fd
commit c8e7c3b343
2 changed files with 17 additions and 56 deletions
Download patch file Download diff file Expand all files Collapse all files

49
src/sqlast/mod.rs
View file

@ -30,6 +30,14 @@ pub use self::value::Value;
pub use self::sql_operator::SQLOperator;
/// Like `vec.join(", ")`, but for any types implementing ToString.
fn comma_separated_string<T: ToString>(vec: &Vec<T>) -> String {
vec.iter()
.map(|a| a.to_string())
.collect::<Vec<String>>()
.join(", ")
}
/// Identifier name, in the originally quoted form (e.g. `"id"`)
pub type SQLIdent = String;
@ -123,10 +131,7 @@ impl ToString for ASTNode {
"{} {}IN ({})",
expr.as_ref().to_string(),
if *negated { "NOT " } else { "" },
list.iter()
.map(|a| a.to_string())
.collect::<Vec<String>>()
.join(", ")
comma_separated_string(list)
),
ASTNode::SQLInSubquery {
expr,
@ -279,11 +284,7 @@ impl ToString for SQLStatement {
" VALUES({})",
values
.iter()
.map(|row| row
.iter()
.map(|c| c.to_string())
.collect::<Vec<String>>()
.join(", "))
.map(|row| comma_separated_string(row))
.collect::<Vec<String>>()
.join(", ")
);
@ -297,14 +298,7 @@ impl ToString for SQLStatement {
} => {
let mut s = format!("COPY {}", table_name.to_string());
if columns.len() > 0 {
s += &format!(
" ({})",
columns
.iter()
.map(|c| c.to_string())
.collect::<Vec<String>>()
.join(", ")
);
s += &format!(" ({})", comma_separated_string(columns));
}
s += " FROM stdin; ";
if values.len() > 0 {
@ -327,14 +321,7 @@ impl ToString for SQLStatement {
} => {
let mut s = format!("UPDATE {}", table_name.to_string());
if assignments.len() > 0 {
s += &format!(
"{}",
assignments
.iter()
.map(|ass| ass.to_string())
.collect::<Vec<String>>()
.join(", ")
);
s += &format!("{}", comma_separated_string(assignments));
}
if let Some(selection) = selection {
s += &format!(" WHERE {}", selection.to_string());
@ -373,11 +360,7 @@ impl ToString for SQLStatement {
} if *external => format!(
"CREATE EXTERNAL TABLE {} ({}) STORED AS {} LOCATION '{}'",
name.to_string(),
columns
.iter()
.map(|c| c.to_string())
.collect::<Vec<String>>()
.join(", "),
comma_separated_string(columns),
file_format.as_ref().map(|f| f.to_string()).unwrap(),
location.as_ref().unwrap()
),
@ -390,11 +373,7 @@ impl ToString for SQLStatement {
} => format!(
"CREATE TABLE {} ({})",
name.to_string(),
columns
.iter()
.map(|c| c.to_string())
.collect::<Vec<String>>()
.join(", ")
comma_separated_string(columns)
),
SQLStatement::SQLAlterTable { name, operation } => {
format!("ALTER TABLE {} {}", name.to_string(), operation.to_string())

24
src/sqlast/query.rs
View file

@ -29,14 +29,7 @@ impl ToString for SQLQuery {
}
s += &self.body.to_string();
if let Some(ref order_by) = self.order_by {
s += &format!(
" ORDER BY {}",
order_by
.iter()
.map(|o| o.to_string())
.collect::<Vec<String>>()
.join(", ")
);
s += &format!(" ORDER BY {}", comma_separated_string(order_by));
}
if let Some(ref limit) = self.limit {
s += &format!(" LIMIT {}", limit.to_string());
@ -130,11 +123,7 @@ impl ToString for SQLSelect {
let mut s = format!(
"SELECT{} {}",
if self.distinct { " DISTINCT" } else { "" },
self.projection
.iter()
.map(|p| p.to_string())
.collect::<Vec<String>>()
.join(", ")
comma_separated_string(&self.projection)
);
if let Some(ref relation) = self.relation {
s += &format!(" FROM {}", relation.to_string());
@ -146,14 +135,7 @@ impl ToString for SQLSelect {
s += &format!(" WHERE {}", selection.to_string());
}
if let Some(ref group_by) = self.group_by {
s += &format!(
" GROUP BY {}",
group_by
.iter()
.map(|g| g.to_string())
.collect::<Vec<String>>()
.join(", ")
);
s += &format!(" GROUP BY {}", comma_separated_string(group_by));
}
if let Some(ref having) = self.having {
s += &format!(" HAVING {}", having.to_string());