Add ElasticSearchLiteQueryCreator

Author: swtwsk

CHANGELOG.md

@@ -18,6 +18,10 @@ New properties are:
   (along with a "default"
   [GenericGetQueryCreator](src/main/java/com/getindata/connectors/http/internal/table/lookup/querycreators/GenericGetQueryCreator.java)
   implementation) for customization of queries prepared by Lookup Source for its HTTP requests.
+- Add [ElasticSearchLiteQueryCreator](src/main/java/com/getindata/connectors/http/internal/table/lookup/querycreators/ElasticSearchLiteQueryCreator.java)
+  that prepares [`q` parameter query](https://www.elastic.co/guide/en/elasticsearch/reference/current/search-search.html#search-api-query-params-q)
+  using Lucene query string syntax (in first versions of ElasticSearch called
+  [Search _Lite_](https://www.elastic.co/guide/en/elasticsearch/guide/current/search-lite.html)).
 
 ## [0.4.0] - 2022-08-31
 

src/main/java/com/getindata/connectors/http/internal/table/lookup/querycreators/ElasticSearchLiteQueryCreator.java

@@ -0,0 +1,34 @@
+package com.getindata.connectors.http.internal.table.lookup.querycreators;
+
+import java.util.List;
+import java.util.stream.Collectors;
+
+import com.getindata.connectors.http.LookupArg;
+import com.getindata.connectors.http.LookupQueryCreator;
+
+/**
+ * A {@link LookupQueryCreator} that prepares <a href="https://www.elastic.co/guide/en/elasticsearch/reference/current/search-search.html#search-api-query-params-q"><code>q</code> parameter query</a>
+ * for ElasticSearch <i>Search API</i> using Lucene query string syntax (in first versions of the
+ * ElasticSearch called <a href="https://www.elastic.co/guide/en/elasticsearch/guide/current/search-lite.html">Search <i>Lite</i></a>).
+ */
+public class ElasticSearchLiteQueryCreator implements LookupQueryCreator {
+    private static final String ENCODED_SPACE = "%20";
+    private static final String ENCODED_QUOTATION_MARK = "%22";
+
+    @Override
+    public String createLookupQuery(List<LookupArg> params) {
+        var luceneQuery = params.stream()
+            .map(ElasticSearchLiteQueryCreator::processLookupArg)
+            .collect(Collectors.joining(ENCODED_SPACE + "AND" + ENCODED_SPACE));
+
+        return luceneQuery.isEmpty() ? "" : ("q=" + luceneQuery);
+    }
+
+    private static String processLookupArg(LookupArg arg) {
+        return arg.getArgName()
+               + ":"
+               + ENCODED_QUOTATION_MARK
+               + arg.getArgValue()
+               + ENCODED_QUOTATION_MARK;
+    }
+}

src/main/java/com/getindata/connectors/http/internal/table/lookup/querycreators/ElasticSearchLiteQueryCreatorFactory.java

@@ -0,0 +1,35 @@
+package com.getindata.connectors.http.internal.table.lookup.querycreators;
+
+import java.util.Set;
+
+import org.apache.flink.configuration.ConfigOption;
+
+import com.getindata.connectors.http.LookupQueryCreator;
+import com.getindata.connectors.http.LookupQueryCreatorFactory;
+
+/**
+ * Factory for creating {@link ElasticSearchLiteQueryCreator}.
+ */
+public class ElasticSearchLiteQueryCreatorFactory implements LookupQueryCreatorFactory {
+    public static final String IDENTIFIER = "elasticsearch-lite";
+
+    @Override
+    public LookupQueryCreator createLookupQueryCreator() {
+        return new ElasticSearchLiteQueryCreator();
+    }
+
+    @Override
+    public String factoryIdentifier() {
+        return IDENTIFIER;
+    }
+
+    @Override
+    public Set<ConfigOption<?>> requiredOptions() {
+        return Set.of();
+    }
+
+    @Override
+    public Set<ConfigOption<?>> optionalOptions() {
+        return Set.of();
+    }
+}

src/main/resources/META-INF/services/org.apache.flink.table.factories.Factory

@@ -1,4 +1,5 @@
 com.getindata.connectors.http.internal.table.lookup.HttpLookupTableSourceFactory
+com.getindata.connectors.http.internal.table.lookup.querycreators.ElasticSearchLiteQueryCreatorFactory
 com.getindata.connectors.http.internal.table.lookup.querycreators.GenericGetQueryCreatorFactory
 com.getindata.connectors.http.internal.table.sink.HttpDynamicTableSinkFactory
 com.getindata.connectors.http.internal.table.sink.Slf4jHttpPostRequestCallbackFactory

src/test/java/com/getindata/connectors/http/internal/table/lookup/querycreators/ElasticSearchLiteQueryCreatorTest.java

@@ -0,0 +1,33 @@
+package com.getindata.connectors.http.internal.table.lookup.querycreators;
+
+import java.util.List;
+import java.util.stream.Stream;
+
+import org.junit.jupiter.params.ParameterizedTest;
+import org.junit.jupiter.params.provider.Arguments;
+import org.junit.jupiter.params.provider.MethodSource;
+import static org.assertj.core.api.Assertions.assertThat;
+
+import com.getindata.connectors.http.LookupArg;
+
+public class ElasticSearchLiteQueryCreatorTest {
+    static Stream<Arguments> queryArguments() {
+        return Stream.of(
+            Arguments.of(List.of(), ""),
+            Arguments.of(List.of(new LookupArg("key1", "val1")), "q=key1:%22val1%22"),
+            Arguments.of(List.of(
+                new LookupArg("key1", "val1"),
+                new LookupArg("key2", "val2"),
+                new LookupArg("key3", "3")
+            ), "q=key1:%22val1%22%20AND%20key2:%22val2%22%20AND%20key3:%223%22")
+        );
+    }
+
+    @ParameterizedTest
+    @MethodSource("queryArguments")
+    public void testGenericGetQueryCreation(List<LookupArg> args, String expectedQuery) {
+        var queryCreator = new ElasticSearchLiteQueryCreator();
+        var createdQuery = queryCreator.createLookupQuery(args);
+        assertThat(createdQuery).isEqualTo(expectedQuery);
+    }
+}