allow to customize json search function

Author: Martin Fekete

rsql-jpa/src/main/java/io/github/perplexhub/rsql/jsonb/JsonbExpressionBuilder.java

@@ -161,10 +161,6 @@ public ArgValue convert(String s) {
             Map.entry(BETWEEN, "(%1$s >= %2$s && %1$s <= %3$s)")
     );
 
-    private static final String JSONB_PATH_EXISTS = "jsonb_path_exists";
-
-    private static final String JSONB_PATH_EXISTS_TZ = "jsonb_path_exists_tz";
-
     private final ComparisonOperator operator;
     private final String keyPath;
     private final List<ArgValue> values;

rsql-jpa/src/main/java/io/github/perplexhub/rsql/jsonb/JsonbSupport.java

@@ -1,35 +1,43 @@
 package io.github.perplexhub.rsql.jsonb;
 
 
-import static io.github.perplexhub.rsql.RSQLVisitorBase.getEntityManagerMap;
-
-import java.lang.reflect.Field;
-import java.util.EnumSet;
-import java.util.Map;
-import java.util.Optional;
-import java.util.Set;
-
 import cz.jirutka.rsql.parser.ast.ComparisonNode;
 import cz.jirutka.rsql.parser.ast.ComparisonOperator;
 import io.github.perplexhub.rsql.RSQLOperators;
 import io.github.perplexhub.rsql.RSQLVisitorBase;
 import io.github.perplexhub.rsql.ResolvedExpression;
 import jakarta.persistence.Column;
 import jakarta.persistence.criteria.CriteriaBuilder;
-import jakarta.persistence.criteria.Expression;
 import jakarta.persistence.criteria.Path;
-import jakarta.persistence.criteria.Predicate;
 import jakarta.persistence.metamodel.Attribute;
 import jakarta.persistence.metamodel.ManagedType;
 import org.springframework.orm.jpa.vendor.Database;
 
+import java.lang.reflect.Field;
+import java.util.EnumSet;
+import java.util.Map;
+import java.util.Optional;
+import java.util.Set;
+
+import static io.github.perplexhub.rsql.RSQLVisitorBase.getEntityManagerMap;
+
 /**
  * Support for jsonb expression.
  */
 public class JsonbSupport {
 
     public static boolean DATE_TIME_SUPPORT = false;
 
+    /**
+     * Postgresql {@code jsonb_path_exists} function to use
+     */
+    public static String JSONB_PATH_EXISTS = "jsonb_path_exists";
+
+    /**
+     * Postgresql {@code jsonb_path_exists_tz} function to use
+     */
+    public static String JSONB_PATH_EXISTS_TZ = "jsonb_path_exists_tz";
+
     private static final Set<Database> JSON_SUPPORT = EnumSet.of(Database.POSTGRESQL);
 
     private static final Map<ComparisonOperator, ComparisonOperator> NEGATE_OPERATORS =

rsql-jpa/src/test/java/io/github/perplexhub/rsql/jsonb/JsonbExpressionBuilderTest.java

@@ -12,7 +12,8 @@
 import java.util.Objects;
 import java.util.stream.Stream;
 
-import static org.junit.jupiter.api.Assertions.*;
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertThrows;
 import static org.junit.jupiter.params.provider.Arguments.arguments;
 
 class JsonbExpressionBuilderTest {
@@ -37,6 +38,27 @@ void testJsonbPathExpressionWithTemporal(ComparisonOperator operator, String key
         assertEquals(expectedJsonbPath, expression.jsonbPath());
     }
 
+    @ParameterizedTest
+    @MethodSource("customized")
+    void testJsonbPathExpressionCustomized(ComparisonOperator operator, String keyPath, List<String> arguments, String expectedJsonbFunction, String expectedJsonbPath) {
+        String jsonbPathExists = JsonbSupport.JSONB_PATH_EXISTS;
+        String jsonbPathExistsTz = JsonbSupport.JSONB_PATH_EXISTS_TZ;
+        try {
+            JsonbSupport.JSONB_PATH_EXISTS = "my_jsonb_path_exists";
+            JsonbSupport.JSONB_PATH_EXISTS_TZ = "my_jsonb_path_exists_tz";
+            JsonbSupport.DATE_TIME_SUPPORT = true;
+            JsonbExpressionBuilder builder = new JsonbExpressionBuilder(operator, keyPath, arguments);
+            var expression = builder.getJsonPathExpression();
+            assertEquals(expectedJsonbFunction, expression.jsonbFunction());
+            assertEquals(expectedJsonbPath, expression.jsonbPath());
+        } catch (Exception e) {
+            throw e;
+        } finally {
+            JsonbSupport.JSONB_PATH_EXISTS = jsonbPathExists;
+            JsonbSupport.JSONB_PATH_EXISTS_TZ = jsonbPathExistsTz;
+        }
+    }
+
     static Stream<Arguments> data() {
         return Stream.of(
                 allOperators(),
@@ -78,6 +100,17 @@ static Stream<Arguments> conversion() {
         ).filter(Objects::nonNull);
     }
 
+    static Stream<Arguments> customized() {
+
+        return Stream.of(
+                arguments(RSQLOperators.EQUAL, "json.equal_key", Collections.singletonList("value"), "my_jsonb_path_exists", "$.equal_key ? (@ == \"value\")"),
+                arguments(RSQLOperators.GREATER_THAN, "json.greater_than_key", Collections.singletonList("value"), "my_jsonb_path_exists", "$.greater_than_key ? (@ > \"value\")"),
+                arguments(RSQLOperators.EQUAL, "json.equal_key", Collections.singletonList("1970-01-01T00:00:00.000"), "my_jsonb_path_exists", "$.equal_key ? (@.datetime() == \"1970-01-01T00:00:00.000\".datetime())"),
+                arguments(RSQLOperators.EQUAL, "json.equal_key", Collections.singletonList("1970-01-01T00:00:00.000Z"), "my_jsonb_path_exists_tz", "$.equal_key ? (@.datetime() == \"1970-01-01T00:00:00.000Z\".datetime())"),
+                null
+        ).filter(Objects::nonNull);
+    }
+
     static Stream<Arguments> temporal() {
 
         return Stream.of(