mindol1004/spring-data-mongodb
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Polishing.

Browse Source 
See #4139
Original pull request: #4182.
This commit is contained in:
Christoph Strobl
2022-09-22 13:03:42 +02:00
committed by Mark Paluch
parent 72d82d3083
commit dd446472bc
2 changed files with 16 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
spring-data-mongodb/src/main/java/org/springframework/data/mongodb/core/aggregation/SelectionOperators.java
View File

@@ -55,6 +55,14 @@ public class SelectionOperators {
return new Bottom(Collections.emptyMap());
}
/**
* @param numberOfResults Limits the number of returned elements to the given value.
* @return new instance of {@link Bottom}.
*/
public static Bottom bottom(int numberOfResults) {
return bottom().limit(numberOfResults);
}
/**
* Limits the number of returned elements to the given value.
*
@@ -239,14 +247,14 @@ public class SelectionOperators {
* @return new instance of {@link First}.
*/
public static First first() {
return new First(Collections.emptyMap()).limit(1);
return new First(Collections.emptyMap());
}
/**
* @return new instance of {@link First}.
*/
public static First first(int numberOfResults) {
return new First(Collections.emptyMap()).limit(numberOfResults);
return first().limit(numberOfResults);
}
/**
@@ -333,14 +341,14 @@ public class SelectionOperators {
* @return new instance of {@link Last}.
*/
public static Last last() {
return new Last(Collections.emptyMap()).limit(1);
return new Last(Collections.emptyMap());
}
/**
* @return new instance of {@link Last}.
*/
public static Last last(int numberOfResults) {
return new Last(Collections.emptyMap()).limit(numberOfResults);
return last().limit(numberOfResults);
}
/**

8
spring-data-mongodb/src/test/java/org/springframework/data/mongodb/core/aggregation/SelectionOperatorUnitTests.java
View File

@@ -74,8 +74,8 @@ class SelectionOperatorUnitTests {
@Test // GH-4139
void bottomNRenderedCorrectly() {
Document document = SelectionOperators.Bottom.bottom().output(Fields.fields("playerId", "score"))
.sortBy(Sort.by(Direction.DESC, "score")).limit(3).toDocument(Aggregation.DEFAULT_CONTEXT);
Document document = SelectionOperators.Bottom.bottom(3).output(Fields.fields("playerId", "score"))
.sortBy(Sort.by(Direction.DESC, "score")).toDocument(Aggregation.DEFAULT_CONTEXT);
assertThat(document).isEqualTo(Document.parse("""
{
@@ -114,8 +114,8 @@ class SelectionOperatorUnitTests {
@Test // GH-4139
void topNRenderedCorrectly() {
Document document = SelectionOperators.Top.top().output(Fields.fields("playerId", "score"))
.sortBy(Sort.by(Direction.DESC, "score")).limit(3).toDocument(Aggregation.DEFAULT_CONTEXT);
Document document = SelectionOperators.Top.top(3).output(Fields.fields("playerId", "score"))
.sortBy(Sort.by(Direction.DESC, "score")).toDocument(Aggregation.DEFAULT_CONTEXT);
assertThat(document).isEqualTo(Document.parse("""
{