Can you explain why you want to implement a different sort first? There
may be other ways of achieving the same thing.
Upayavira
On Sun, Jan 6, 2013, at 01:32 AM, andy wrote:
> Hi,
>
> Maybe this is an old thread or maybe it's different with previous one.
>
> I want to custom solr sort and pass solr param from client to solr
> server,
> so I implemented SearchComponent which named MySortComponent in my code,
> and also implemented FieldComparatorSource and FieldComparator. when I
> use
> "mysearch" requesthandler(see following codes), I found that custom sort
> just effect on the current page when I got multiple page results, but the
> sort is expected when I sets the rows which contains all the results.
> Does
> anybody know how to solve it or the reason?
>
> code snippet:
>
> public class MySortComponent extends SearchComponent implements
> SolrCoreAware {
>
> public void inform(SolrCore arg0) {
> }
>
> @Override
> public void prepare(ResponseBuilder rb) throws IOException {
> SolrParams params = rb.req.getParams();
> String uid = params.get("uid")
> private RestTemplate restTemplate = new RestTemplate();
>
> MyComparatorSource comparator = new MyComparatorSource(uid);
> SortSpec sortSpec = rb.getSortSpec();
> if (sortSpec.getSort() == null) {
> sortSpec.setSort(new Sort(new SortField[] {
> new SortField("relation",
> comparator),SortField.FIELD_SCORE }));
>
> } else {
>
> SortField[] current = sortSpec.getSort().getSort();
> ArrayList<SortField> sorts = new ArrayList<SortField>(
> current.length + 1);
> sorts.add(new SortField("relation", comparator));
> for (SortField sf : current) {
> sorts.add(sf);
> }
> sortSpec.setSort(new Sort(sorts.toArray(new
> SortField[sorts.size()])));
>
> }
>
> }
>
> @Override
> public void process(ResponseBuilder rb) throws IOException {
>
> }
>
> //
> ---------------------------------------------------------------------------------
> // SolrInfoMBean
> //
> ---------------------------------------------------------------------------------
>
> @Override
> public String getDescription() {
> return "Custom Sorting";
> }
>
> @Override
> public String getSource() {
> return "";
> }
>
> @Override
> public URL[] getDocs() {
> try {
> return new URL[] { new URL(
> "http://wiki.apache.org/solr/QueryComponent";) };
> } catch (MalformedURLException e) {
> throw new RuntimeException(e);
> }
> }
>
> public class MyComparatorSource extends FieldComparatorSource {
> private BitSet dg1;
> private BitSet dg2;
> private BitSet dg3;
>
> public MyComparatorSource(String uid) throws IOException {
>
> SearchResponse responseBody = restTemplate.postForObject(
> "http://search.test.com/userid/search/"; + uid, null,
> SearchResponse.class);
>
> String d1 = responseBody.getOneDe();
> String d2 = responseBody.getTwoDe();
> String d3 = responseBody.getThreeDe();
>
> if (StringUtils.hasLength(d1)) {
> byte[] bytes = Base64.decodeBase64(d1);
> dg1 = BitSetHelper.loadFromBzip2ByteArray(bytes);
> }
>
> if (StringUtils.hasLength(d2)) {
> byte[] bytes = Base64.decodeBase64(d2);
> dg2 = BitSetHelper.loadFromBzip2ByteArray(bytes);
> }
>
> if (StringUtils.hasLength(d3)) {
> byte[] bytes = Base64.decodeBase64(d3);
> dg3 = BitSetHelper.loadFromBzip2ByteArray(bytes);
> }
>
> }
>
> @Override
> public FieldComparator newComparator(String fieldname,
> final int numHits, int sortPos, boolean reversed)
> throws IOException {
> return new RelationComparator(fieldname, numHits);
> }
>
> class RelationComparator extends FieldComparator {
> private int[] uidDoc;
> private float[] values;
> private float bottom;
> String fieldName;
>
> public RelationComparator(String fieldName, int numHits)
> throws IOException {
> values = new float[numHits];
> this.fieldName = fieldName;
> }
>
> @Override
> public int compare(int slot1, int slot2) {
> if (values[slot1] > values[slot2])
> return -1;
> if (values[slot1] < values[slot2])
> return 1;
> return 0;
> }
>
> @Override
> public int compareBottom(int doc) throws IOException {
> float docDistance = getRelation(doc);
> if (bottom < docDistance)
> return -1;
> if (bottom > docDistance)
> return 1;
> return 0;
> }
>
> @Override
> public void copy(int slot, int doc) throws IOException {
> values[slot] = getRelation(doc);
>
> }
>
> @Override
> public void setBottom(int slot) {
> bottom = values[slot];
> }
>
> @Override
> public FieldComparator<Integer> setNextReader(
> AtomicReaderContext ctx) throws IOException {
> uidDoc = FieldCache.DEFAULT.getInts(ctx.reader(),
> "userID",
> true);
> return this;
> }
>
> @Override
> public Float value(int slot) {
> return new Float(values[slot]);
> }
>
> private float getRelation(int doc) throws IOException {
> if (dg3.get(uidDoc[doc])) {
> return 3.0f;
> } else if (dg2.get(uidDoc[doc])) {
> return 4.0f;
> } else if (dg1.get(uidDoc[doc])) {
> return 5.0f;
> } else {
> return 1.0f;
> }
> }
>
> @Override
> public int compareDocToValue(int arg0, Object arg1)
> throws IOException {
> // TODO Auto-generated method stub
> return 0;
> }
> }
>
> }
> }
>
>
> and solrconfig.xml configuration is
>
> <searchComponent name="mySortComponent"
> class="com.tianji.solr.handler.component.MySortComponent"/>
>
> <requestHandler name="/mysearch" class="solr.SearchHandler">
> <arr name="last-components">
> <str>mySortComponent</str>
> </arr>
> </requestHandler>
>
>
> Thanks
> Andy
>
>
>
>
> --
> View this message in context:
> http://lucene.472066.n3.nabble.com/custom-solr-sort-tp4031014.html
> Sent from the Solr - User mailing list archive at Nabble.com.
