use std::future::Future;
use tokio_postgres::{
Client, Error, Row, RowStream, Statement, ToStatement, Transaction,
types::{BorrowToSql, ToSql},
};
pub trait GenericClient: Send + Sync {
fn stmt_cache() -> bool {
false
}
fn prepare(&self, query: &str) -> impl Future<Output = Result<Statement, Error>> + Send;
fn execute<T>(
&self,
query: &T,
params: &[&(dyn ToSql + Sync)],
) -> impl Future<Output = Result<u64, Error>> + Send
where
T: ?Sized + ToStatement + Sync + Send;
fn query_one<T>(
&self,
statement: &T,
params: &[&(dyn ToSql + Sync)],
) -> impl Future<Output = Result<Row, Error>> + Send
where
T: ?Sized + ToStatement + Sync + Send;
fn query_opt<T>(
&self,
statement: &T,
params: &[&(dyn ToSql + Sync)],
) -> impl Future<Output = Result<Option<Row>, Error>> + Send
where
T: ?Sized + ToStatement + Sync + Send;
fn query<T>(
&self,
query: &T,
params: &[&(dyn ToSql + Sync)],
) -> impl Future<Output = Result<Vec<Row>, Error>> + Send
where
T: ?Sized + ToStatement + Sync + Send;
fn query_raw<T, I>(
&self,
statement: &T,
params: I,
) -> impl Future<Output = Result<RowStream, Error>> + Send
where
T: ?Sized + ToStatement + Sync + Send,
I: IntoIterator + Sync + Send,
I::IntoIter: ExactSizeIterator,
I::Item: BorrowToSql;
}
impl GenericClient for Transaction<'_> {
async fn prepare(&self, query: &str) -> Result<Statement, Error> {
Transaction::prepare(self, query).await
}
async fn execute<T>(&self, query: &T, params: &[&(dyn ToSql + Sync)]) -> Result<u64, Error>
where
T: ?Sized + ToStatement + Sync + Send,
{
Transaction::execute(self, query, params).await
}
async fn query_one<T>(
&self,
statement: &T,
params: &[&(dyn ToSql + Sync)],
) -> Result<Row, Error>
where
T: ?Sized + ToStatement + Sync + Send,
{
Transaction::query_one(self, statement, params).await
}
async fn query_opt<T>(
&self,
statement: &T,
params: &[&(dyn ToSql + Sync)],
) -> Result<Option<Row>, Error>
where
T: ?Sized + ToStatement + Sync + Send,
{
Transaction::query_opt(self, statement, params).await
}
async fn query<T>(&self, query: &T, params: &[&(dyn ToSql + Sync)]) -> Result<Vec<Row>, Error>
where
T: ?Sized + ToStatement + Sync + Send,
{
Transaction::query(self, query, params).await
}
async fn query_raw<T, I>(&self, statement: &T, params: I) -> Result<RowStream, Error>
where
T: ?Sized + ToStatement + Sync + Send,
I: IntoIterator + Sync + Send,
I::IntoIter: ExactSizeIterator,
I::Item: BorrowToSql,
{
Transaction::query_raw(self, statement, params).await
}
}
impl GenericClient for Client {
async fn prepare(&self, query: &str) -> Result<Statement, Error> {
Client::prepare(self, query).await
}
async fn execute<T>(&self, query: &T, params: &[&(dyn ToSql + Sync)]) -> Result<u64, Error>
where
T: ?Sized + ToStatement + Sync + Send,
{
Client::execute(self, query, params).await
}
async fn query_one<T>(
&self,
statement: &T,
params: &[&(dyn ToSql + Sync)],
) -> Result<Row, Error>
where
T: ?Sized + ToStatement + Sync + Send,
{
Client::query_one(self, statement, params).await
}
async fn query_opt<T>(
&self,
statement: &T,
params: &[&(dyn ToSql + Sync)],
) -> Result<Option<Row>, Error>
where
T: ?Sized + ToStatement + Sync + Send,
{
Client::query_opt(self, statement, params).await
}
async fn query<T>(&self, query: &T, params: &[&(dyn ToSql + Sync)]) -> Result<Vec<Row>, Error>
where
T: ?Sized + ToStatement + Sync + Send,
{
Client::query(self, query, params).await
}
async fn query_raw<T, I>(&self, statement: &T, params: I) -> Result<RowStream, Error>
where
T: ?Sized + ToStatement + Sync + Send,
I: IntoIterator + Sync + Send,
I::IntoIter: ExactSizeIterator,
I::Item: BorrowToSql,
{
Client::query_raw(self, statement, params).await
}
}