Make hooks module private and relevant types

bikeshedder · Oct 18, 2021 · f35ac2a · f35ac2a
1 parent 2c6f899
commit f35ac2a
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 7 deletions.
diff --git a/src/managed/hooks.rs b/src/managed/hooks.rs
@@ -176,7 +176,7 @@ impl<M: Manager> HookVec<M> {
 /// Collection of all the hooks that can be configured for a [`Pool`].
 ///
 /// [`Pool`]: super::Pool
-pub struct Hooks<M: Manager> {
+pub(crate) struct Hooks<M: Manager> {
     pub(crate) post_create: HookVec<M>,
     pub(crate) pre_recycle: HookVec<M>,
     pub(crate) post_recycle: HookVec<M>,

diff --git a/src/managed/mod.rs b/src/managed/mod.rs
@@ -54,7 +54,7 @@ mod builder;
 mod config;
 mod dropguard;
 mod errors;
-pub mod hooks;
+mod hooks;
 mod metrics;
 pub mod reexports;
 pub mod sync;
@@ -77,6 +77,7 @@ use tokio::sync::{Semaphore, TryAcquireError};
 
 use crate::runtime::Runtime;
 
+pub use hooks::{Hook, HookError, HookErrorCause};
 pub use metrics::Metrics;
 
 pub use crate::Status;

diff --git a/src/managed/reexports.rs b/src/managed/reexports.rs
@@ -8,6 +8,10 @@
 //! ```
 
 pub use crate::{
-    managed::{PoolConfig, Timeouts},
+    managed::{
+        hooks::{Hook, HookError, HookErrorCause},
+        metrics::Metrics,
+        PoolConfig, Timeouts,
+    },
     Runtime,
 };
diff --git a/tests/managed_hooks.rs b/tests/managed_hooks.rs
@@ -4,10 +4,7 @@ use std::sync::atomic::{AtomicUsize, Ordering};
 
 use async_trait::async_trait;
 
-use deadpool::managed::{
-    hooks::{Hook, HookError, HookErrorCause},
-    Manager, Pool, PoolError, RecycleResult,
-};
+use deadpool::managed::{Hook, HookError, HookErrorCause, Manager, Pool, PoolError, RecycleResult};
 
 struct Computer {
     next_id: AtomicUsize,