rust/greptimedb
1
0
Fork 0
mirror of https://github.com/GreptimeTeam/greptimedb.git synced 2026-05-27 02:10:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add logical plan based prepare statement for postgresql (#1813)

Browse Source 
* feat: add logical plan based prepare statement for postgresql

* refactor: correct more types

* Update src/servers/src/postgres/types.rs

Co-authored-by: LFC <bayinamine@gmail.com>

* fix: address review issues

* test: add datetime in integration tests

---------

Co-authored-by: LFC <bayinamine@gmail.com>
This commit is contained in:
Ning Sun
2023-07-11 11:07:18 +08:00
committed by GitHub
parent c615fb2a93
commit f293126315
11 changed files with 865 additions and 506 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
src/sql/src/statements/query.rs
View File

@@ -14,7 +14,6 @@
use std::fmt;
use datatypes::prelude::ConcreteDataType;
use sqlparser::ast::Query as SpQuery;
use crate::error::Error;
@@ -23,7 +22,6 @@ use crate::error::Error;
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct Query {
pub inner: SpQuery,
pub param_types: Vec<ConcreteDataType>,
}
/// Automatically converts from sqlparser Query instance to SqlQuery.
@@ -31,10 +29,7 @@ impl TryFrom<SpQuery> for Query {
type Error = Error;
fn try_from(q: SpQuery) -> Result<Self, Self::Error> {
Ok(Query {
inner: q,
param_types: vec![],
})
Ok(Query { inner: q })
}
}
@@ -46,27 +41,9 @@ impl TryFrom<Query> for SpQuery {
}
}
impl Query {
pub fn param_types(&self) -> &Vec<ConcreteDataType> {
&self.param_types
}
pub fn param_types_mut(&mut self) -> &mut Vec<ConcreteDataType> {
&mut self.param_types
}
}
impl fmt::Display for Query {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "{} ", self.inner)?;
write!(f, "[")?;
for i in 0..self.param_types.len() {
write!(f, "{}", self.param_types[i])?;
if i != self.param_types.len() - 1 {
write!(f, ",")?;
}
}
write!(f, "]")?;
write!(f, "{}", self.inner)?;
Ok(())
}
}
@@ -95,7 +72,7 @@ mod test {
create_query("select * from abc where x = 1 and y = 7")
.unwrap()
.to_string(),
"SELECT * FROM abc WHERE x = 1 AND y = 7 []"
"SELECT * FROM abc WHERE x = 1 AND y = 7"
);
assert_eq!(
create_query(
@@ -103,7 +80,7 @@ mod test {
)
.unwrap()
.to_string(),
"SELECT * FROM abc LEFT JOIN bcd WHERE abc.a = 1 AND bcd.d = 7 AND abc.id = bcd.id []"
"SELECT * FROM abc LEFT JOIN bcd WHERE abc.a = 1 AND bcd.d = 7 AND abc.id = bcd.id"
);
}
}