mirror of https://github.com/apache/lucene.git
SOLR-264: Added RandomSortField a random order SortComparatorSource. Thanks yonik for the performance hints
git-svn-id: https://svn.apache.org/repos/asf/lucene/solr/trunk@548594 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
parent
1981c7ab29
commit
59ed1b5bf1
|
@ -65,6 +65,9 @@ New Features
|
||||||
requests depending on the request attributes. This class will likely
|
requests depending on the request attributes. This class will likely
|
||||||
replace XmlUpdateRequestHandler. (Thorsten Scherler, ryan)
|
replace XmlUpdateRequestHandler. (Thorsten Scherler, ryan)
|
||||||
|
|
||||||
|
9. SOLR-264: Added RandomSortField, a utility field with a random sort order.
|
||||||
|
This field type should only be used for sorting. (ryan)
|
||||||
|
|
||||||
Changes in runtime behavior
|
Changes in runtime behavior
|
||||||
|
|
||||||
Optimizations
|
Optimizations
|
||||||
|
|
|
@ -0,0 +1,108 @@
|
||||||
|
/**
|
||||||
|
* Licensed to the Apache Software Foundation (ASF) under one or more
|
||||||
|
* contributor license agreements. See the NOTICE file distributed with
|
||||||
|
* this work for additional information regarding copyright ownership.
|
||||||
|
* The ASF licenses this file to You under the Apache License, Version 2.0
|
||||||
|
* (the "License"); you may not use this file except in compliance with
|
||||||
|
* the License. You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package org.apache.solr.schema;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.util.Random;
|
||||||
|
|
||||||
|
import org.apache.lucene.document.Fieldable;
|
||||||
|
import org.apache.lucene.index.IndexReader;
|
||||||
|
import org.apache.lucene.search.ScoreDoc;
|
||||||
|
import org.apache.lucene.search.ScoreDocComparator;
|
||||||
|
import org.apache.lucene.search.SortComparatorSource;
|
||||||
|
import org.apache.lucene.search.SortField;
|
||||||
|
import org.apache.solr.request.TextResponseWriter;
|
||||||
|
import org.apache.solr.request.XMLWriter;
|
||||||
|
import org.apache.solr.search.function.ValueSource;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Utility Field used for random sorting. It should not be passed a value.
|
||||||
|
*
|
||||||
|
* To enable random sorting, you will need to add something like this
|
||||||
|
* to the schema.xml
|
||||||
|
*
|
||||||
|
* <types>
|
||||||
|
* ...
|
||||||
|
* <fieldType name="random" class="solr.RandomSortField" />
|
||||||
|
* ...
|
||||||
|
* </types>
|
||||||
|
* <fields>
|
||||||
|
* ...
|
||||||
|
* <field name="random" type="random" indexed="true" stored="false"/>
|
||||||
|
* ...
|
||||||
|
* </fields>
|
||||||
|
*
|
||||||
|
* @author ryan
|
||||||
|
* @version $Id$
|
||||||
|
* @since solr 1.3
|
||||||
|
*/
|
||||||
|
public class RandomSortField extends FieldType
|
||||||
|
{
|
||||||
|
/** Special comparator for sorting hits in random order */
|
||||||
|
private static final ScoreDocComparator COMPARE = new ScoreDocComparator() {
|
||||||
|
final Random rand = new Random();
|
||||||
|
|
||||||
|
public int compare (ScoreDoc i, ScoreDoc j) {
|
||||||
|
return (rand.nextBoolean()) ? 1 : -1; //rand.nextInt() >>> 31; ??
|
||||||
|
}
|
||||||
|
public Comparable sortValue (ScoreDoc i) {
|
||||||
|
return new Float(rand.nextFloat());
|
||||||
|
}
|
||||||
|
public int sortType() {
|
||||||
|
return SortField.CUSTOM;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
/** use random sorting order. */
|
||||||
|
private static class RandomSort extends SortField {
|
||||||
|
public RandomSort( String n )
|
||||||
|
{
|
||||||
|
super( n, SortField.CUSTOM );
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public SortComparatorSource getFactory() {
|
||||||
|
return new SortComparatorSource() {
|
||||||
|
public ScoreDocComparator newComparator(IndexReader reader, String fieldname) throws IOException {
|
||||||
|
return COMPARE;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public SortField getSortField(SchemaField field,boolean reverse) {
|
||||||
|
return new RandomSort(field.getName());
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public ValueSource getValueSource(SchemaField field) {
|
||||||
|
throw new UnsupportedOperationException("Random field does not have a value source");
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void write(XMLWriter xmlWriter, String name, Fieldable f) {}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void write(TextResponseWriter writer, String name, Fieldable f) {}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue