N1QL DSL WIP

ygree · Jul 9, 2018 · e9b43d8 · e9b43d8
1 parent add5fb2
commit e9b43d8
Show file tree

Hide file tree

Showing 2 changed files with 46 additions and 0 deletions.
diff --git a/readside-impl/src/main/java/com/lightbend/readside/impl/ReadSideRepository.java b/readside-impl/src/main/java/com/lightbend/readside/impl/ReadSideRepository.java
@@ -7,6 +7,9 @@
 import com.couchbase.client.java.document.json.JsonObject;
 import com.couchbase.client.java.query.N1qlQuery;
 import com.couchbase.client.java.query.ParameterizedN1qlQuery;
+import com.couchbase.client.java.query.dsl.Expression;
+import com.couchbase.client.java.query.dsl.functions.ConditionalFunctions;
+import com.couchbase.client.java.query.dsl.path.UpdateSetPath;
 import com.lightbend.couchbase.Couchbase;
 import rx.Observable;
 import utils.RxJava8Utils;
@@ -16,6 +19,12 @@
 import java.util.Optional;
 import java.util.concurrent.CompletionStage;
 
+import static com.couchbase.client.java.query.Update.update;
+import static com.couchbase.client.java.query.dsl.Expression.*;
+import static com.couchbase.client.java.query.dsl.functions.ArrayFunctions.arrayIfNull;
+import static com.couchbase.client.java.query.dsl.functions.ArrayFunctions.arrayPrepend;
+import static com.couchbase.client.java.query.dsl.functions.ConditionalFunctions.*;
+
 @Singleton
 public class ReadSideRepository {
 
@@ -39,10 +48,19 @@ public CompletionStage<Done> updateMessage(String name, String message) {
         String queryText = "UPDATE test USE KEYS $1 SET messages = ARRAY_PREPEND($2, IFNULL(messages, [])), message = $2;";
         ParameterizedN1qlQuery query = N1qlQuery.parameterized(queryText, JsonArray.from(docId, message));
 
+/* TODO: it doesn't translates into a correct query
+        UpdateSetPath updateSetPath =
+                update("test")
+                        .useKeys(docId)
+                        .set("message", message)
+                        .set("messages", arrayPrepend(x(message), ifNull(x("messages"), x(JsonArray.empty()))));
+*/
+
         Observable<Done> result = bucket
                 .flatMap(b -> b.insert(doc).map(x -> b))
                 .onExceptionResumeNext(bucket)
                 .flatMap(b -> b.query(query))
+//                .flatMap(b -> b.query(updateSetPath))
                 .map(v -> Done.getInstance());
 
         return RxJava8Utils.fromSingleObservable(result);

diff --git a/readside-impl/src/test/java/com/lightbend/readside/impl/N1qlDslTests.java b/readside-impl/src/test/java/com/lightbend/readside/impl/N1qlDslTests.java
@@ -0,0 +1,28 @@
+package com.lightbend.readside.impl;
+
+import com.couchbase.client.java.document.json.JsonArray;
+import com.couchbase.client.java.query.dsl.path.UpdateSetPath;
+import org.junit.Assert;
+import org.junit.Test;
+
+import static com.couchbase.client.java.query.Update.update;
+import static com.couchbase.client.java.query.dsl.Expression.x;
+import static com.couchbase.client.java.query.dsl.functions.ArrayFunctions.arrayPrepend;
+import static com.couchbase.client.java.query.dsl.functions.ConditionalFunctions.ifNull;
+
+public class N1qlDslTests {
+
+    @Test
+    public void test() {
+
+        UpdateSetPath updateSetPath =
+                update("test")
+                        .useKeys("$1")
+                        .set("messages", arrayPrepend(ifNull(x("messages"), x(JsonArray.empty())), x("$2")))
+                        .set("message", "$2");
+
+
+        Assert.assertEquals("UPDATE test USE KEYS $1 SET messages = ARRAY_PREPEND($2, IFNULL(messages, [])), message = $2;", updateSetPath.toString());
+
+    }
+}