Apache
/
lucene
mirror of https://github.com/apache/lucene.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Allow open-ended ranges in Intervals range (#13859) 

Currently IntervalsSource.range function closed intervals.
This will allow open-ended ranges

Relates to #13562
This commit is contained in:
Mayya Sharipova 2024-10-07 13:48:12 -04:00 committed by GitHub
parent 0fb7f9f89d
commit 970672ceab
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 50 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
lucene/queries/src/java/org/apache/lucene/queries/intervals/Intervals.java
View File

@ -247,8 +247,10 @@ public final class Intervals {
* Return an {@link IntervalsSource} over the disjunction of all terms that fall within the given
* range
*
* @param lowerTerm The term text at the lower end of the range
* @param upperTerm The term text at the upper end of the range
* @param lowerTerm The term text at the lower end of the range; can be {@code null} to indicate
* an open-ended range at this end
* @param upperTerm The term text at the upper end of the range; can be {@code null} to indicate
* an open-ended range at this end
* @param includeLower If true, the <code>lowerTerm</code> is included in the range
* @param includeUpper If true, the <code>upperTerm</code> is included in the range
* @throws IllegalStateException if the range expands to more than {@link #DEFAULT_MAX_EXPANSIONS}
@ -266,8 +268,10 @@ public final class Intervals {
* <p>WARNING: Setting {@code maxExpansions} to higher than the default value of {@link
* #DEFAULT_MAX_EXPANSIONS} can be both slow and memory-intensive
*
* @param lowerTerm The term text at the lower end of the range
* @param upperTerm The term text at the upper end of the range
* @param lowerTerm The term text at the lower end of the range; can be {@code null} to indicate
* an open-ended range at this end
* @param upperTerm The term text at the upper end of the range; can be {@code null} to indicate
* an open-ended range at this end
* @param includeLower If true, the <code>lowerTerm</code> is included in the range
* @param includeUpper If true, the <code>upperTerm</code> is included in the range
* @param maxExpansions the maximum number of terms to expand to
@ -286,9 +290,9 @@ public final class Intervals {
StringBuilder buffer = new StringBuilder();
buffer.append("{");
buffer.append(lowerTerm.utf8ToString());
buffer.append(lowerTerm == null ? "* " : lowerTerm.utf8ToString());
buffer.append(",");
buffer.append(upperTerm.utf8ToString());
buffer.append(upperTerm == null ? "*" : upperTerm.utf8ToString());
buffer.append("}");
return new MultiTermIntervalsSource(ca, maxExpansions, buffer.toString());
}

40
lucene/queries/src/test/org/apache/lucene/queries/intervals/TestIntervals.java
View File

@ -1138,6 +1138,46 @@ public class TestIntervals extends LuceneTestCase {
checkVisits(source, 1);
}
public void testOpenEndedRange() throws IOException {
{
IntervalsSource source = Intervals.range(new BytesRef("porridge"), null, false, false);
checkIntervals(
source,
"field1",
5,
new int[][] {
{3, 3},
{9, 9, 10, 10, 14, 14, 18, 18, 22, 22, 26, 26, 27, 27},
{9, 9, 10, 10, 11, 11, 14, 14, 18, 18, 22, 22, 26, 26},
{8, 8},
{9, 9, 10, 10, 12, 12, 14, 14, 18, 18, 21, 21},
{}
});
MatchesIterator mi = getMatches(source, 3, "field1");
assertNotNull(mi);
assertMatch(mi, 8, 8, 37, 41);
}
{
IntervalsSource source = Intervals.range(null, new BytesRef("anyone"), false, true);
checkIntervals(
source,
"field1",
1,
new int[][] {
{4, 4},
{},
{},
{},
{},
{}
});
MatchesIterator mi = getMatches(source, 0, "field1");
assertNotNull(mi);
assertMatch(mi, 4, 4, 23, 29);
}
}
public void testWrappedFilters() throws IOException {
IntervalsSource source =
Intervals.or(