From efd89bd52147db2494cc10ff66f156663eadbf3d Mon Sep 17 00:00:00 2001
From: Serge Rider <serge@jkiss.org>
Date: Thu, 16 Dec 2021 10:40:09 +0300
Subject: [PATCH] CB-1504 DynamoDB: delete documents by row value

---
 .../src/io/cloudbeaver/service/sql/WebSQLProcessor.java       | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/server/bundles/io.cloudbeaver.server/src/io/cloudbeaver/service/sql/WebSQLProcessor.java b/server/bundles/io.cloudbeaver.server/src/io/cloudbeaver/service/sql/WebSQLProcessor.java
index 46d46667bd..a149908f9d 100644
--- a/server/bundles/io.cloudbeaver.server/src/io/cloudbeaver/service/sql/WebSQLProcessor.java
+++ b/server/bundles/io.cloudbeaver.server/src/io/cloudbeaver/service/sql/WebSQLProcessor.java
@@ -487,8 +487,10 @@ public class WebSQLProcessor {
                     }
                     Map<DBDAttributeBinding, Object> delKeyAttributes = new LinkedHashMap<>();
 
+                    boolean isDocumentKey = keyAttributes.length == 1 && keyAttributes[0].getDataKind() == DBPDataKind.DOCUMENT && dataContainer instanceof DBSDocumentLocator;
+
                     for (int i = 0; i < allAttributes.length; i++) {
-                        if (ArrayUtils.contains(keyAttributes, allAttributes[i])) {
+                        if (isDocumentKey || ArrayUtils.contains(keyAttributes, allAttributes[i])) {
                             Object realCellValue = convertInputCellValue(session, allAttributes[i],
                                 keyData.get(i));
                             delKeyAttributes.put(allAttributes[i], realCellValue);