Add expression support for median aggregation.

Original Pull Request: #4515
spring-projects · Sep 28, 2023 · f84a62b · f84a62b
1 parent d506cd0
commit f84a62b
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/...mongodb/src/main/java/org/springframework/data/mongodb/core/spel/MethodReferenceNode.java b/...mongodb/src/main/java/org/springframework/data/mongodb/core/spel/MethodReferenceNode.java
@@ -231,6 +231,8 @@ public class MethodReferenceNode extends ExpressionNode {
 				.mappingParametersTo("n", "input"));
 		map.put("percentile", mapArgRef().forOperator("$percentile") //
 				.mappingParametersTo("input", "p", "method"));
+		map.put("median", mapArgRef().forOperator("$median") //
+				.mappingParametersTo("input", "method"));
 
 		// TYPE OPERATORS
 		map.put("type", singleArgRef().forOperator("$type"));

diff --git a/...org/springframework/data/mongodb/core/aggregation/SpelExpressionTransformerUnitTests.java b/...org/springframework/data/mongodb/core/aggregation/SpelExpressionTransformerUnitTests.java
@@ -1269,6 +1269,16 @@ void shouldRenderPercentile() {
 			.isEqualTo("{ $percentile : { input : \"$score\", p : [0.4, 0.85], method : \"approximate\" }}");
 	}
 
+	@Test // GH-4472
+	void shouldRenderMedian() {
+
+		assertThat(transform("median(new String[]{\"$scoreOne\", \"$scoreTwo\" }, \"approximate\")"))
+				.isEqualTo("{ $median : { input : [\"$scoreOne\", \"$scoreTwo\"], method : \"approximate\" }}");
+
+		assertThat(transform("median(score, \"approximate\")"))
+				.isEqualTo("{ $median : { input : \"$score\", method : \"approximate\" }}");
+	}
+
 	private Document transform(String expression, Object... params) {
 		return (Document) transformer.transform(expression, Aggregation.DEFAULT_CONTEXT, params);
 	}