Move key spec generation to association query

Author: Burak Serdar

crud/src/main/java/com/redhat/lightblue/assoc/ep/Assemble.java

@@ -218,9 +218,7 @@ public void commit() {
                 // Try to build an index from results
                 MemDocIndex docIndex=null;
                 if(aq.getQuery()!=null&&numSlots>MEM_INDEX_THRESHOLD) {
-                    // Lets see if we can build a key spec from this query
-                    GetIndexKeySpec giks=new GetIndexKeySpec(aq.getQueryFieldInfo());
-                    KeySpec keySpec=giks.iterate(aq.getQuery());
+                    KeySpec keySpec=aq.getIndexKeySpec();
                     LOGGER.debug("In-memory index key spec:{}",keySpec);
                     if(keySpec!=null) {
                         // There is a key spec, meaning we can index the docs

crud/src/main/java/com/redhat/lightblue/assoc/ep/AssociationQuery.java

@@ -12,6 +12,9 @@
 import com.redhat.lightblue.assoc.QueryFieldInfo;
 import com.redhat.lightblue.assoc.AnalyzeQuery;
 
+import com.redhat.lightblue.mindex.GetIndexKeySpec;
+import com.redhat.lightblue.mindex.KeySpec;
+
 import com.redhat.lightblue.query.QueryExpression;
 
 /**
@@ -25,6 +28,8 @@ public class AssociationQuery {
     // If non-null, query is either always true or always false
     private final Boolean always;
     private final List<QueryFieldInfo> qfi;
+    // In-memory index key spec
+    private final KeySpec keySpec;
 
     public AssociationQuery(CompositeMetadata root,
                             CompositeMetadata currentEntity,
@@ -61,11 +66,18 @@ public AssociationQuery(CompositeMetadata root,
             } else {
                 always=null;
             }
+            keySpec=null;
         } else {
             query = Searches.and(queries);
+            GetIndexKeySpec giks=new GetIndexKeySpec(qfi);
+            keySpec=giks.iterate(query);
             always=null;
         }
     }
+
+    public KeySpec getIndexKeySpec() {
+        return keySpec;
+    }
 
     public List<QueryFieldInfo> getQueryFieldInfo() {
         return qfi;