feat: initial support for table branching and tagging (#19035)

* branching and tag

* table lvt

* add test

* update

* fix review comments

* fix

* update

Author: zhyass

Cargo.lock

@@ -107,6 +107,7 @@ databend-enterprise-resources-management = { path = "src/query/ee_features/resou
 databend-enterprise-row-access-policy-feature = { path = "src/query/ee_features/row_access_policy" }
 databend-enterprise-storage-encryption = { path = "src/query/ee_features/storage_encryption" }
 databend-enterprise-stream-handler = { path = "src/query/ee_features/stream_handler" }
+databend-enterprise-table-ref-handler = { path = "src/query/ee_features/table_ref_handler" }
 databend-enterprise-vacuum-handler = { path = "src/query/ee_features/vacuum_handler" }
 databend-enterprise-virtual-column = { path = "src/query/ee_features/virtual_column" }
 databend-functions-scalar-arithmetic = { path = "src/query/functions/src/scalars/arithmetic" }

Cargo.toml

@@ -16,7 +16,6 @@ hyper-util = { workspace = true }
 prost = { workspace = true }
 serde = { workspace = true }
 tonic = { workspace = true }
-#tonic-prost = { workspace = true }
 
 [build-dependencies]
 lenient_semver = { workspace = true }

src/common/cloud_control/Cargo.toml

@@ -333,6 +333,20 @@ build_exceptions! {
     UnsupportedEngineParams(2703),
 }
 
+// Table reference Errors [2745-2749]
+build_exceptions! {
+    /// Unknown reference
+    UnknownReference(2745),
+    /// Reference already exists
+    ReferenceAlreadyExists(2746),
+    /// Illegal reference
+    IllegalReference(2747),
+    /// Mismatched reference type
+    MismatchedReferenceType(2748),
+    /// Reference expired
+    ReferenceExpired(2749),
+}
+
 // License Errors [1401-1404]
 build_exceptions! {
     /// License key parse error
@@ -509,7 +523,7 @@ build_exceptions! {
     IllegalUser(2218),
 }
 
-// Database and Catalog Management Errors [2301-2317, 2321-2323]
+// Database and Catalog Management Errors [2301-2317, 2321-2327]
 build_exceptions! {
     /// Database already exists
     DatabaseAlreadyExists(2301),
@@ -549,6 +563,8 @@ build_exceptions! {
     RowAccessPolicyAlreadyExists(2324),
     /// General failures met while garbage collecting database meta
     GeneralDbGcFailure(2325),
+    /// Table snapshot expired
+    TableSnapshotExpired(2327),
 }
 
 // Stage and Connection Errors [2501-2505, 2510-2512]

src/common/exception/src/exception_code.rs

@@ -39,6 +39,8 @@ pub enum Feature {
     StorageEncryption,
     #[serde(alias = "stream", alias = "STREAM")]
     Stream,
+    #[serde(alias = "table_ref", alias = "TABLE_REF")]
+    TableRef,
     #[serde(alias = "attach_table", alias = "ATTACH_TABLE")]
     AttacheTable,
     #[serde(alias = "amend_table", alias = "AMEND_TABLE")]
@@ -80,6 +82,7 @@ impl fmt::Display for Feature {
             Feature::ComputedColumn => write!(f, "computed_column"),
             Feature::StorageEncryption => write!(f, "storage_encryption"),
             Feature::Stream => write!(f, "stream"),
+            Feature::TableRef => write!(f, "table_ref"),
             Feature::AttacheTable => write!(f, "attach_table"),
             Feature::AmendTable => write!(f, "amend_table"),
             Feature::SystemManagement => write!(f, "system_management"),
@@ -118,6 +121,7 @@ impl Feature {
             | (Feature::Vacuum, Feature::Vacuum)
             | (Feature::LicenseInfo, Feature::LicenseInfo)
             | (Feature::Stream, Feature::Stream)
+            | (Feature::TableRef, Feature::TableRef)
             | (Feature::DataMask, Feature::DataMask)
             | (Feature::RowAccessPolicy, Feature::RowAccessPolicy)
             | (Feature::VirtualColumn, Feature::VirtualColumn)
@@ -226,6 +230,10 @@ mod tests {
             Feature::Stream,
             serde_json::from_str::<Feature>("\"Stream\"").unwrap()
         );
+        assert_eq!(
+            Feature::TableRef,
+            serde_json::from_str::<Feature>("\"TableRef\"").unwrap()
+        );
         assert_eq!(
             Feature::AttacheTable,
             serde_json::from_str::<Feature>("\"ATTACH_TABLE\"").unwrap()
@@ -287,6 +295,7 @@ mod tests {
                 Feature::ComputedColumn,
                 Feature::StorageEncryption,
                 Feature::Stream,
+                Feature::TableRef,
                 Feature::AttacheTable,
                 Feature::AmendTable,
                 Feature::HilbertClustering,
@@ -298,7 +307,7 @@ mod tests {
         };
 
         assert_eq!(
-            "LicenseInfo{ type: enterprise, org: databend, tenants: [databend_tenant,foo], features: [amend_table,attach_table,computed_column,data_mask,hilbert_clustering,license_info,private_task,row_access_policy,storage_encryption,stream,system_history,vacuum,virtual_column,workload_group] }",
+            "LicenseInfo{ type: enterprise, org: databend, tenants: [databend_tenant,foo], features: [amend_table,attach_table,computed_column,data_mask,hilbert_clustering,license_info,private_task,row_access_policy,storage_encryption,stream,system_history,table_ref,vacuum,virtual_column,workload_group] }",
             license_info.to_string()
         );
     }

src/common/license/src/license.rs

@@ -10,7 +10,6 @@ edition = { workspace = true }
 jiff = { workspace = true }
 
 [dev-dependencies]
-rand = { workspace = true }
 
 [lints]
 workspace = true