Add some more query examples

Author: karesti

infinispan-remote/query/src/main/java/org/infinispan/tutorial/simple/remote/query/InfinispanRemoteQuery.java

@@ -8,6 +8,7 @@
 import org.infinispan.tutorial.simple.connect.TutorialsConnectorHelper;
 
 import java.net.URI;
+import java.util.Arrays;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
@@ -25,19 +26,22 @@
 public class InfinispanRemoteQuery {
 
    public static final String INDEXED_PEOPLE_CACHE = "indexedPeopleCache";
+   public static final String INDEXED_TEAM_CACHE = "indexedTeamCache";
    static RemoteCacheManager client;
    static RemoteCache<PersonKey, Person> peopleCache;
+   static RemoteCache<String, Team> teamCache;
 
    public static void main(String[] args) throws Exception {
       connectToInfinispan();
-
-      addDataToCache();
+      addDataToPeopleCache();
       queryAll();
       queryWithWhereStatementOnValues();
       queryByKey();
       queryWithProjection();
       deleteByQuery();
 
+      addDataToTeamCache();
+      queryWithScoreAndFilterOnNestedValues();
       disconnect(false);
    }
 
@@ -101,6 +105,31 @@ static List<Person> queryWithWhereStatementOnValues() {
       return queryResult;
    }
 
+   private static void queryWithScoreAndFilterOnNestedValues() {
+      System.out.println("== Query and filter on nested values of the team");
+      Query<Team> query = teamCache.query("FROM tutorial.Team t join t.players p where p.bornIn = :bornIn");
+      // Set the parameter value
+      query.setParameter("bornIn", "London");
+      List<Team> teamsWithPeopleFromLondon = query.execute().list();
+      System.out.println("There are only 2 teams " + teamsWithPeopleFromLondon.size());
+      System.out.println(teamsWithPeopleFromLondon);
+
+      // Get the score
+      Query<Object[]> queryWithProjectionAndScore = teamCache.query("select t.teamName, score(t) FROM tutorial.Team t where t.points=88");
+      List<Object[]> results = queryWithProjectionAndScore.execute().list();
+      System.out.println("Team with 88 points: " + results.size());
+      System.out.println("Team name: " + results.get(0)[0]);
+      System.out.println("Score: " + results.get(0)[1]);
+
+      // With filter in nested values
+      Query<Object[]> queryProjection = teamCache.query("select t.teamName, score(t) FROM tutorial.Team t join t.players p where p.lastName = :lastName");
+      queryProjection.setParameter("lastName", "Granger");
+      results = queryProjection.execute().list();
+      System.out.println("Hermione number of teams: " + results.size());
+      System.out.println("Hermione team name: " + results.get(0)[0]);
+   }
+
+
    static void connectToInfinispan() throws Exception {
       ConfigurationBuilder builder = TutorialsConnectorHelper.connectionConfig();
 
@@ -109,7 +138,9 @@ static void connectToInfinispan() throws Exception {
 
       // Use indexed cache
       URI indexedCacheURI = InfinispanRemoteQuery.class.getClassLoader().getResource("indexedCache.xml").toURI();
+      URI teamCacheURI = InfinispanRemoteQuery.class.getClassLoader().getResource("teamCache.xml").toURI();
       builder.remoteCache(INDEXED_PEOPLE_CACHE).configurationURI(indexedCacheURI);
+      builder.remoteCache(INDEXED_TEAM_CACHE).configurationURI(teamCacheURI);
 
       // Connect to the server
       client = TutorialsConnectorHelper.connect(builder);
@@ -119,9 +150,10 @@ static void connectToInfinispan() throws Exception {
 
       // Get the people cache, create it if needed with the default configuration
       peopleCache = client.getCache(INDEXED_PEOPLE_CACHE);
+      teamCache = client.getCache(INDEXED_TEAM_CACHE);
    }
 
-   static void addDataToCache() {
+   static void addDataToPeopleCache() {
       // Create the persons dataset to be stored in the cache
       Map<PersonKey, Person> people = new HashMap<>();
       people.put(new PersonKey("1", "hgranger"),
@@ -137,6 +169,34 @@ static void addDataToCache() {
       peopleCache.putAll(people);
    }
 
+   static void addDataToTeamCache() {
+      // Create the persons dataset to be stored in the cache
+      Map<String, Team> teams = new HashMap<>();
+
+      Team team1 = new Team("Heroic team", 900, Arrays.asList(
+              new Person("Hermione", "Granger", 1990, "London"),
+              new Person("Neville", "Longbottom", 1990, "Manchester"),
+              new Person("Luna", "Lovegood", 1991, "London")
+      ));
+
+      Team team2 = new Team("Villains team", 88, Arrays.asList(
+              new Person("Draco", "Malfoy", 1988, "London"),
+              new Person("Bellatrix", "Lestrange", 1955, "Bristol"),
+              new Person("Dolores", "Umbridge", 1950, "Bristol")
+      ));
+
+      Team team3 = new Team("I don't care team", 1500, Arrays.asList(
+              new Person("Lavender", "Brown", 1990, "Bristol"),
+              new Person("Seamus", "Finnigan", 1990, "Manchester")
+      ));
+
+      teams.put("1", team1);
+      teams.put("2", team2);
+      teams.put("3", team3);
+
+      teamCache.putAll(teams);
+   }
+
    public static void disconnect(boolean removeCaches) {
       if (removeCaches) {
          client.administration().removeCache(INDEXED_PEOPLE_CACHE);

infinispan-remote/query/src/main/java/org/infinispan/tutorial/simple/remote/query/Team.java

@@ -0,0 +1,19 @@
+package org.infinispan.tutorial.simple.remote.query;
+
+import org.infinispan.api.annotations.indexing.Basic;
+import org.infinispan.api.annotations.indexing.Embedded;
+import org.infinispan.api.annotations.indexing.Indexed;
+import org.infinispan.protostream.annotations.Proto;
+
+import java.util.List;
+
+@Proto
+@Indexed
+public record Team(
+        @Basic(projectable = true)
+        String teamName,
+        @Basic(projectable = true)
+        Integer points,
+        @Embedded
+        List<Person> players){
+}

infinispan-remote/query/src/main/java/org/infinispan/tutorial/simple/remote/query/TutorialSchema.java

@@ -3,6 +3,6 @@
 import org.infinispan.protostream.GeneratedSchema;
 import org.infinispan.protostream.annotations.ProtoSchema;
 
-@ProtoSchema(schemaPackageName = "tutorial", includeClasses = { Person.class, PersonKey.class })
+@ProtoSchema(schemaPackageName = "tutorial", includeClasses = { Team.class, Person.class, PersonKey.class })
 public interface TutorialSchema extends GeneratedSchema {
 }

infinispan-remote/query/src/main/resources/teamCache.xml

@@ -0,0 +1,11 @@
+<distributed-cache statistics="true">
+    <encoding>
+        <key media-type="application/x-protostream"/>
+        <value media-type="application/x-protostream"/>
+    </encoding>
+    <indexing enabled="true" storage="filesystem" startup-mode="AUTO">
+        <indexed-entities>
+            <indexed-entity>tutorial.Team</indexed-entity>
+        </indexed-entities>
+    </indexing>
+</distributed-cache>

infinispan-remote/query/src/test/java/org/infinispan/tutorial/simple/remote/query/InfinispanRemoteQueryTest.java

@@ -32,7 +32,7 @@ public void testRemoteQuery() {
         List<Person> people = InfinispanRemoteQuery.queryAll();
         assertEquals(0, people.size());
 
-        InfinispanRemoteQuery.addDataToCache();
+        InfinispanRemoteQuery.addDataToPeopleCache();
 
         people = InfinispanRemoteQuery.queryAll();
         assertEquals(4, people.size());