otel-arrow-dfe-engine 0.61.0

Async pipeline engine
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101

// Copyright The OpenTelemetry Authors
// SPDX-License-Identifier: Apache-2.0

//! Pipeline-scoped topic binding with resolved wiring defaults.

use std::ops::Deref;

use crate::topic::handle::TopicHandle;
use otel_arrow_dfe_config::topic::{TopicAckPropagationMode, TopicQueueOnFullPolicy};

/// Pipeline-scoped topic binding with resolved wiring defaults.
///
/// The binding wraps a pure [`TopicHandle`] with defaults that are resolved by
/// the controller for one pipeline instance, such as how topic exporters should
/// behave when balanced queues are full and whether topic receivers/exporters
/// should bridge Ack/Nack across the topic hop by default.
pub struct PipelineTopicBinding<T: Send + Sync + 'static> {
    handle: TopicHandle<T>,
    queue_on_full_default: TopicQueueOnFullPolicy,
    ack_propagation_mode_default: TopicAckPropagationMode,
}

impl<T: Send + Sync + 'static> Clone for PipelineTopicBinding<T> {
    fn clone(&self) -> Self {
        Self {
            handle: self.handle.clone(),
            queue_on_full_default: self.queue_on_full_default.clone(),
            ack_propagation_mode_default: self.ack_propagation_mode_default,
        }
    }
}

impl<T: Send + Sync + 'static> PipelineTopicBinding<T> {
    /// Create a binding with the default pipeline wiring policies.
    #[must_use]
    pub fn new(handle: TopicHandle<T>) -> Self {
        Self {
            handle,
            queue_on_full_default: TopicQueueOnFullPolicy::Block,
            ack_propagation_mode_default: TopicAckPropagationMode::Disabled,
        }
    }

    /// Return a cloned binding with the topic-level default full-queue policy.
    #[must_use]
    pub fn with_default_queue_on_full(&self, policy: TopicQueueOnFullPolicy) -> Self {
        Self {
            handle: self.handle.clone(),
            queue_on_full_default: policy,
            ack_propagation_mode_default: self.ack_propagation_mode_default,
        }
    }

    /// Return a cloned binding with the topic-level default Ack/Nack propagation mode.
    #[must_use]
    pub fn with_default_ack_propagation_mode(&self, mode: TopicAckPropagationMode) -> Self {
        Self {
            handle: self.handle.clone(),
            queue_on_full_default: self.queue_on_full_default.clone(),
            ack_propagation_mode_default: mode,
        }
    }

    /// The underlying pure topic runtime handle.
    #[must_use]
    pub const fn handle(&self) -> &TopicHandle<T> {
        &self.handle
    }

    /// Consume the binding and return the underlying topic handle.
    #[must_use]
    pub fn into_handle(self) -> TopicHandle<T> {
        self.handle
    }

    /// Pipeline-level default behavior when the queue is full.
    #[must_use]
    pub fn default_queue_on_full(&self) -> TopicQueueOnFullPolicy {
        self.queue_on_full_default.clone()
    }

    /// Pipeline-level default mode for cross-pipeline Ack/Nack propagation.
    #[must_use]
    pub const fn default_ack_propagation_mode(&self) -> TopicAckPropagationMode {
        self.ack_propagation_mode_default
    }
}

impl<T: Send + Sync + 'static> From<TopicHandle<T>> for PipelineTopicBinding<T> {
    fn from(value: TopicHandle<T>) -> Self {
        Self::new(value)
    }
}

impl<T: Send + Sync + 'static> Deref for PipelineTopicBinding<T> {
    type Target = TopicHandle<T>;

    fn deref(&self) -> &Self::Target {
        &self.handle
    }
}