BATCH UPDATE/DELETE queries for per row updates with NoTx

ydb/core/kqp/common/batch/params.h

@@ -0,0 +1,15 @@
+#pragma once
+
+#include <util/generic/fwd.h>
+
+namespace NKikimr::NKqp::NBatchParams {
+
+const TString Header = "$_kqp_batch_";
+const TString IsFirstQuery = Header + "is_first_query";
+const TString IsLastQuery = Header + "is_last_query";
+const TString IsInclusiveLeft = Header + "is_inclusive_left";
+const TString IsInclusiveRight = Header + "is_inclusive_right";
+const TString Begin = Header + "begin_"; // begin_N
+const TString End = Header + "end_";     // end_N
+
+} // namespace NKikimr::NKqp::NPartitionedExecuter

ydb/core/kqp/common/kqp_resolve.cpp

@@ -14,6 +14,59 @@ using namespace NMiniKQL;
 using namespace NYql;
 using namespace NYql::NNodes;
 
+NUdf::TUnboxedValue MakeDefaultValueByType(const NKikimr::NMiniKQL::TType* type) {
+    auto dataType = static_cast<const NKikimr::NMiniKQL::TDataType*>(type);
+    switch (dataType->GetSchemeType()) {
+        case NUdf::TDataType<bool>::Id:
+            return NUdf::TUnboxedValuePod(false);
+        case NUdf::TDataType<ui8>::Id:
+        case NUdf::TDataType<i8>::Id:
+        case NUdf::TDataType<ui16>::Id:
+        case NUdf::TDataType<i16>::Id:
+        case NUdf::TDataType<i32>::Id:
+        case NUdf::TDataType<NUdf::TDate32>::Id:
+        case NUdf::TDataType<ui32>::Id:
+        case NUdf::TDataType<i64>::Id:
+        case NUdf::TDataType<NUdf::TDatetime64>::Id:
+        case NUdf::TDataType<NUdf::TTimestamp64>::Id:
+        case NUdf::TDataType<NUdf::TInterval64>::Id:
+        case NUdf::TDataType<ui64>::Id:
+        case NUdf::TDataType<float>::Id:
+        case NUdf::TDataType<double>::Id:
+        case NUdf::TDataType<NUdf::TDate>::Id:
+        case NUdf::TDataType<NUdf::TDatetime>::Id:
+        case NUdf::TDataType<NUdf::TTimestamp>::Id:
+        case NUdf::TDataType<NUdf::TInterval>::Id:
+            return NUdf::TUnboxedValuePod(0);
+        case NUdf::TDataType<NUdf::TJson>::Id:
+        case NUdf::TDataType<NUdf::TUtf8>::Id:
+        case NUdf::TDataType<NUdf::TJsonDocument>::Id:
+        case NUdf::TDataType<NUdf::TDyNumber>::Id:
+            return NKikimr::NMiniKQL::MakeString("");
+        case NUdf::TDataType<NUdf::TTzDate>::Id:
+        case NUdf::TDataType<NUdf::TTzDatetime>::Id:
+        case NUdf::TDataType<NUdf::TTzTimestamp>::Id:
+        case NUdf::TDataType<NUdf::TTzDate32>::Id:
+        case NUdf::TDataType<NUdf::TTzDatetime64>::Id:
+        case NUdf::TDataType<NUdf::TTzTimestamp64>::Id: {
+            return NUdf::TUnboxedValuePod(ValueFromString(NUdf::GetDataSlot(dataType->GetSchemeType()), ""));
+        }
+        case NUdf::TDataType<NUdf::TDecimal>::Id:
+            return NUdf::TUnboxedValuePod(NYql::NDecimal::FromHalfs(0, 0));
+        case NUdf::TDataType<NUdf::TUuid>::Id: {
+            union {
+                ui64 half[2];
+                char bytes[16];
+            } buf;
+            buf.half[0] = 0;
+            buf.half[1] = 0;
+            return NKikimr::NMiniKQL::MakeString(NUdf::TStringRef(buf.bytes, 16));
+        }
+        default:
+            return NKikimr::NMiniKQL::MakeString("");
+    }
+}
+
 TVector<TCell> MakeKeyCells(const NKikimr::NUdf::TUnboxedValue& value, const TVector<NScheme::TTypeInfo>& keyColumnTypes,
     const TVector<ui32>& keyColumnIndices, const NMiniKQL::TTypeEnvironment& typeEnv, bool copyValues)
 {

ydb/core/kqp/common/kqp_resolve.h

@@ -11,6 +11,7 @@
 #include <ydb/core/tx/scheme_cache/scheme_cache.h>
 
 #include <yql/essentials/minikql/mkql_node.h>
+#include <yql/essentials/minikql/mkql_string_util.h>
 
 #include <util/generic/map.h>
 
@@ -232,6 +233,8 @@ class TKqpTableKeys {
     THashMap<TTableId, TTable> TablesById;
 };
 
+NUdf::TUnboxedValue MakeDefaultValueByType(const NKikimr::NMiniKQL::TType* type);
+
 TVector<TCell> MakeKeyCells(const NKikimr::NUdf::TUnboxedValue& value, const TVector<NScheme::TTypeInfo>& keyColumnTypes,
     const TVector<ui32>& keyColumnIndices, const NMiniKQL::TTypeEnvironment& typeEnv, bool copyValues);