The request should contain only known partitions (#20547) (#20567)

Author: Alek5andr-Kotov

ydb/core/kqp/topics/kqp_topics.cpp

@@ -219,6 +219,11 @@ ui64 TTopicPartitionOperations::GetTabletId() const
     return *TabletId_;
 }
 
+bool TTopicPartitionOperations::HasTabletId() const
+{
+    return TabletId_.Defined();
+}
+
 void TTopicPartitionOperations::SetTabletId(ui64 value)
 {
     Y_ABORT_UNLESS(TabletId_.Empty());
@@ -424,6 +429,17 @@ bool TTopicOperations::ProcessSchemeCacheNavigate(const NSchemeCache::TSchemeCac
         }
     }
 
+    for (const auto& [key, operations] : Operations_) {
+        if (!operations.HasTabletId()) {
+            builder << "Topic '" << key.Topic_ << "'. Unknown partition " << key.Partition_;
+
+            status = Ydb::StatusIds::SCHEME_ERROR;
+            message = std::move(builder);
+
+            return false;
+        }
+    }
+
     status = Ydb::StatusIds::SUCCESS;
     message = "";
 

ydb/core/kqp/topics/kqp_topics.h

@@ -91,6 +91,7 @@ class TTopicPartitionOperations {
 
     void SetTabletId(ui64 value);
     ui64 GetTabletId() const;
+    bool HasTabletId() const;
 
     TMaybe<TString> GetTopicName() const;
 

ydb/services/persqueue_v1/ut/topic_service_ut.cpp

@@ -311,6 +311,20 @@ Y_UNIT_TEST_F(UnknownTopic, TUpdateOffsetsInTransactionFixture) {
     UNIT_ASSERT_VALUES_EQUAL(response.operation().status(), Ydb::StatusIds::SCHEME_ERROR);
 }
 
+Y_UNIT_TEST_F(UnknownPartition, TUpdateOffsetsInTransactionFixture) {
+    auto response = Call_UpdateOffsetsInTransaction({
+        TTopic{.Path=VALID_TOPIC_PATH, .Partitions={
+            TPartition{.Id=1000, .Offsets={
+                TOffsetRange{.Begin=0, .End=2}
+            }},
+            TPartition{.Id=2000, .Offsets={
+                TOffsetRange{.Begin=1, .End=2}
+            }}
+        }}
+    });
+    UNIT_ASSERT_VALUES_EQUAL(response.operation().status(), Ydb::StatusIds::SCHEME_ERROR);
+}
+
 Y_UNIT_TEST_F(UseDoubleSlashInTopicPath, TUpdateOffsetsInTransactionFixture) {
     TestTopicPaths("//Root//PQ//rt3.dc1--topic1", "/Root/PQ/rt3.dc1--topic1");
 }