solr 自定义函数 评分函数

lx86

solrpackage com.yujz;import java.io.IOException;import java.util.Map;import org.apache.lucene.index.AtomicReaderContext;import org.apache.lucene.queries.function.FunctionValues;import org.apache.lucene.queries.function.ValueSource;import org.apache.lucene.queries.function.docvalues.IntDocValues;importorg.apache.lucene.queries.function.valuesource.OrdFieldSource;public class B258DynamicSource extends ValueSource {public ValueSource[] valueSource;public ValueSource fieldSource;public B258DynamicSource(ValueSource[] valueSource, ValueSource fieldSource) {this.valueSource= valueSource;this.fieldSource = fieldSource;}@SuppressWarnings(&quot;rawtypes&quot;)@Overridepublic FunctionValues getValues(Map context,AtomicReaderContext readerContext) throws IOException {final FunctionValues fieldVals = fieldSource.getValues(context,readerContext);finalFunctionValues[] valsArr = new FunctionValues[valueSource.length];System.err.println(&quot;valueSource.length-------------&quot; &#43; valueSource.length);for (int i = 0; i < valueSource.length; i&#43;&#43;) {valsArr = valueSource.getValues(context, readerContext);System.err.println(&quot;valsArr&quot;);}returnnew IntDocValues(this) {@Overridepublic int intVal(int doc) {String source = fieldVals.strVal(doc);System.err.println(&quot;source------------------&quot; &#43; source);System.err.println(&quot;valsArr.length-----&quot; &#43; valsArr.length);System.err.println(&quot;doc----&quot; &#43; doc);for (FunctionValuesfv : valsArr) {int ss = fv.intVal(doc);System.err.println(&quot;args-----&quot; &#43; ss);if (doc > 7) {return 2;}}return 1;}@Overridepublic String toString(int doc) {return name() &#43; '(' &#43; fieldVals.strVal(doc) &#43; ')';}};}@Overridepublic boolean equals(Object o) {returno != null && o.getClass() == OrdFieldSource.class && this.fieldSource.equals(((B258DynamicSource)o).fieldSource);}@Overridepublic int hashCode() {return B258DynamicSource.class.hashCode() &#43; fieldSource.hashCode();}@Overridepublic String description() {returnname();}public String name() {return &quot;b258dynamic&quot;;}}package com.yujz;import java.util.List;import org.apache.lucene.queries.function.ValueSource;import org.apache.solr.common.SolrException;import org.apache.solr.schema.IntField;import org.apache.solr.schema.SchemaField;importorg.apache.solr.search.FunctionQParser;import org.apache.solr.search.SyntaxError;import org.apache.solr.search.ValueSourceParser;public class B258DynamicSourceParser extends ValueSourceParser{public ValueSource parse(FunctionQParser fp) throws SyntaxError{String field = fp.parseArg();ValueSource v1 = getValueSource(fp, field); List sources = fp.parseValueSourceList();return new B258DynamicSource(sources.toArray(new ValueSource[sources.size()]),v1);}public ValueSource getValueSource(FunctionQParser fp, Stringfield) { if (field == null) return null; SchemaField f = fp.getReq().getSchema().getField(field); if (f.getType().getClass() == IntField.class) { throw new SolrException(SolrException.ErrorCode.BAD_REQUEST, &quot;Can't use ms() function on non-numeric legacy datefield &quot; &#43; field); } return f.getType().getValueSource(f, fp); }}         版权声明：本文为博主原创文章，未经博主允许不得转载。