SOLR-14419: adding {param:ref} to Query DSL

This commit is contained in:
Mikhail Khludnev 2020-04-26 23:04:00 +03:00
parent 5251c40679
commit 2c8cfa678b
3 changed files with 73 additions and 18 deletions

View File

@ -125,6 +125,8 @@ Improvements
themselves. The collection hints are pushed down to the policy engine so operations for non-matching collections themselves. The collection hints are pushed down to the policy engine so operations for non-matching collections
are not computed at all. (ab, shalin) are not computed at all. (ab, shalin)
* SOLR-14419: json.queries as well as other parameters might be referred via {"param":"ref"} in Query DSL (Mikhail Khludnev)
Optimizations Optimizations
--------------------- ---------------------
* SOLR-8306: Do not collect expand documents when expand.rows=0 (Marshall Sanders, Amelia Henderson) * SOLR-8306: Do not collect expand documents when expand.rows=0 (Marshall Sanders, Amelia Henderson)

View File

@ -98,6 +98,18 @@ class JsonQueryConverter {
qtype = map.keySet().iterator().next(); qtype = map.keySet().iterator().next();
// FUTURE: might want to recurse here instead to handle nested tags (and add tagName as a parameter?) // FUTURE: might want to recurse here instead to handle nested tags (and add tagName as a parameter?)
} }
} else {
if (qtype.equals("param")) {
boolean toplevel;
if (toplevel=(builder.length() == 0)) {
builder.append("{!v=");
}
builder.append("$").append(map.get("param"));
if (toplevel) {
builder.append("}");
}
return;
}
} }
StringBuilder subBuilder = useSubBuilder ? new StringBuilder() : builder; StringBuilder subBuilder = useSubBuilder ? new StringBuilder() : builder;
@ -113,6 +125,9 @@ class JsonQueryConverter {
if (useSubBuilder) { if (useSubBuilder) {
builder.append('$').append(putParam(subBuilder.toString(), additionalParams)); builder.append('$').append(putParam(subBuilder.toString(), additionalParams));
} }
} else {
if(map.size()==1 && map.keySet().iterator().next().equals("param")) {
builder.append("v").append("=$").append(map.get("param")).append(" ");
} else { } else {
for (Map.Entry<String, Object> entry : map.entrySet()) { for (Map.Entry<String, Object> entry : map.entrySet()) {
String key = entry.getKey(); String key = entry.getKey();
@ -138,4 +153,5 @@ class JsonQueryConverter {
} }
} }
} }
}
} }

View File

@ -24,6 +24,7 @@ import org.apache.solr.JSONTestUtil;
import org.apache.solr.SolrTestCaseHS; import org.apache.solr.SolrTestCaseHS;
import org.apache.solr.common.SolrException; import org.apache.solr.common.SolrException;
import org.apache.solr.common.params.CommonParams; import org.apache.solr.common.params.CommonParams;
import org.apache.solr.common.params.ModifiableSolrParams;
import org.apache.solr.request.SolrQueryRequest; import org.apache.solr.request.SolrQueryRequest;
import org.apache.solr.search.CaffeineCache; import org.apache.solr.search.CaffeineCache;
import org.apache.solr.search.DocSet; import org.apache.solr.search.DocSet;
@ -179,7 +180,7 @@ public class TestJsonRequest extends SolrTestCaseHS {
, "response/docs==[{id:'5', x:5.5},{id:'4', x:5.5}]" , "response/docs==[{id:'5', x:5.5},{id:'4', x:5.5}]"
); );
doParamRefDslTest(client);
// test templating before parsing JSON // test templating before parsing JSON
client.testJQ( params("json","${OPENBRACE} query:'cat_s:A' ${CLOSEBRACE}", "json","${OPENBRACE} filter:'where_s:NY'${CLOSEBRACE}", "OPENBRACE","{", "CLOSEBRACE","}") client.testJQ( params("json","${OPENBRACE} query:'cat_s:A' ${CLOSEBRACE}", "json","${OPENBRACE} filter:'where_s:NY'${CLOSEBRACE}", "OPENBRACE","{", "CLOSEBRACE","}")
@ -407,6 +408,42 @@ public class TestJsonRequest extends SolrTestCaseHS {
} }
private static void doParamRefDslTest(Client client) throws Exception {
// referencing in dsl //nestedqp
client.testJQ( params("json","{query: {query: {param:'ref1'}}}", "ref1","{!field f=cat_s}A")
, "response/numFound==2"
);
// referencing json string param
client.testJQ( params("json", random().nextBoolean() ?
"{query:{query:{param:'ref1'}}}" // nestedqp
: "{query: {query: {query:{param:'ref1'}}}}", // nestedqp, v local param
"json",random().nextBoolean()
? "{params:{ref1:'{!field f=cat_s}A'}}" // string param
: "{queries:{ref1:{field:{f:cat_s,query:A}}}}" ) // qdsl
, "response/numFound==2"
);
{ // shortest top level ref
final ModifiableSolrParams params = params("json","{query:{param:'ref1'}}");
if (random().nextBoolean()) {
params.add("ref1","cat_s:A"); // either to plain string
} else {
params.add("json","{queries:{ref1:{field:{f:cat_s,query:A}}}}");// or to qdsl
}
client.testJQ( params, "response/numFound==2");
} // ref in bool must
client.testJQ( params("json","{query:{bool: {must:[{param:fq1},{param:fq2}]}}}",
"json","{params:{fq1:'cat_s:A', fq2:'where_s:NY'}}", "json.fields", "id")
, "response/docs==[{id:'1'}]"
);// referencing dsl&strings from filters objs&array
client.testJQ( params("json.filter","{param:fq1}","json.filter","{param:fq2}",
"json", random().nextBoolean() ?
"{queries:{fq1:{lucene:{query:'cat_s:A'}}, fq2:{lucene:{query:'where_s:NY'}}}}" :
"{params:{fq1:'cat_s:A', fq2:'where_s:NY'}}",
"json.fields", "id", "q", "*:*")
, "response/docs==[{id:'1'}]"
);
}
private static void testFilterCachingLocally(Client client) throws Exception { private static void testFilterCachingLocally(Client client) throws Exception {
if(client.getClientProvider()==null) { if(client.getClientProvider()==null) {
final SolrQueryRequest request = req(); final SolrQueryRequest request = req();