rust/greptimedb
1
0
Fork 0
mirror of https://github.com/GreptimeTeam/greptimedb.git synced 2025-12-27 08:29:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: add table region key to delete in upgrade tool (#2214)

Browse Source
This commit is contained in:
shuiyisong
2023-08-21 16:16:10 +08:00
committed by GitHub
parent c5f507c20e
commit 9352649f22
1 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/cmd/src/cli/upgrade.rs
View File

@@ -133,17 +133,18 @@ impl MigrateTableMetadata {
);
while let Some((key, value)) = stream.try_next().await.context(error::IterStreamSnafu)? {
self.migrate_table_route_key(value).await?;
let table_id = self.migrate_table_route_key(value).await?;
keys.push(key);
keys.push(TableRegionKey::new(table_id).as_raw_key())
}
info!("Total migrated TableRouteKeys: {}", keys.len());
info!("Total migrated TableRouteKeys: {}", keys.len() / 2);
self.delete_migrated_keys(keys).await;
Ok(())
}
async fn migrate_table_route_key(&self, value: TableRouteValue) -> Result<()> {
async fn migrate_table_route_key(&self, value: TableRouteValue) -> Result<u32> {
let table_route = TableRoute::try_from_raw(
&value.peers,
value.table_route.expect("expected table_route"),
@@ -152,7 +153,8 @@ impl MigrateTableMetadata {
let new_table_value = NextTableRouteValue::new(table_route.region_routes);
let new_key = NextTableRouteKey::new(table_route.table.id as u32);
let table_id = table_route.table.id as u32;
let new_key = NextTableRouteKey::new(table_id);
info!("Creating '{new_key}'");
if self.dryrun {
@@ -168,7 +170,7 @@ impl MigrateTableMetadata {
.unwrap();
}
Ok(())
Ok(table_id)
}
async fn migrate_schema_keys(&self) -> Result<()> {
@@ -310,7 +312,7 @@ impl MigrateTableMetadata {
async fn delete_migrated_keys(&self, keys: Vec<Vec<u8>>) {
for keys in keys.chunks(PAGE_SIZE) {
info!("Deleting {} TableGlobalKeys", keys.len());
info!("Deleting {} keys", keys.len());
let req = BatchDeleteRequest {
keys: keys.to_vec(),
prev_kv: false,