Fix: reset RawTypeV3, when Type is set (and vice versa)

commit_hash:af63b4f5a0c2db692a0224330b79c5088c6982dc

1 files changed, 21 insertions, 0 deletions

diff --git a/yt/cpp/mapreduce/interface/common.cpp b/yt/cpp/mapreduce/interface/common.cpp
index 831762fa95..966be8341f 100644
--- a/yt/cpp/mapreduce/interface/common.cpp
+++ b/yt/cpp/mapreduce/interface/common.cpp
@@ -348,6 +348,7 @@ TColumnSchema& TColumnSchema::Type(const NTi::TTypePtr& type) &
 {
     Y_ABORT_UNLESS(type.Get(), "Cannot create column schema with nullptr type");
     TypeV3_ = type;
+    RawTypeV3_ = {};
     return *this;
 }
 
@@ -355,6 +356,7 @@ TColumnSchema TColumnSchema::Type(const NTi::TTypePtr& type) &&
 {
     Y_ABORT_UNLESS(type.Get(), "Cannot create column schema with nullptr type");
     TypeV3_ = type;
+    RawTypeV3_ = {};
     return *this;
 }
 
@@ -388,6 +390,25 @@ TColumnSchema TColumnSchema::Type(EValueType type, bool required) &&
     return Type(ToTypeV3(type, required));
 }
 
+const TMaybe<TNode>& TColumnSchema::RawTypeV3() const
+{
+    return RawTypeV3_;
+}
+
+TColumnSchema& TColumnSchema::RawTypeV3(TNode rawTypeV3) &
+{
+    RawTypeV3_ = std::move(rawTypeV3);
+    TypeV3_ = nullptr;
+    return *this;
+}
+
+TColumnSchema TColumnSchema::RawTypeV3(TNode rawTypeV3) &&
+{
+    RawTypeV3_ = std::move(rawTypeV3);
+    TypeV3_ = nullptr;
+    return *this;
+}
+
 bool operator==(const TColumnSchema& lhs, const TColumnSchema& rhs)
 {
     return