common_meta/reconciliation/reconcile_database/
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::reconciliation::reconcile_database::{ReconcileDatabaseContext, State};
23
24#[derive(Debug, Serialize, Deserialize)]
25pub(crate) struct ReconcileDatabaseEnd;
26
27#[async_trait::async_trait]
28#[typetag::serde]
29impl State for ReconcileDatabaseEnd {
30    async fn next(
31        &mut self,
32        ctx: &mut ReconcileDatabaseContext,
33        procedure_ctx: &ProcedureContext,
34    ) -> Result<(Box<dyn State>, Status)> {
35        info!(
36            "Database reconciliation completed. schema: {}, catalog: {}, procedure_id: {}, metrics: {}, elapsed: {:?}",
37            ctx.persistent_ctx.schema,
38            ctx.persistent_ctx.catalog,
39            procedure_ctx.procedure_id,
40            ctx.metrics(),
41            ctx.volatile_ctx.start_time.elapsed(),
42        );
43        Ok((Box::new(ReconcileDatabaseEnd), Status::done()))
44    }
45
46    fn as_any(&self) -> &dyn Any {
47        self
48    }
49}