fix serde of RemoteDocLevelMonitorInput (opensearch-project#666)

Signed-off-by: Subhobrata Dey <sbcd90@gmail.com>
jowg-amazon · Jul 2, 2024 · 211e4b6 · 211e4b6
1 parent e18d899
commit 211e4b6
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/...otlin/org/opensearch/commons/alerting/model/remote/monitors/RemoteDocLevelMonitorInput.kt b/...otlin/org/opensearch/commons/alerting/model/remote/monitors/RemoteDocLevelMonitorInput.kt
@@ -68,7 +68,10 @@ data class RemoteDocLevelMonitorInput(val input: BytesReference, val docLevelMon
                 when (fieldName) {
                     RemoteMonitorInput.INPUT_FIELD -> bytes = xcp.binaryValue()
                     RemoteMonitorInput.INPUT_SIZE -> size = xcp.intValue()
-                    Input.Type.DOCUMENT_LEVEL_INPUT.value -> docLevelMonitorInput = DocLevelMonitorInput.parse(xcp)
+                    Input.Type.DOCUMENT_LEVEL_INPUT.value -> {
+                        docLevelMonitorInput = DocLevelMonitorInput.parse(xcp)
+                        XContentParserUtils.ensureExpectedToken(XContentParser.Token.END_OBJECT, xcp.nextToken(), xcp)
+                    }
                 }
             }
             val input = BytesReference.fromByteBuffer(ByteBuffer.wrap(bytes, 0, size))