From 7722b663d6d05273385d2647ff638a2dff787b30 Mon Sep 17 00:00:00 2001
From: "myroslav.kosinskyi" <kosinski@ukr.net>
Date: Thu, 6 Oct 2022 17:42:50 +0300
Subject: [PATCH] Change Streams fullDocumentBeforeChange support

Closes #4187
---
 .../data/mongodb/core/ChangeStreamEvent.java  | 20 ++++---------------
 1 file changed, 4 insertions(+), 16 deletions(-)

diff --git a/spring-data-mongodb/src/main/java/org/springframework/data/mongodb/core/ChangeStreamEvent.java b/spring-data-mongodb/src/main/java/org/springframework/data/mongodb/core/ChangeStreamEvent.java
index 33a561a2a6..4a92c68e87 100644
--- a/spring-data-mongodb/src/main/java/org/springframework/data/mongodb/core/ChangeStreamEvent.java
+++ b/spring-data-mongodb/src/main/java/org/springframework/data/mongodb/core/ChangeStreamEvent.java
@@ -155,33 +155,21 @@ public String getCollectionName() {
 	@Nullable
 	public T getBody() {
 
-		if (raw == null) {
+		if (raw == null || raw.getFullDocument() == null) {
 			return null;
 		}
 
-		Document fullDocument = raw.getFullDocument();
-
-		if (fullDocument == null) {
-			return targetType.cast(fullDocument);
-		}
-
-		return getConvertedFullDocument(fullDocument);
+		return getConvertedFullDocument(raw.getFullDocument());
 	}
 
 	@Nullable
 	public T getBodyBeforeChange() {
 
-		if (raw == null) {
+		if (raw == null || raw.getFullDocumentBeforeChange() == null) {
 			return null;
 		}
 
-		Document fullDocumentBeforeChange = raw.getFullDocumentBeforeChange();
-
-		if (fullDocumentBeforeChange == null) {
-			return targetType.cast(fullDocumentBeforeChange);
-		}
-
-		return getConvertedFullDocumentBeforeChange(fullDocumentBeforeChange);
+		return getConvertedFullDocumentBeforeChange(raw.getFullDocumentBeforeChange());
 	}
 
 	@SuppressWarnings("unchecked")