feat: expose RegionMigrationProcedureTask

GreptimeTeam · Jan 3, 2024 · 5f7cfc7 · 5f7cfc7
1 parent 2c17a94
commit 5f7cfc7
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/src/meta-srv/src/procedure/region_migration.rs b/src/meta-srv/src/procedure/region_migration.rs
@@ -42,6 +42,7 @@ use common_procedure::error::{
     Error as ProcedureError, FromJsonSnafu, Result as ProcedureResult, ToJsonSnafu,
 };
 use common_procedure::{Context as ProcedureContext, LockKey, Procedure, Status};
+pub use manager::RegionMigrationProcedureTask;
 use serde::{Deserialize, Serialize};
 use snafu::{location, Location, OptionExt, ResultExt};
 use store_api::storage::RegionId;

diff --git a/src/meta-srv/src/procedure/region_migration/manager.rs b/src/meta-srv/src/procedure/region_migration/manager.rs
@@ -56,13 +56,24 @@ impl Drop for RegionMigrationProcedureGuard {
 }
 
 #[derive(Debug, Clone)]
-pub(crate) struct RegionMigrationProcedureTask {
+pub struct RegionMigrationProcedureTask {
     pub(crate) cluster_id: ClusterId,
     pub(crate) region_id: RegionId,
     pub(crate) from_peer: Peer,
     pub(crate) to_peer: Peer,
 }
 
+impl RegionMigrationProcedureTask {
+    pub fn new(cluster_id: ClusterId, region_id: RegionId, from_peer: Peer, to_peer: Peer) -> Self {
+        Self {
+            cluster_id,
+            region_id,
+            from_peer,
+            to_peer,
+        }
+    }
+}
+
 impl Display for RegionMigrationProcedureTask {
     fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
         write!(
@@ -223,7 +234,7 @@ impl RegionMigrationManager {
     }
 
     /// Submits a new region migration procedure.
-    pub(crate) async fn submit_procedure(
+    pub async fn submit_procedure(
         &self,
         task: RegionMigrationProcedureTask,
     ) -> Result<Option<ProcedureId>> {