Line data    Source code

       1              : use crate::query::RowStream;
       2              : use crate::types::Type;
       3              : use crate::{Client, Error, Transaction};
       4              : use async_trait::async_trait;
       5              : use postgres_protocol2::Oid;
       6              : 
       7              : mod private {
       8              :     pub trait Sealed {}
       9              : }
      10              : 
      11              : /// A trait allowing abstraction over connections and transactions.
      12              : ///
      13              : /// This trait is "sealed", and cannot be implemented outside of this crate.
      14              : #[async_trait]
      15              : pub trait GenericClient: private::Sealed {
      16              :     /// Like `Client::query_raw_txt`.
      17              :     async fn query_raw_txt<S, I>(&self, statement: &str, params: I) -> Result<RowStream, Error>
      18              :     where
      19              :         S: AsRef<str> + Sync + Send,
      20              :         I: IntoIterator<Item = Option<S>> + Sync + Send,
      21              :         I::IntoIter: ExactSizeIterator + Sync + Send;
      22              : 
      23              :     /// Query for type information
      24              :     async fn get_type(&self, oid: Oid) -> Result<Type, Error>;
      25              : }
      26              : 
      27              : impl private::Sealed for Client {}
      28              : 
      29              : #[async_trait]
      30              : impl GenericClient for Client {
      31            0 :     async fn query_raw_txt<S, I>(&self, statement: &str, params: I) -> Result<RowStream, Error>
      32            0 :     where
      33            0 :         S: AsRef<str> + Sync + Send,
      34            0 :         I: IntoIterator<Item = Option<S>> + Sync + Send,
      35            0 :         I::IntoIter: ExactSizeIterator + Sync + Send,
      36            0 :     {
      37            0 :         self.query_raw_txt(statement, params).await
      38            0 :     }
      39              : 
      40              :     /// Query for type information
      41            0 :     async fn get_type(&self, oid: Oid) -> Result<Type, Error> {
      42            0 :         self.get_type(oid).await
      43            0 :     }
      44              : }
      45              : 
      46              : impl private::Sealed for Transaction<'_> {}
      47              : 
      48              : #[async_trait]
      49              : #[allow(clippy::needless_lifetimes)]
      50              : impl GenericClient for Transaction<'_> {
      51            0 :     async fn query_raw_txt<S, I>(&self, statement: &str, params: I) -> Result<RowStream, Error>
      52            0 :     where
      53            0 :         S: AsRef<str> + Sync + Send,
      54            0 :         I: IntoIterator<Item = Option<S>> + Sync + Send,
      55            0 :         I::IntoIter: ExactSizeIterator + Sync + Send,
      56            0 :     {
      57            0 :         self.query_raw_txt(statement, params).await
      58            0 :     }
      59              : 
      60              :     /// Query for type information
      61            0 :     async fn get_type(&self, oid: Oid) -> Result<Type, Error> {
      62            0 :         self.client().get_type(oid).await
      63            0 :     }
      64              : }