fix warehouse uid modify to id

TCeason · TCeason · commit 94f3c60a1ed8 · 2025-02-20T14:53:56.000+08:00
diff --git a/src/meta/app/src/principal/ownership_object.rs b/src/meta/app/src/principal/ownership_object.rs
@@ -84,7 +84,7 @@ impl fmt::Display for OwnershipObject {
             }
             OwnershipObject::UDF { name } => write!(f, "UDF {name}"),
             OwnershipObject::Stage { name } => write!(f, "STAGE {name}"),
-            OwnershipObject::Warehouse { id: uid } => write!(f, "Warehouse {uid}"),
+            OwnershipObject::Warehouse { id } => write!(f, "Warehouse {id}"),
         }
     }
 }
@@ -124,7 +124,7 @@ impl KeyCodec for OwnershipObject {
             }
             OwnershipObject::Stage { name } => b.push_raw("stage-by-name").push_str(name),
             OwnershipObject::UDF { name } => b.push_raw("udf-by-name").push_str(name),
-            OwnershipObject::Warehouse { id: uid } => b.push_raw("warehouse-by-uid").push_str(uid),
+            OwnershipObject::Warehouse { id } => b.push_raw("warehouse-by-id").push_str(id),
         }
     }
 
@@ -171,13 +171,13 @@ impl KeyCodec for OwnershipObject {
                 let name = p.next_str()?;
                 Ok(OwnershipObject::UDF { name })
             }
-            "warehouse-by-uid" => {
-                let uid = p.next_str()?;
-                Ok(OwnershipObject::Warehouse { id: uid })
+            "warehouse-by-id" => {
+                let id = p.next_str()?;
+                Ok(OwnershipObject::Warehouse { id })
             }
             _ => Err(kvapi::KeyError::InvalidSegment {
                 i: p.index(),
-                expect: "database-by-id|database-by-catalog-id|table-by-id|table-by-catalog-id|stage-by-name|udf-by-name|warehouse-by-uid"
+                expect: "database-by-id|database-by-catalog-id|table-by-id|table-by-catalog-id|stage-by-name|udf-by-name|warehouse-by-id"
                     .to_string(),
                 got: q.to_string(),
             }),
diff --git a/src/meta/app/src/principal/tenant_ownership_object_ident.rs b/src/meta/app/src/principal/tenant_ownership_object_ident.rs
@@ -265,7 +265,7 @@ mod tests {
             );
 
             let key = role_grantee.to_string_key();
-            assert_eq!("__fd_object_owners/test/warehouse-by-uid/n87s", key);
+            assert_eq!("__fd_object_owners/test/warehouse-by-id/n87s", key);
 
             let parsed = TenantOwnershipObjectIdent::from_str_key(&key).unwrap();
             assert_eq!(role_grantee, parsed);
diff --git a/src/meta/proto-conv/src/ownership_from_to_protobuf_impl.rs b/src/meta/proto-conv/src/ownership_from_to_protobuf_impl.rs
@@ -89,8 +89,8 @@ impl FromToProto for mt::principal::OwnershipObject {
                 stage,
             }) => Ok(mt::principal::OwnershipObject::Stage { name: stage }),
             pb::ownership_object::Object::Warehouse(
-                pb::ownership_object::OwnershipWarehouseObject { uid },
-            ) => Ok(mt::principal::OwnershipObject::Warehouse { id: uid }),
+                pb::ownership_object::OwnershipWarehouseObject { id },
+            ) => Ok(mt::principal::OwnershipObject::Warehouse { id }),
         }
     }
 
@@ -126,9 +126,9 @@ impl FromToProto for mt::principal::OwnershipObject {
                     stage: name.clone(),
                 }),
             ),
-            mt::principal::OwnershipObject::Warehouse { id: uid } => {
+            mt::principal::OwnershipObject::Warehouse { id } => {
                 Some(pb::ownership_object::Object::Warehouse(
-                    pb::ownership_object::OwnershipWarehouseObject { uid: uid.clone() },
+                    pb::ownership_object::OwnershipWarehouseObject { id: id.clone() },
                 ))
             }
         };
diff --git a/src/meta/proto-conv/tests/it/main.rs b/src/meta/proto-conv/tests/it/main.rs
@@ -117,5 +117,4 @@ mod v116_marked_deleted_index_meta;
 mod v117_webhdfs_add_disable_list_batch;
 mod v118_webhdfs_add_user_name;
 mod v119_virtual_column;
-
 mod v120_warehouse_ownershipobject;
diff --git a/src/meta/proto-conv/tests/it/v120_warehouse_ownershipobject.rs b/src/meta/proto-conv/tests/it/v120_warehouse_ownershipobject.rs
@@ -38,17 +38,17 @@ use crate::common;
 #[test]
 fn test_decode_v120_grant_object() -> anyhow::Result<()> {
     let role_info_v120 = vec![
-        10, 2, 114, 49, 18, 172, 1, 10, 21, 10, 8, 10, 0, 160, 6, 117, 168, 6, 24, 16, 128, 128,
-        128, 1, 160, 6, 117, 168, 6, 24, 10, 31, 10, 21, 18, 13, 10, 7, 100, 101, 102, 97, 117,
-        108, 116, 18, 2, 100, 98, 160, 6, 117, 168, 6, 24, 16, 2, 160, 6, 117, 168, 6, 24, 10, 35,
+        10, 2, 114, 49, 18, 172, 1, 10, 21, 10, 8, 10, 0, 160, 6, 120, 168, 6, 24, 16, 128, 128,
+        128, 1, 160, 6, 120, 168, 6, 24, 10, 31, 10, 21, 18, 13, 10, 7, 100, 101, 102, 97, 117,
+        108, 116, 18, 2, 100, 98, 160, 6, 120, 168, 6, 24, 16, 2, 160, 6, 120, 168, 6, 24, 10, 35,
         10, 25, 26, 17, 10, 7, 100, 101, 102, 97, 117, 108, 116, 18, 2, 100, 98, 26, 2, 116, 98,
-        160, 6, 117, 168, 6, 24, 16, 2, 160, 6, 117, 168, 6, 24, 10, 22, 10, 12, 34, 4, 10, 2, 102,
-        49, 160, 6, 117, 168, 6, 24, 16, 1, 160, 6, 117, 168, 6, 24, 10, 24, 10, 12, 42, 4, 10, 2,
-        115, 49, 160, 6, 117, 168, 6, 24, 16, 128, 128, 32, 160, 6, 117, 168, 6, 24, 10, 21, 10, 8,
-        10, 0, 160, 6, 117, 168, 6, 24, 16, 254, 255, 191, 1, 160, 6, 117, 168, 6, 24, 160, 6, 117,
+        160, 6, 120, 168, 6, 24, 16, 2, 160, 6, 120, 168, 6, 24, 10, 22, 10, 12, 34, 4, 10, 2, 102,
+        49, 160, 6, 120, 168, 6, 24, 16, 1, 160, 6, 120, 168, 6, 24, 10, 24, 10, 12, 42, 4, 10, 2,
+        115, 49, 160, 6, 120, 168, 6, 24, 16, 128, 128, 32, 160, 6, 120, 168, 6, 24, 10, 21, 10, 8,
+        10, 0, 160, 6, 120, 168, 6, 24, 16, 254, 255, 191, 1, 160, 6, 120, 168, 6, 24, 160, 6, 120,
         168, 6, 24, 26, 23, 49, 57, 55, 48, 45, 48, 49, 45, 48, 49, 32, 48, 48, 58, 48, 48, 58, 48,
         48, 32, 85, 84, 67, 34, 23, 49, 57, 55, 48, 45, 48, 49, 45, 48, 49, 32, 48, 48, 58, 48, 48,
-        58, 48, 48, 32, 85, 84, 67, 160, 6, 117, 168, 6, 24,
+        58, 48, 48, 32, 85, 84, 67, 160, 6, 120, 168, 6, 24,
     ];
     let want = || mt::principal::RoleInfo {
         name: "r1".to_string(),
@@ -91,7 +91,7 @@ fn test_decode_v120_grant_object() -> anyhow::Result<()> {
     };
 
     common::test_pb_from_to(func_name!(), want())?;
-    common::test_load_old(func_name!(), role_info_v120.as_slice(), 117, want())?;
+    common::test_load_old(func_name!(), role_info_v120.as_slice(), 120, want())?;
 
     Ok(())
 }
@@ -100,7 +100,7 @@ fn test_decode_v120_grant_object() -> anyhow::Result<()> {
 fn test_decode_v120_ownership() -> anyhow::Result<()> {
     let ownership_info_v120 = vec![
         10, 2, 114, 49, 18, 19, 42, 11, 10, 9, 97, 117, 110, 105, 113, 117, 101, 105, 100, 160, 6,
-        117, 168, 6, 24, 160, 6, 117, 168, 6, 24,
+        120, 168, 6, 24, 160, 6, 120, 168, 6, 24,
     ];
 
     let want = || mt::principal::OwnershipInfo {
@@ -110,7 +110,7 @@ fn test_decode_v120_ownership() -> anyhow::Result<()> {
         },
     };
     common::test_pb_from_to(func_name!(), want())?;
-    common::test_load_old(func_name!(), ownership_info_v120.as_slice(), 117, want())?;
+    common::test_load_old(func_name!(), ownership_info_v120.as_slice(), 120, want())?;
 
     Ok(())
 }
diff --git a/src/meta/protos/proto/ownership.proto b/src/meta/protos/proto/ownership.proto
@@ -47,7 +47,7 @@ message OwnershipObject {
   }
 
   message OwnershipWarehouseObject {
-    string uid = 1;
+    string id = 1;
   }
 
   oneof object {

Original file line number	Diff line number	Diff line change
`@@ -84,7 +84,7 @@ impl fmt::Display for OwnershipObject {`
`84`	`84`	`}`
`85`	`85`	`OwnershipObject::UDF { name } => write!(f, "UDF {name}"),`
`86`	`86`	`OwnershipObject::Stage { name } => write!(f, "STAGE {name}"),`
`87`		`- OwnershipObject::Warehouse { id: uid } => write!(f, "Warehouse {uid}"),`
	`87`	`+ OwnershipObject::Warehouse { id } => write!(f, "Warehouse {id}"),`
`88`	`88`	`}`
`89`	`89`	`}`
`90`	`90`	`}`
`@@ -124,7 +124,7 @@ impl KeyCodec for OwnershipObject {`
`124`	`124`	`}`
`125`	`125`	`OwnershipObject::Stage { name } => b.push_raw("stage-by-name").push_str(name),`
`126`	`126`	`OwnershipObject::UDF { name } => b.push_raw("udf-by-name").push_str(name),`
`127`		`- OwnershipObject::Warehouse { id: uid } => b.push_raw("warehouse-by-uid").push_str(uid),`
	`127`	`+ OwnershipObject::Warehouse { id } => b.push_raw("warehouse-by-id").push_str(id),`
`128`	`128`	`}`
`129`	`129`	`}`
`130`	`130`
`@@ -171,13 +171,13 @@ impl KeyCodec for OwnershipObject {`
`171`	`171`	`let name = p.next_str()?;`
`172`	`172`	`Ok(OwnershipObject::UDF { name })`
`173`	`173`	`}`
`174`		`- "warehouse-by-uid" => {`
`175`		`- let uid = p.next_str()?;`
`176`		`- Ok(OwnershipObject::Warehouse { id: uid })`
	`174`	`+ "warehouse-by-id" => {`
	`175`	`+ let id = p.next_str()?;`
	`176`	`+ Ok(OwnershipObject::Warehouse { id })`
`177`	`177`	`}`
`178`	`178`	`_ => Err(kvapi::KeyError::InvalidSegment {`
`179`	`179`	`i: p.index(),`
`180`		`- expect: "database-by-id\|database-by-catalog-id\|table-by-id\|table-by-catalog-id\|stage-by-name\|udf-by-name\|warehouse-by-uid"`
	`180`	`+ expect: "database-by-id\|database-by-catalog-id\|table-by-id\|table-by-catalog-id\|stage-by-name\|udf-by-name\|warehouse-by-id"`
`181`	`181`	`.to_string(),`
`182`	`182`	`got: q.to_string(),`
`183`	`183`	`}),`
Original file line number	Diff line number	Diff line change
`@@ -47,7 +47,7 @@ message OwnershipObject {`
`47`	`47`	`}`
`48`	`48`
`49`	`49`	`message OwnershipWarehouseObject {`
`50`		`- string uid = 1;`
	`50`	`+ string id = 1;`
`51`	`51`	`}`
`52`	`52`
`53`	`53`	`oneof object {`