fix(repository-elasticsearch): use a mapping for labels compatible wi…

…th opensearch
kestra-io · Jun 22, 2022 · d87a83c · d87a83c
1 parent 8a17519
commit d87a83c
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 5 deletions.
diff --git a/...csearch/src/main/java/io/kestra/repository/elasticsearch/ElasticSearchFlowRepository.java b/...csearch/src/main/java/io/kestra/repository/elasticsearch/ElasticSearchFlowRepository.java
@@ -205,10 +205,10 @@ public ArrayListTotal<Flow> find(
 
         if (labels != null) {
             labels.forEach((key, value) -> {
+                bool.must(QueryBuilders.termQuery("labelsMap.key", key));
+
                 if (value != null) {
-                    bool.must(QueryBuilders.termQuery("labels." + key, value));
-                } else {
-                    bool.must(QueryBuilders.existsQuery("labels." + key));
+                    bool.must(QueryBuilders.termQuery("labelsMap.value", value));
                 }
             });
         }
@@ -318,6 +318,14 @@ public Flow save(Flow flow, CrudEventType crudEventType) throws ConstraintViolat
         try {
             Map<String, Object> flowMap = JacksonMapper.toMap(flow);
             flowMap.put("sourceCode", YAML_MAPPER.writeValueAsString(flow));
+            if (flow.getLabels() != null) {
+                flowMap.put("labelsMap", flow.getLabels()
+                    .entrySet()
+                    .stream()
+                    .map(e -> Map.of("key", e.getKey(), "value", e.getValue()))
+                    .collect(Collectors.toList())
+                );
+            }
             json = JSON_MAPPER.writeValueAsString(flowMap);
         } catch (JsonProcessingException e) {
             throw new RuntimeException(e);

diff --git a/repository-elasticsearch/src/main/resources/mappings/flow.yml b/repository-elasticsearch/src/main/resources/mappings/flow.yml
@@ -37,10 +37,15 @@ properties:
         type: keyword
       type:
         type: keyword
-  labels:
-    type: flattened
   deleted:
     type: boolean
   sourceCode:
     type: text
     term_vector: with_positions_offsets
+  labelsMap:
+    type: object
+    properties:
+      key:
+        type: keyword
+      value:
+        type: keyword