fix serde of RemoteDocLevelMonitorInput (#666)

Signed-off-by: Subhobrata Dey <[email protected]>
opensearch-project · Jun 5, 2024 · 0d21526 · 0d21526
1 parent 79994f6
commit 0d21526
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))