Fix MODE_UPSERT for update queries (#17002)

Author: dahbka-lis

ydb/core/kqp/opt/kqp_opt_kql.cpp

@@ -204,6 +204,10 @@ TExprNode::TPtr GetPgNotNullColumns(
 
 TExprNode::TPtr IsUpdateSetting(TExprContext& ctx, const TPositionHandle& pos) {
     return Build<TCoNameValueTupleList>(ctx, pos)
+        .Add()
+            .Name().Build("Mode")
+            .Value<TCoAtom>().Build("update")
+        .Build()
         .Add()
             .Name().Build("IsUpdate")
         .Build()
@@ -212,6 +216,10 @@ TExprNode::TPtr IsUpdateSetting(TExprContext& ctx, const TPositionHandle& pos) {
 
 TExprNode::TPtr IsConditionalUpdateSetting(TExprContext& ctx, const TPositionHandle& pos) {
     return Build<TCoNameValueTupleList>(ctx, pos)
+        .Add()
+            .Name().Build("Mode")
+            .Value<TCoAtom>().Build("update")
+        .Build()
         .Add()
             .Name().Build("IsUpdate")
         .Build()

ydb/core/kqp/ut/query/kqp_explain_ut.cpp

@@ -523,7 +523,7 @@ Y_UNIT_TEST_SUITE(KqpExplain) {
         auto deletesConstCount = CountPlanNodesByKv(plan, "Node Type", "Delete-ConstantExpr");
         UNIT_ASSERT_VALUES_EQUAL(deletesConstCount, UseSink ? 0 : 1);
 
-        auto upsertsCount = CountPlanNodesByKv(plan, "Name", "Upsert");
+        auto upsertsCount = CountPlanNodesByKv(plan, "Name", UseSink ? "Update" : "Upsert");
         UNIT_ASSERT_VALUES_EQUAL(upsertsCount, UseSink ? 2 : 2);
 
         auto deletesCount = CountPlanNodesByKv(plan, "Name", "Delete");
@@ -549,7 +549,7 @@ Y_UNIT_TEST_SUITE(KqpExplain) {
         countOperationsByType("reads");
         countOperationsByType("writes");
 
-        UNIT_ASSERT_VALUES_EQUAL(counter["MultiUpsert"], UseSink ? upsertsCount : upsertsConstCount);
+        UNIT_ASSERT_VALUES_EQUAL(counter[UseSink ? "MultiUpdate" : "MultiUpsert"], UseSink ? upsertsCount : upsertsConstCount);
         UNIT_ASSERT_VALUES_EQUAL(counter["MultiErase"], UseSink ? deletesCount : deletesConstCount);
         UNIT_ASSERT_VALUES_EQUAL(counter["FullScan"], fullScansCount);
         UNIT_ASSERT_VALUES_EQUAL(counter["Scan"], rangeScansCount);

ydb/tests/functional/canonical/canondata/test_sql.TestCanonicalFolder1.test_case_write_update.sql-plan_/write_update.sql.plan

@@ -29,7 +29,7 @@
                         "Group",
                         "Name"
                     ],
-                    "type": "MultiUpsert"
+                    "type": "MultiUpdate"
                 }
             ]
         }