SOLR中实现自定义评分函数FunctionQuery

SOLR项目应用中&＃xff0c;需要对搜索结果自定义排序&＃xff0c;比如对一些质量好的网站权重高点。这里就需要定义一个自定义的valueSourceParser来实现。

先实现一个自定义的&＃xff1a;B258DynamicSourceParser。主要代码如下&＃xff1a;

public class B258DynamicSourceParser extends ValueSourceParser{public ValueSource parse(FunctionQParser fp) throws SyntaxError {String first &＃61; fp.parseArg();String second &＃61; fp.parseArg();ValueSource v1 &＃61; getValueSource(fp, first);return new B258DynamicSource(v1,second);}public ValueSource getValueSource(FunctionQParser fp, String arg) {if (arg &＃61;&＃61; null) return null;SchemaField f &＃61; fp.getReq().getSchema().getField(arg);if (f.getType().getClass() &＃61;&＃61; DateField.class) {throw new SolrException(SolrException.ErrorCode.BAD_REQUEST, "Can&＃39;t use ms() function on non-numeric legacy date field " &＃43; arg);}return f.getType().getValueSource(f, fp);}}

同时定义一个B258DynamicSource类。

public class B258DynamicSource extends ValueSource{public String str;public ValueSource field;public B258DynamicSource(ValueSource v1,String str){this.field &＃61; v1;this.str &＃61; str;}&＃64;Overridepublic FunctionValues getValues(Map context,AtomicReaderContext readerContext) throws IOException {final FunctionValues fieldVals &＃61; field.getValues(context, readerContext);return new FloatDocValues(this) {&＃64;Overridepublic float floatVal(int doc) {String domain &＃61; fieldVals.strVal(doc);//根据domain做不同的权重加权float f &＃61; 10f;return f;}&＃64;Overridepublic String toString(int doc) {return name() &＃43; &＃39;(&＃39; &＃43; fieldVals.strVal(doc) &＃43; &＃39;)&＃39;;}};}&＃64;Overridepublic boolean equals(Object o) {// TODO Auto-generated method stub//重要return true;}&＃64;Overridepublic int hashCode() {// TODO Auto-generated method stubreturn 0;}&＃64;Overridepublic String description() {return name();}public String name() {return "b258dynamic";}}

定义好了Funtion后&＃xff0c;需要在solrconfig.xml中定义该插件并引用。

在对应的solr.StandardRequestHandler节点中通过下面代码来引用。

258text0.8brb258dynamic(domain,dd)