Testing filtering queries

corda · May 13, 2024 · 241310b · 241310b
1 parent bd416c8
commit 241310b
Show file tree

Hide file tree

Showing 2 changed files with 49 additions and 2 deletions.
diff --git a/...racts/src/main/kotlin/com/r3/developers/customStateJSONandQuery/states/CustomChatQuery.kt b/...racts/src/main/kotlin/com/r3/developers/customStateJSONandQuery/states/CustomChatQuery.kt
@@ -1,6 +1,10 @@
 package com.r3.developers.customStateJSONandQuery.states
 
+import net.corda.v5.ledger.utxo.StateAndRef
+import net.corda.v5.ledger.utxo.query.VaultNamedQueryCollector
 import net.corda.v5.ledger.utxo.query.VaultNamedQueryFactory
+import net.corda.v5.ledger.utxo.query.VaultNamedQueryStateAndRefFilter
+import net.corda.v5.ledger.utxo.query.VaultNamedQueryStateAndRefTransformer
 import net.corda.v5.ledger.utxo.query.registration.VaultNamedQueryBuilderFactory
 
 class ChatCustomQueryFactory : VaultNamedQueryFactory {
@@ -16,5 +20,49 @@ class ChatCustomQueryFactory : VaultNamedQueryFactory {
                 "WHERE visible_states.custom_representation -> 'com.r3.developers.customStateJSONandQuery.states.ChatState' ->> 'messageContentFrom' = :nameOfSender"
             )
             .register()
+        vaultNamedQueryBuilderFactory.create("GET_MSG_FROM_ALICE")
+            .whereJson(
+                "WHERE visible_states.custom_representation ? 'com.r3.developers.cordapptemplate.utxoexample.states.ChatState' "
+            )
+            .filter(CustomQueryFilter())
+            .register()
+
+        vaultNamedQueryBuilderFactory.create("GET_ALL_MSGS_FROM")
+            .whereJson(
+                "WHERE visible_states.custom_representation ? 'com.r3.developers.cordapptemplate.utxoexample.states.ChatState' "
+            )
+            .map(CustomQueryTransformer())
+            .register()
+
+        vaultNamedQueryBuilderFactory.create("GET_MSG_AMOUNT")
+            .whereJson(
+                "WHERE visible_states.custom_representation -> 'com.r3.developers.cordapptemplate.utxoexample.states.ChatState' ->> 'messageContentFrom' = :nameOfSender"
+            )
+            .collect(CustomQueryCollector())
+            .register()
+    }
+}
+
+class CustomQueryFilter : VaultNamedQueryStateAndRefFilter<ChatState> {
+    override fun filter(data: StateAndRef<ChatState>, parameters: MutableMap<String, Any>): Boolean {
+        return true
+    }
+}
+
+class CustomQueryTransformer : VaultNamedQueryStateAndRefTransformer<ChatState, String> {
+    override fun transform(data: StateAndRef<ChatState>, parameters: MutableMap<String, Any>): String {
+        return data.state.contractState.messageFrom.toString()
+    }
+}
+
+class CustomQueryCollector : VaultNamedQueryCollector<String, Int> {
+    override fun collect(
+        resultSet: MutableList<String>,
+        parameters: MutableMap<String, Any>
+    ): VaultNamedQueryCollector.Result<Int> {
+        return VaultNamedQueryCollector.Result(
+            listOf(resultSet.size),
+            true
+        )
     }
 }
diff --git a/.../kotlin/com/r3/developers/customStateJSONandQuery/workflows/ListChatsByCustomQueryFlow.kt b/.../kotlin/com/r3/developers/customStateJSONandQuery/workflows/ListChatsByCustomQueryFlow.kt
@@ -31,8 +31,7 @@ class ListChatsByCustomQueryFlow : ClientStartableFlow {
         log.info("ListChatsByCustomQueryFlow.call() called")
 
         //this is our custom query
-        val resultSet = ledgerService.query("GET_MSG_FROM", StateAndRef::class.java)
-            .setParameter("nameOfSender", "CN=Alice, OU=Test Dept, O=R3, L=London, C=GB")
+        val resultSet = ledgerService.query("GET_MSG_FROM_ALICE", StateAndRef::class.java)
             .setCreatedTimestampLimit(Instant.now()).setLimit(1000)
             .execute()