messages for shadow partition

ydb/core/persqueue/events/internal.h

@@ -4,6 +4,7 @@
 
 #include <ydb/core/base/row_version.h>
 #include <ydb/core/protos/pqconfig.pb.h>
+#include <ydb/core/persqueue/blob.h>
 #include <ydb/core/persqueue/key.h>
 #include <ydb/core/persqueue/metering_sink.h>
 #include <ydb/core/tablet/tablet_counters.h>
@@ -65,6 +66,18 @@ namespace NPQ {
         {}
     };
 
+    struct TDataKey {
+        TKey Key;
+        ui32 Size;
+        TInstant Timestamp;
+        ui64 CumulativeSize;
+    };
+
+    struct TSeqNoRange {
+        ui64 Min;
+        ui64 Max;
+    };
+
     struct TErrorInfo {
         NPersQueue::NErrorCode::EErrorCode ErrorCode;
         TString ErrorStr;
@@ -171,6 +184,9 @@ struct TEvPQ {
         EvProvideDirectReadInfo,
         EvCheckPartitionStatusRequest,
         EvCheckPartitionStatusResponse,
+        EvGetWriteInfoRequest,
+        EvGetWriteInfoResponse,
+        EvGetWriteInfoError,
         EvEnd
     };
 
@@ -1004,7 +1020,43 @@ struct TEvPQ {
     struct TEvCheckPartitionStatusResponse : public TEventPB<TEvCheckPartitionStatusResponse, NKikimrPQ::TEvCheckPartitionStatusResponse, EvCheckPartitionStatusResponse> {
     };
 
+    struct TEvGetWriteInfoRequest : public TEventLocal<TEvGetWriteInfoRequest, EvGetWriteInfoRequest> {
+        explicit TEvGetWriteInfoRequest(ui32 cookie) :
+            Cookie(cookie)
+        {
+        }
+
+        ui32 Cookie; // ShadowPartitionId
+    };
+
+    struct TEvGetWriteInfoResponse : public TEventLocal<TEvGetWriteInfoResponse, EvGetWriteInfoResponse> {
+        TEvGetWriteInfoResponse(ui32 cookie,
+                                THashMap<TString, NPQ::TSeqNoRange>&& seqNo,
+                                std::deque<NPQ::TDataKey>&& bodyKeys,
+                                TVector<NPQ::TClientBlob>&& head) :
+            Cookie(cookie),
+            SeqNo(std::move(seqNo)),
+            BodyKeys(std::move(bodyKeys)),
+            Head(std::move(head))
+        {
+        }
+
+        ui32 Cookie; // ShadowPartitionId
+        THashMap<TString, NPQ::TSeqNoRange> SeqNo; // SourceId -> (MinSeqNo, MaxSeqNo)
+        std::deque<NPQ::TDataKey> BodyKeys;
+        TVector<NPQ::TClientBlob> Head;
+    };
+
+    struct TEvGetWriteInfoError : public TEventLocal<TEvGetWriteInfoError, EvGetWriteInfoError> {
+        ui32 Cookie; // ShadowPartitionId
+        TString Message;
 
+        TEvGetWriteInfoError(ui32 cookie, TString message) :
+            Cookie(cookie),
+            Message(std::move(message))
+        {
+        }
+    };
 };
 
 } //NKikimr

ydb/core/persqueue/partition_types.h

@@ -120,13 +120,6 @@ struct TMessage {
     #undef DEFINE_CHECKER_GETTER
 };
 
-struct TDataKey {
-    TKey Key;
-    ui32 Size;
-    TInstant Timestamp;
-    ui64 CumulativeSize;
-};
-
 
 } // namespace NKikimr::NPQ