meta_srv/procedure/repartition/
repartition_end.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 std::any::Any;
16
17use common_procedure::{Context as ProcedureContext, Status};
18use common_telemetry::info;
19use serde::{Deserialize, Serialize};
20
21use crate::error::Result;
22use crate::procedure::repartition::{Context, State};
23
24#[derive(Debug, Clone, Serialize, Deserialize)]
25pub struct RepartitionEnd;
26
27#[async_trait::async_trait]
28#[typetag::serde]
29impl State for RepartitionEnd {
30    async fn next(
31        &mut self,
32        ctx: &mut Context,
33        _procedure_ctx: &ProcedureContext,
34    ) -> Result<(Box<dyn State>, Status)> {
35        info!(
36            "Repartition procedure finished, metrics: {}",
37            ctx.volatile_ctx.metrics
38        );
39        Ok((Box::new(RepartitionEnd), Status::done()))
40    }
41
42    fn as_any(&self) -> &dyn Any {
43        self
44    }
45}