如何实现Solr自定义评分查询

（一）背景介绍

大多数时候我们使用lucene/solr/elasticsearch自带的评分查询都是没问题的，当然这也仅仅限于简单的业务或者对搜索排名
不敏感的场景中，假设业务方要求有若干业务因子要干扰到排名，同时还不能放弃框架本身的文本相似度评分，那么应该怎么做呢？
这种场景尤其是在电商类的一些垂直搜索中体现比较明显，比如，新商品加分，口碑好的加分，图片清晰的加分，商品介绍详细的加分，大促的商品加分等等等等，那么如何把众多的业务因素加到的总的评分里面?

（二）实现策略

（1）在索引的时候把众多的业务评分计算提前计算好，存储成一个字段，然后查询的时候根据这个字段排名。
评价：比较简单暴力，适合加权固定，不经常改变评分因素的业务使用，查询性能最高  ，更新，改变，调试评分比较麻烦

（2）在索引的时候把众多的业务评分因子都索引成一个字段，在查询的时候动态获取各个字段评分计算后，加入总的评分从而影响最终的排名。
评价：对动态更改评分支持比较好，查询性能稍差

（3）对于业务的评分因素，动静分离，静态评分因子长期不变的，就全部计算完存储成一个字段，动态的也存储若干字段，最终的评分由静态业务评分+动态业务评分+相似度评分综合得出，从而影响最终的排名。
评价：合理规划评分因素，动静分离，算是业务与技术的一个折中


有关lucene的自定义评分组件，可以参考以前的文章：
http://qindongliang.iteye.com/blog/2008672

下面来简述下如何在solr中，实现开发自定义评分组件，solr基于lucene，总体来说
与lucene大同小异，需要自己开发几个包装的类即可，在elasticsearch中也是如此，后面有机会，再探讨在elasticsearch中自定义评分的实现

（三）实现方式

（A）通过重写QueryParser实现

1，继承CustomScoreProvider类，重写customScore方法，从DocValues中动态读取评分有关因子，计算后，影响总评分
2，继承CustomScoreQuery类，重写getCustomScoreProvider方法，需要用到1，返回自定义的CustomScoreProvider类
到此，在lucene中就完事了，但是在solr中我们还需要继续
3，继承QParser类，重写parse方法，需要用到2，并在构造方法中，完成一些必须的初始化操作
4，继承QParserPlugin类，重写createParser方法，需要用到3，至此，代码完成
打包项目成一个jar，拷贝至server\solr-webapp\webapp\WEB-INF\lib中
5，在solrconfig.xml中，注册我们写的插件：

<queryParser name="myqp" class="com.easy.custom.queryparser.MyQueryParserPlugin">
  <lst name="words">
       <str name="word">easy_money</str>
       <str name="word">easy_count</str>
       <str name="word">easy_test</str>
     </lst>
</queryParser>

6， 重启solr，或者reload指定的core
7，打开solr的ui页面，指定defType，测试搜索，如果log不报错，就证明使用成功了





（B）通过Function Query实现

函数查询一项强大的功能，solr已经自带了
许多的函数，比如数学函数，聚合函数等，详细资料可看官网的wiki资料
https://cwiki.apache.org/confluence/display/solr/Function+Queries

1，继承ValueSource类，重写getValues方法，并在返回的方法中，完成评分计算逻辑
2，继承ValueSourceParser类，并重写parser方法，返回1定义的类，
建议在parser方法里面，获取ValueSource然后传入自定义的ValueSource类里面复用，
不建议直接从DocValues里面读取，因为基于这个IndexSearch的打开的ValueSource耗费资源更少。至此，代码完成打包项目成一个jar，拷贝至server\solr-webapp\webapp\WEB-INF\lib中
3，在solrconfig.xml中，注册我们的组件：

//此处，也可也定义需要传入的参数
<valueSourceParser name="myfunc" class="com.easy.custom.function.MyValueParser"  />   

4，打开solr的ui页面进行查询，不报错的话，即查询成功，可以看到和我们第一种方式的结果是一致的



（四）总结

其实核心功能还是使用lucene实现的，solr/es则是在lucene的基础上提供了强大灵活的插件机制，这样以来，我们就能更容易实现一些我们特殊需求的定制化。

全部代码，已经放在github上：
https://github.com/qindongliang/solr-custom-score




有什么问题可以扫码关注微信公众号：我是攻城师(woshigcs)，在后台留言咨询。
技术债不能欠，健康债更不能欠， 求道之路，我们同行。

评论

2 楼 qindongliang1922 2016-09-26  

chucklee 写道

defType使用edismax，通过boost查询，和文中的方式有什么区别呢，没太理解文中自己要实现的部分具体是什么内容。
个人认为：计算出来各个业务因子的得分之后，存储在各个字段中，通过solr自己提供的函数写一个公式，查询的时候指定公式给boost是不是就可以了呢

如果得分，是查询时候动态生成的，你怎么做

1 楼 chucklee 2016-09-26  

defType使用edismax，通过boost查询，和文中的方式有什么区别呢，没太理解文中自己要实现的部分具体是什么内容。
个人认为：计算出来各个业务因子的得分之后，存储在各个字段中，通过solr自己提供的函数写一个公式，查询的时候指定公式给boost是不是就可以了呢