Function should_sample_statement

src/adapter/src/statement_logging.rs:744–757

Helper function to decide whether to sample a statement execution. Returns `true` if the statement should be sampled based on the sample rate. If `reproducible_rng` is `Some`, uses the provided RNG for reproducible sampling (used in tests). If `reproducible_rng` is `None`, uses the thread-local RNG.

(
    sample_rate: f64,
    reproducible_rng: Option<&mut rand_chacha::ChaCha8Rng>,
)

/// If `reproducible_rng` is `Some`, uses the provided RNG for reproducible sampling (used in tests).
/// If `reproducible_rng` is `None`, uses the thread-local RNG.
pub(crate) fn should_sample_statement(
    sample_rate: f64,
    reproducible_rng: Option<&mut rand_chacha::ChaCha8Rng>,
) -> bool {
    let distribution = Bernoulli::new(sample_rate).unwrap_or_else(|_| {
        soft_panic_or_log!("statement_logging_sample_rate is out of range [0, 1]");
        Bernoulli::new(0.0).expect("0.0 is valid for Bernoulli")
    });
    if let Some(rng) = reproducible_rng {
        distribution.sample(rng)
    } else {
        distribution.sample(&mut rand::rng())
    }
}

/// Helper function to serialize statement parameters for logging.
fn serialize_params(params: &Params) -> Vec<Option<String>> {