feat: write logical region to metric engine (#2759)

* transform write request Signed-off-by: Ruihang Xia <[email protected]> * add tests for put request Signed-off-by: Ruihang Xia <[email protected]> * use table_id instead of metric_name Signed-off-by: Ruihang Xia <[email protected]> * fix typo Signed-off-by: Ruihang Xia <[email protected]> * CR sugg. Signed-off-by: Ruihang Xia <[email protected]> * define random state as const Signed-off-by: Ruihang Xia <[email protected]> --------- Signed-off-by: Ruihang Xia <[email protected]>
GreptimeTeam · Nov 17, 2023 · ac4b6cd · ac4b6cd
1 parent 3ab4947
commit ac4b6cd
Show file tree

Hide file tree

Showing 10 changed files with 446 additions and 43 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -63,6 +63,7 @@ edition = "2021"
 license = "Apache-2.0"
 
 [workspace.dependencies]
+ahash = { version = "0.8", features = ["compile-time-rng"] }
 aquamarine = "0.3"
 arrow = { version = "47.0" }
 arrow-array = "47.0"

diff --git a/src/metric-engine/Cargo.toml b/src/metric-engine/Cargo.toml
@@ -5,6 +5,7 @@ edition.workspace = true
 license.workspace = true
 
 [dependencies]
+ahash.workspace = true
 api.workspace = true
 async-trait.workspace = true
 base64.workspace = true

diff --git a/src/metric-engine/src/data_region.rs b/src/metric-engine/src/data_region.rs
@@ -13,12 +13,15 @@
 // limitations under the License.
 
 use api::v1::SemanticType;
+use common_query::Output;
 use common_telemetry::tracing::warn;
 use mito2::engine::MitoEngine;
 use snafu::ResultExt;
 use store_api::metadata::ColumnMetadata;
 use store_api::region_engine::RegionEngine;
-use store_api::region_request::{AddColumn, AlterKind, RegionAlterRequest, RegionRequest};
+use store_api::region_request::{
+    AddColumn, AlterKind, RegionAlterRequest, RegionPutRequest, RegionRequest,
+};
 use store_api::storage::RegionId;
 
 use crate::error::{
@@ -122,6 +125,18 @@ impl DataRegion {
 
         Ok(())
     }
+
+    pub async fn write_data(
+        &self,
+        region_id: RegionId,
+        request: RegionPutRequest,
+    ) -> Result<Output> {
+        let region_id = utils::to_data_region_id(region_id);
+        self.mito
+            .handle_request(region_id, RegionRequest::Put(request))
+            .await
+            .context(MitoWriteOperationSnafu)
+    }
 }
 
 #[cfg(test)]
@@ -184,7 +199,7 @@ mod test {
         let expected = vec![
             "greptime_timestamp",
             "greptime_value",
-            "__metric",
+            "__table_id",
             "__tsid",
             "job",
             "tag2",