phillipleblanc opened a new pull request, #1400:
URL: https://github.com/apache/datafusion-ballista/pull/1400
# Which issue does this PR close?
Closes #1367
# Rationale for this change
Adds extension points to customize Ballista gRPC client connections for TLS
and custom header support.
# What changes are included in this PR?
- Add with_ballista_grpc_metadata() to inject custom headers (e.g., API
keys) into all gRPC requests
- Add with_ballista_override_create_grpc_client_endpoint() callback for
custom endpoint configuration (TLS, mTLS)
- Add with_ballista_use_tls() flag for cluster-wide TLS setting
- Ensure GrpcClientConfig (timeouts, keepalives) is applied to all client
connections
Example:
```rust
use std::sync::Arc;
use tonic::transport::{Certificate, ClientTlsConfig, Endpoint};
let ca_cert = std::fs::read("ca.pem")?;
let tls_config = ClientTlsConfig::new()
.ca_certificate(Certificate::from_pem(&ca_cert))
.domain_name("ballista.example.com");
let ctx = SessionContext::remote("df://scheduler:50050")
.await?
.with_ballista_override_create_grpc_client_endpoint(Arc::new(move |ep:
Endpoint| {
Ok(ep.tls_config(tls_config.clone())?)
}))
.with_ballista_use_tls(true);
```
# Are there any user-facing changes?
No, all changes are additive.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
