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

鸬鹚洲

　　SOLR项目应用中，需要对搜索结果自定义排序，比如对一些质量好的网站权重高点。这里就需要定义一个自定义的valueSourceParser来实现。
　　先实现一个自定义的：B258DynamicSourceParser。主要代码如下：

public class B258DynamicSourceParser extends ValueSourceParser{
public ValueSource parse(FunctionQParser fp) throws SyntaxError {
String first = fp.parseArg();
String second = fp.parseArg();
ValueSource v1 = getValueSource(fp, first);
return new B258DynamicSource(v1,second);
}
public ValueSource getValueSource(FunctionQParser fp, String arg) {
if (arg == null) return null;
SchemaField f = fp.getReq().getSchema().getField(arg);
if (f.getType().getClass() == DateField.class) {
throw new SolrException(SolrException.ErrorCode.BAD_REQUEST, "Can't use ms() function on non-numeric legacy date field " + 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 = v1;
this.str = str;
}
@Override
public FunctionValues getValues(Map context,
AtomicReaderContext readerContext) throws IOException {
final FunctionValues fieldVals =  field.getValues(context, readerContext);
return new FloatDocValues(this) {
@Override
public float floatVal(int doc) {
String domain = fieldVals.strVal(doc);//根据domain做不同的权重加权
float f = 10f;
return f;
}
@Override
public String toString(int doc) {
return name() + '(' + fieldVals.strVal(doc) + ')';
}
};
}
@Override
public boolean equals(Object o) {
// TODO Auto-generated method stub//重要
return true;
}
@Override
public int hashCode() {
// TODO Auto-generated method stub
return 0;
}
@Override
public String description() {
return name();
}
public String name() {
return "b258dynamic";
}
}
　　定义好了Funtion后，需要在solrconfig.xml中定义该插件并引用。

<valueSourceParser name=&quot;b258dynamic&quot;
class=&quot;org.lifeba.lucene.queries.function.B258DynamicSourceParser&quot; />
　　在对应的solr.StandardRequestHandler节点中通过下面代码来引用。

<lst name=&quot;defaults&quot;>
<str name=&quot;defType&quot;>258</str>
<str name=&quot;qf&quot;>text</str>
<str name=&quot;tie&quot;>0.8</str>
<str name=&quot;boost&quot;>br</str>
<str name=&quot;dynamic&quot;>b258dynamic(domain,dd)</str>
</lst>

         
版权声明：本文为博主原创文章，未经博主允许不得转载。