Skip to main content

tests_fuzz/ir/
repartition_expr.rs

1// Copyright 2023 Greptime Team
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 partition::expr::PartitionExpr;
16use serde::{Deserialize, Serialize};
17
18use crate::ir::Ident;
19
20#[derive(Debug, Clone, Serialize, Deserialize)]
21pub struct SplitPartitionExpr {
22    pub table_name: Ident,
23    pub target: PartitionExpr,
24    pub into: Vec<PartitionExpr>,
25    #[serde(default = "default_wait")]
26    pub wait: bool,
27}
28
29#[derive(Debug, Clone, Serialize, Deserialize)]
30pub struct MergePartitionExpr {
31    pub table_name: Ident,
32    pub targets: Vec<PartitionExpr>,
33    #[serde(default = "default_wait")]
34    pub wait: bool,
35}
36
37#[derive(Debug, Clone, Serialize, Deserialize)]
38pub enum RepartitionExpr {
39    Split(SplitPartitionExpr),
40    Merge(MergePartitionExpr),
41}
42
43const fn default_wait() -> bool {
44    true
45}