risingwave_stream/from_proto/
barrier_recv.rs

1// Copyright 2023 RisingWave Labs
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7//     http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15use risingwave_pb::stream_plan::BarrierRecvNode;
16
17use super::*;
18use crate::executor::BarrierRecvExecutor;
19
20pub struct BarrierRecvExecutorBuilder;
21
22impl_stream_node_body!(BarrierRecv(BarrierRecvNode) => BarrierRecvExecutorBuilder);
23
24impl ExecutorBuilder for BarrierRecvExecutorBuilder {
25    type Node = BarrierRecvNode;
26
27    async fn new_boxed_executor(
28        params: ExecutorParams,
29        _node: &Self::Node,
30        _store: impl StateStore,
31    ) -> StreamResult<Executor> {
32        assert!(
33            params.input.is_empty(),
34            "barrier receiver should not have input"
35        );
36
37        let barrier_receiver = params
38            .local_barrier_manager
39            .subscribe_barrier(params.actor_context.id);
40
41        let exec = BarrierRecvExecutor::new(params.actor_context, barrier_receiver);
42        Ok((params.info, exec).into())
43    }
44}
risingwave_stream/from_proto/barrier_recv.rs

risingwave_stream/from_proto/
barrier_recv.rs
