risingwave_meta_model/
actor_dispatcher.rs

1// Copyright 2025 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 std::hash::Hash;
16
17use risingwave_pb::stream_plan::PbDispatcherType;
18use sea_orm::entity::prelude::*;
19use serde::{Deserialize, Serialize};
20
21#[derive(
22    Hash, Copy, Clone, Debug, PartialEq, Eq, EnumIter, DeriveActiveEnum, Serialize, Deserialize,
23)]
24#[sea_orm(rs_type = "String", db_type = "string(None)")]
25pub enum DispatcherType {
26    #[sea_orm(string_value = "HASH")]
27    Hash,
28    #[sea_orm(string_value = "BROADCAST")]
29    Broadcast,
30    #[sea_orm(string_value = "SIMPLE")]
31    Simple,
32    #[sea_orm(string_value = "NO_SHUFFLE")]
33    NoShuffle,
34}
35
36impl From<PbDispatcherType> for DispatcherType {
37    fn from(val: PbDispatcherType) -> Self {
38        match val {
39            PbDispatcherType::Unspecified => unreachable!(),
40            PbDispatcherType::Hash => DispatcherType::Hash,
41            PbDispatcherType::Broadcast => DispatcherType::Broadcast,
42            PbDispatcherType::Simple => DispatcherType::Simple,
43            PbDispatcherType::NoShuffle => DispatcherType::NoShuffle,
44        }
45    }
46}
47
48impl From<DispatcherType> for PbDispatcherType {
49    fn from(val: DispatcherType) -> Self {
50        match val {
51            DispatcherType::Hash => PbDispatcherType::Hash,
52            DispatcherType::Broadcast => PbDispatcherType::Broadcast,
53            DispatcherType::Simple => PbDispatcherType::Simple,
54            DispatcherType::NoShuffle => PbDispatcherType::NoShuffle,
55        }
56    }
57}
risingwave_meta_model/actor_dispatcher.rs

risingwave_meta_model/
actor_dispatcher.rs
