fix: new multiple write requests when tag names are different in prom…

… write (#888)
apache · May 11, 2023 · 9e85ce0 · 9e85ce0
1 parent b27a72f
commit 9e85ce0
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/proxy/src/http/prom.rs b/proxy/src/http/prom.rs
@@ -407,7 +407,8 @@ fn convert_write_request(req: WriteRequest) -> Result<Vec<WriteTableRequest>> {
             .unzip();
 
         req_by_metric
-            .entry(metric.to_string())
+            // fields with same tag names will be grouped together
+            .entry((metric.clone(), tag_names.clone()))
             .or_insert_with(|| WriteTableRequest {
                 table: metric,
                 tag_names,

diff --git a/proxy/src/influxdb/types.rs b/proxy/src/influxdb/types.rs
@@ -491,12 +491,16 @@ pub(crate) fn convert_write_request(req: WriteRequest) -> Result<Vec<WriteTableR
         tag_set.sort_unstable_by(|a, b| a.0.cmp(&b.0));
         // sort by field key
         line.field_set.sort_unstable_by(|a, b| a.0.cmp(&b.0));
-
+        let tag_names = tag_set
+            .iter()
+            .map(|(tagk, _)| tagk.to_string())
+            .collect::<Vec<_>>();
         let req_for_one_measurement = req_by_measurement
-            .entry(line.series.measurement.to_string())
+            // fields with same tag names will be grouped together
+            .entry((line.series.measurement.to_string(), tag_names.clone()))
             .or_insert_with(|| WriteTableRequest {
                 table: line.series.measurement.to_string(),
-                tag_names: tag_set.iter().map(|(tagk, _)| tagk.to_string()).collect(),
+                tag_names,
                 field_names: line
                     .field_set
                     .iter()