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 {}
@@ -79,6 +80,9 @@ pub trait GenericClient: private::Sealed {
     /// Like `Client::batch_execute`.
     async fn batch_execute(&self, query: &str) -> Result<(), 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;
 }
@@ -173,6 +177,11 @@ impl GenericClient for Client {
         Ok(())
     }
 
+    /// Query for type information
+    async fn get_type(&self, oid: Oid) -> Result<Type, Error> {
+        self.get_type(oid).await
+    }
+
     fn client(&self) -> &Client {
         self
     }
@@ -270,6 +279,11 @@ impl GenericClient for Transaction<'_> {
         Ok(())
     }
 
+    /// 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()
     }