mirror of https://github.com/apache/lucene.git
forgot to add
git-svn-id: https://svn.apache.org/repos/asf/lucene/dev/branches/lucene4547@1436547 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
parent
24d701180e
commit
f3736807bb
|
@ -0,0 +1,136 @@
|
||||||
|
package org.apache.lucene.index;
|
||||||
|
|
||||||
|
/*
|
||||||
|
* 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.
|
||||||
|
*/
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.util.Comparator;
|
||||||
|
|
||||||
|
import org.apache.lucene.util.Bits;
|
||||||
|
import org.apache.lucene.util.BytesRef;
|
||||||
|
|
||||||
|
/** Implements a {@link TermsEnum} wrapping a provided
|
||||||
|
* {@link SortedDocValues}. */
|
||||||
|
|
||||||
|
public class SortedDocValuesTermsEnum extends TermsEnum {
|
||||||
|
private final SortedDocValues values;
|
||||||
|
private int currentOrd = -1;
|
||||||
|
private final BytesRef term = new BytesRef();
|
||||||
|
|
||||||
|
public SortedDocValuesTermsEnum(SortedDocValues values) {
|
||||||
|
this.values = values;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public SeekStatus seekCeil(BytesRef text, boolean useCache /* ignored */) throws IOException {
|
||||||
|
int ord = values.lookupTerm(text, term);
|
||||||
|
// nocommit >= 0!?
|
||||||
|
if (ord >= 0) {
|
||||||
|
currentOrd = ord;
|
||||||
|
term.offset = 0;
|
||||||
|
term.copyBytes(text);
|
||||||
|
return SeekStatus.FOUND;
|
||||||
|
} else {
|
||||||
|
currentOrd = -ord-1;
|
||||||
|
if (currentOrd == values.getValueCount()) {
|
||||||
|
return SeekStatus.END;
|
||||||
|
} else {
|
||||||
|
// nocommit hmm can we avoid this "extra" lookup?:
|
||||||
|
values.lookupOrd(currentOrd, term);
|
||||||
|
return SeekStatus.NOT_FOUND;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean seekExact(BytesRef text, boolean useCache) throws IOException {
|
||||||
|
int ord = values.lookupTerm(text, term);
|
||||||
|
if (ord >= 0) {
|
||||||
|
term.copyBytes(text);
|
||||||
|
currentOrd = ord;
|
||||||
|
return true;
|
||||||
|
} else {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void seekExact(long ord) throws IOException {
|
||||||
|
assert ord >= 0 && ord < values.getValueCount();
|
||||||
|
currentOrd = (int) ord;
|
||||||
|
values.lookupOrd(currentOrd, term);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public BytesRef next() throws IOException {
|
||||||
|
currentOrd++;
|
||||||
|
if (currentOrd >= values.getValueCount()) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
values.lookupOrd(currentOrd, term);
|
||||||
|
return term;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public BytesRef term() throws IOException {
|
||||||
|
return term;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public long ord() throws IOException {
|
||||||
|
return currentOrd;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public int docFreq() {
|
||||||
|
throw new UnsupportedOperationException();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public long totalTermFreq() {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public DocsEnum docs(Bits liveDocs, DocsEnum reuse, int flags) throws IOException {
|
||||||
|
throw new UnsupportedOperationException();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public DocsAndPositionsEnum docsAndPositions(Bits liveDocs, DocsAndPositionsEnum reuse, int flags) throws IOException {
|
||||||
|
throw new UnsupportedOperationException();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Comparator<BytesRef> getComparator() {
|
||||||
|
return BytesRef.getUTF8SortedAsUnicodeComparator();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void seekExact(BytesRef term, TermState state) throws IOException {
|
||||||
|
assert state != null && state instanceof OrdTermState;
|
||||||
|
this.seekExact(((OrdTermState)state).ord);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public TermState termState() throws IOException {
|
||||||
|
OrdTermState state = new OrdTermState();
|
||||||
|
state.ord = currentOrd;
|
||||||
|
return state;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in New Issue