add function

Author: adriangb

src/channel_resolver_ext.rs

@@ -23,6 +23,11 @@ pub(crate) fn get_distributed_channel_resolver(
         .ok_or_else(|| exec_datafusion_err!("ChannelResolver not present in the session config"))
 }
 
+/// Check if there is a distributed channel resolver registered with this [`SessionConfig`].
+pub fn has_distributed_channel_resolver(cfg: &SessionConfig) -> bool {
+    cfg.get_extension::<ChannelResolverExtension>().is_some()
+}
+
 #[derive(Clone)]
 struct ChannelResolverExtension(Arc<dyn ChannelResolver + Send + Sync>);
 

src/lib.rs

@@ -14,7 +14,7 @@ mod protobuf;
 #[cfg(any(feature = "integration", test))]
 pub mod test_utils;
 
-pub use channel_resolver_ext::{BoxCloneSyncChannel, ChannelResolver};
+pub use channel_resolver_ext::{has_distributed_channel_resolver, BoxCloneSyncChannel, ChannelResolver};
 pub use distributed_ext::DistributedExt;
 pub use distributed_physical_optimizer_rule::DistributedPhysicalOptimizerRule;
 pub use execution_plans::{