get type generic client

Author: conradludgate

tokio-postgres/src/generic_client.rs

@@ -2,6 +2,7 @@ use crate::query::RowStream;
 use crate::types::{BorrowToSql, ToSql, Type};
 use crate::{Client, Error, Row, Statement, ToStatement, Transaction};
 use async_trait::async_trait;
+use postgres_protocol::Oid;
 
 mod private {
     pub trait Sealed {}
@@ -76,6 +77,9 @@ pub trait GenericClient: private::Sealed {
     /// Like `Client::transaction`.
     async fn transaction(&mut self) -> Result<Transaction<'_>, Error>;
 
+    /// Query for type information
+    async fn get_type(&self, oid: Oid) -> Result<Type, Error>;
+
     /// Returns a reference to the underlying `Client`.
     fn client(&self) -> &Client;
 }
@@ -165,6 +169,11 @@ impl GenericClient for Client {
         self.transaction().await
     }
 
+    /// Query for type information
+    async fn get_type(&self, oid: Oid) -> Result<Type, Error> {
+        self.get_type(oid).await
+    }
+
     fn client(&self) -> &Client {
         self
     }
@@ -257,6 +266,11 @@ impl GenericClient for Transaction<'_> {
         self.transaction().await
     }
 
+    /// Query for type information
+    async fn get_type(&self, oid: Oid) -> Result<Type, Error> {
+        self.client().get_type(oid).await
+    }
+
     fn client(&self) -> &Client {
         self.client()
     }