org.apache.solr.search.FunctionQParser Java Examples

private MultiValueSource parseSfield(FunctionQParser fp) throws SyntaxError {
  String sfield = fp.getParam(SpatialParams.FIELD);
  if (sfield == null) return null;
  SchemaField sf = fp.getReq().getSchema().getField(sfield);
  FieldType type = sf.getType();
  if (type instanceof AbstractSpatialFieldType) {
    @SuppressWarnings({"rawtypes"})
    AbstractSpatialFieldType asft = (AbstractSpatialFieldType) type;
    return new SpatialStrategyMultiValueSource(asft.getStrategy(sfield), asft.getDistanceUnits());
  }
  ValueSource vs = type.getValueSource(sf, fp);
  if (vs instanceof MultiValueSource) {
    return (MultiValueSource)vs;
  }
  throw new SyntaxError("Spatial field must implement MultiValueSource or extend AbstractSpatialFieldType:" + sf);
}
 

@Override
public ValueSource parse(FunctionQParser fp) throws SyntaxError {
  List<Double> percentiles = new ArrayList<>();
  ValueSource vs = fp.parseValueSource();
  while (fp.hasMoreArguments()) {
    double val = fp.parseDouble();
    if (val<0 || val>100) {
      throw new SyntaxError("requested percentile must be between 0 and 100.  got " + val);
    }
    percentiles.add(val);
  }

  if (percentiles.isEmpty()) {
    throw new SyntaxError("expected percentile(valsource,percent1[,percent2]*)  EXAMPLE:percentile(myfield,50)");
  }

  return new PercentileAgg(vs, percentiles);
}
 

@Override
public ValueSource parse(FunctionQParser fp) throws SyntaxError {
  ValueSource source = fp.parseValueSource();
  final float nvl = fp.parseFloat();

  return new SimpleFloatFunction(source) {
    @Override
  protected String name() {
      return "nvl";
    }

    @Override
    protected float func(int doc, FunctionValues vals) throws IOException {
      float v = vals.floatVal(doc);
      if (v == nvlFloatValue) {
        return nvl;
      } else {
        return v;
      }
    }
  };
}
 

@Override
public ValueSource parse(FunctionQParser fp) throws SyntaxError {
    String field=fp.parseArg();                          // eg. cl_hi
    String featureString = fp.parseArg();
    // System.out.println(featureString);
    byte[] hist= Base64.decodeBase64(featureString);     // eg. FQY5DhMYDg0ODg0PEBEPDg4ODg8QEgsgEBAQEBAgEBAQEBA=
    double maxDistance = Double.MAX_VALUE;
    if (fp.hasMoreArguments()) {                           // if there is a third argument, it's the max value to return if there is none. Note the query cache is not updated upon parameter change.
        maxDistance = Double.parseDouble(fp.parseArg());
    }
    return new LireValueSource(field, hist, maxDistance);
}
 

@SuppressWarnings({ "rawtypes" })
private FunctionValues functionValues(NamedList initArgs, String arg) throws Exception {
  FunctionQParser fqp = mockFunctionQParser(arg);
  XJoinValueSourceParser vsp = new XJoinValueSourceParser();
  vsp.init(initArgs);
  ValueSource vs = vsp.parse(fqp);
  return vs.getValues(null, searcher.getAtomicReader().getContext());
}
 

/**
 * Provide a ValueSource for external process results, which are obtained from the
 * request context (having been placed there by XJoinSearchComponent).
 */
@Override
public ValueSource parse(FunctionQParser fqp) throws SyntaxError {
  String componentName = this.componentName != null ? this.componentName : fqp.parseArg();
  String attribute = this.attribute != null ? this.attribute : fqp.parseArg();
  
  XJoinSearchComponent xJoin = (XJoinSearchComponent)fqp.getReq().getCore().getSearchComponent(componentName);
  String joinField = xJoin.getJoinField();
  XJoinResults<?> results = (XJoinResults<?>)fqp.getReq().getContext().get(xJoin.getResultsTag());
  if (results == null) {
    throw new RuntimeException("No xjoin results in request context");
  }
  return new XJoinValueSource(joinField, results, attribute);
}
 

@SuppressWarnings({ "rawtypes" })
private FunctionValues functionValues(NamedList initArgs, String arg) throws Exception {
  FunctionQParser fqp = mockFunctionQParser(arg);
  XJoinValueSourceParser vsp = new XJoinValueSourceParser();
  vsp.init(initArgs);
  ValueSource vs = vsp.parse(fqp);
  return vs.getValues(null, searcher.getLeafReader().getContext());
}
 

/**
 * Provide a ValueSource for external process results, which are obtained from the
 * request context (having been placed there by XJoinSearchComponent).
 */
@Override
public ValueSource parse(FunctionQParser fqp) throws SyntaxError {
  String componentName = this.componentName != null ? this.componentName : fqp.parseArg();
  String attribute = this.attribute != null ? this.attribute : fqp.parseArg();
  
  XJoinSearchComponent xJoin = (XJoinSearchComponent)fqp.getReq().getCore().getSearchComponent(componentName);
  String joinField = xJoin.getJoinField();
  XJoinResults<?> results = (XJoinResults<?>)fqp.getReq().getContext().get(xJoin.getResultsTag());
  if (results == null) {
    throw new RuntimeException("No xjoin results in request context");
  }
  return new XJoinValueSource(joinField, results, attribute);
}
 

@SuppressWarnings({ "rawtypes" })
private FunctionValues functionValues(NamedList initArgs, String arg) throws Exception {
  FunctionQParser fqp = mockFunctionQParser(arg);
  XJoinValueSourceParser vsp = new XJoinValueSourceParser();
  vsp.init(initArgs);
  ValueSource vs = vsp.parse(fqp);
  return vs.getValues(null, searcher.getSlowAtomicReader().getContext());
}
 

/**
 * Provide a ValueSource for external process results, which are obtained from the
 * request context (having been placed there by XJoinSearchComponent).
 */
@Override
public ValueSource parse(FunctionQParser fqp) throws SyntaxError {
  String componentName = this.componentName != null ? this.componentName : fqp.parseArg();
  String attribute = this.attribute != null ? this.attribute : fqp.parseArg();
  
  XJoinSearchComponent xJoin = (XJoinSearchComponent)fqp.getReq().getCore().getSearchComponent(componentName);
  String joinField = xJoin.getJoinField();
  XJoinResults<?> results = (XJoinResults<?>)fqp.getReq().getContext().get(xJoin.getResultsTag());
  if (results == null) {
    throw new RuntimeException("No xjoin results in request context");
  }
  return new XJoinValueSource(joinField, results, attribute);
}
 

@Override
public ValueSource parse(FunctionQParser fp) throws SyntaxError {
  String key = fp.parseArg();
  double val = fp.parseDouble();
  
  AtomicInteger counter = new AtomicInteger();
  if (null != counters.putIfAbsent(key, counter)) {
    throw new IllegalArgumentException("Key has already been used: " + key);
  } 
  return new CountDocsValueSource(counter, val);
}
 

@Override
public ValueSource parse(FunctionQParser fp) throws SyntaxError {
  parses.incrementAndGet();
  final String what = fp.hasMoreArguments() ? fp.parseId() : "wrap";

  switch (what) {
    case "wrap": return new DebugAgg(fp);
    case "numShards": return new DebugAggNumShards();
    default: /* No-Op */
  }
  throw new RuntimeException("No idea what to do with " + what);
}
 

@Override
public ValueSource parse(FunctionQParser fp) throws SyntaxError {
  
  final String sortFieldName = fp.parseArg();
  final Query query;
  if (fp.hasMoreArguments()){
    query = fp.parseNestedQuery();
  } else {
    query = fp.subQuery(fp.getParam(CommonParams.Q), null).getQuery();
  }
  
  BitSetProducer parentFilter;
  BitSetProducer childFilter;
  SchemaField sf;
  try {
    AllParentsAware bjQ;
    if (!(query instanceof AllParentsAware)) {
      throw new SyntaxError("expect a reference to block join query "+
            AllParentsAware.class.getSimpleName()+" in "+fp.getString());
    }
    bjQ = (AllParentsAware) query;
    
    parentFilter = BlockJoinParentQParser.getCachedBitSetProducer(fp.getReq(), bjQ.getParentQuery());
    childFilter = BlockJoinParentQParser.getCachedBitSetProducer(fp.getReq(), bjQ.getChildQuery());

    if (sortFieldName==null || sortFieldName.equals("")) {
      throw new SyntaxError ("field is omitted in "+fp.getString());
    }
    
    sf = fp.getReq().getSchema().getFieldOrNull(sortFieldName);
    if (null == sf) {
      throw new SyntaxError
        (NAME+" sort param field \""+ sortFieldName+"\" can't be found in schema");
    }
  } catch (SyntaxError e) {
    log.error("can't parse {}", fp.getString(), e);
    throw e;
  }
  return new BlockJoinSortFieldValueSource(childFilter, parentFilter, sf);
}
 

/** Parses simple strings like "avg(x)" in the context of optional local params (may be null) */
private AggValueSource parseStatWithParams(String key, SolrParams localparams, String stat) throws SyntaxError {
  SolrQueryRequest req = getSolrRequest();
  FunctionQParser parser = new FunctionQParser(stat, localparams, req.getParams(), req);
  AggValueSource agg = parser.parseAgg(FunctionQParser.FLAG_DEFAULT);
  return agg;
}
 

private MultiValueSource parsePoint(FunctionQParser fp) throws SyntaxError {
  String ptStr = fp.getParam(SpatialParams.POINT);
  if (ptStr == null) return null;
  Point point = SpatialUtils.parsePointSolrException(ptStr, SpatialContext.GEO);
  //assume Lat Lon order
  return new VectorValueSource(
      Arrays.<ValueSource>asList(new DoubleConstValueSource(point.getY()), new DoubleConstValueSource(point.getX())));
}
 

@Override
public ValueSource parse(FunctionQParser fq) throws SyntaxError {
    return new FunctionValueSource(maxYears,money_maxTimes,money_base,fq.parseValueSourceList());
}
 

private FunctionQParser mockFunctionQParser(String arg) throws Exception {
  FunctionQParser fqp = mock(FunctionQParser.class);
  when(fqp.getReq()).thenReturn(sqr);    
  when(fqp.parseArg()).thenReturn(arg);
  return fqp;
}
 

public DebugAgg(FunctionQParser fp) throws SyntaxError { 
  super("debug");
  this.localParams = fp.getLocalParams();
  this.inner = fp.hasMoreArguments() ? fp.parseAgg(FunctionQParser.FLAG_IS_AGG) : new CountAgg();
}
 

private FunctionQParser mockFunctionQParser(String arg) throws Exception {
  FunctionQParser fqp = mock(FunctionQParser.class);
  when(fqp.getReq()).thenReturn(sqr);    
  when(fqp.parseArg()).thenReturn(arg);
  return fqp;
}
 

private FunctionQParser mockFunctionQParser(String arg) throws Exception {
  FunctionQParser fqp = mock(FunctionQParser.class);
  when(fqp.getReq()).thenReturn(sqr);    
  when(fqp.parseArg()).thenReturn(arg);
  return fqp;
}