added ensure_initialized

hstack · Feb 4, 2025 · 4d9c113 · 4d9c113
1 parent 927f7fb
commit 4d9c113
Show file tree

Hide file tree

Showing 5 changed files with 15 additions and 4 deletions.
diff --git a/crates/aws/src/lib.rs b/crates/aws/src/lib.rs
@@ -99,7 +99,6 @@ pub fn register_handlers(_additional_prefixes: Option<Url>) {
 #[ctor::ctor]
 fn init() {
     register_handlers(None);
-    tracing::info!("AWS initialized");
 }
 
 /// Representation of a log entry stored in DynamoDb

diff --git a/crates/azure/src/lib.rs b/crates/azure/src/lib.rs
@@ -73,5 +73,4 @@ pub fn register_handlers(_additional_prefixes: Option<Url>) {
 #[ctor::ctor]
 fn init() {
     register_handlers(None);
-    tracing::info!("Azure initialized");
 }
diff --git a/crates/deltalake/src/lib.rs b/crates/deltalake/src/lib.rs
@@ -1,3 +1,4 @@
+use std::sync::Once;
 /**
  * The deltalake crate is currently just a meta-package shim for deltalake-core
  */
@@ -11,3 +12,17 @@ pub use deltalake_azure as azure;
 pub use deltalake_gcp as gcp;
 #[cfg(feature = "hdfs")]
 pub use deltalake_hdfs as hdfs;
+
+pub fn ensure_initialized() {
+    let once = Once::new();
+    once.call_once(|| {
+        #[cfg(any(feature = "s3", feature = "s3-native-tls"))]
+        aws::register_handlers(None);
+        #[cfg(feature = "azure")]
+        azure::register_handlers(None);
+        #[cfg(feature = "gcs")]
+        gcp::register_handlers(None);
+        #[cfg(feature = "hdfs")]
+        hdfs::register_handlers(None);
+    })
+}
diff --git a/crates/gcp/src/lib.rs b/crates/gcp/src/lib.rs
@@ -74,5 +74,4 @@ pub fn register_handlers(_additional_prefixes: Option<Url>) {
 #[ctor::ctor]
 fn init() {
     register_handlers(None);
-    tracing::info!("GCP initialized");
 }
diff --git a/crates/hdfs/src/lib.rs b/crates/hdfs/src/lib.rs
@@ -50,5 +50,4 @@ pub fn register_handlers(_additional_prefixes: Option<Url>) {
 #[ctor::ctor]
 fn init() {
     register_handlers(None);
-    tracing::info!("HDFS initialized");
 }