Add clear_type_cache to blocking client

Author: sfackler

postgres/src/client.rs

@@ -501,8 +501,8 @@ impl Client {
         Notifications::new(self.connection.as_ref())
     }
 
-    /// Constructs a cancellation token that can later be used to request
-    /// cancellation of a query running on this connection.
+    /// Constructs a cancellation token that can later be used to request cancellation of a query running on this
+    /// connection.
     ///
     /// # Examples
     ///
@@ -541,6 +541,15 @@ impl Client {
         CancelToken::new(self.client.cancel_token())
     }
 
+    /// Clears the client's type information cache.
+    ///
+    /// When user-defined types are used in a query, the client loads their definitions from the database and caches
+    /// them for the lifetime of the client. If those definitions are changed in the database, this method can be used
+    /// to flush the local cache and allow the new, updated definitions to be loaded.
+    pub fn clear_type_cache(&self) {
+        self.client.clear_type_cache();
+    }
+
     /// Determines if the client's connection has already closed.
     ///
     /// If this returns `true`, the client is no longer usable.

tokio-postgres/src/client.rs

@@ -113,7 +113,7 @@ impl InnerClient {
         self.state.lock().types.insert(oid, type_.clone());
     }
 
-    pub fn clear_types(&self) {
+    pub fn clear_type_cache(&self) {
         self.state.lock().types.clear();
     }
 
@@ -180,11 +180,6 @@ impl Client {
         &self.inner
     }
 
-    /// Clears the cache of database types (domain, enum, composition) that are loaded when preparing a query.
-    pub fn clear_types_cache(&self) {
-        self.inner().clear_types()
-    }
-
     #[cfg(feature = "runtime")]
     pub(crate) fn set_socket_config(&mut self, socket_config: SocketConfig) {
         self.socket_config = Some(socket_config);
@@ -493,9 +488,8 @@ impl Client {
         TransactionBuilder::new(self)
     }
 
-    /// Constructs a cancellation token that can later be used to request
-    /// cancellation of a query running on the connection associated with
-    /// this client.
+    /// Constructs a cancellation token that can later be used to request cancellation of a query running on the
+    /// connection associated with this client.
     pub fn cancel_token(&self) -> CancelToken {
         CancelToken {
             #[cfg(feature = "runtime")]
@@ -532,6 +526,15 @@ impl Client {
         self.cancel_token().cancel_query_raw(stream, tls).await
     }
 
+    /// Clears the client's type information cache.
+    ///
+    /// When user-defined types are used in a query, the client loads their definitions from the database and caches
+    /// them for the lifetime of the client. If those definitions are changed in the database, this method can be used
+    /// to flush the local cache and allow the new, updated definitions to be loaded.
+    pub fn clear_type_cache(&self) {
+        self.inner().clear_type_cache();
+    }
+
     /// Determines if the connection to the server has already closed.
     ///
     /// In that case, all future queries will fail.