@@ -70,23 +70,29 @@ class TBatchSplitttingContext {
70
70
71
71
class TSerializedBatch {
72
72
private:
73
- YDB_READONLY_DEF (TString, SchemaData);
74
73
YDB_READONLY_DEF (TString, Data);
75
74
YDB_READONLY (ui32, RowsCount, 0 );
76
75
YDB_READONLY (ui32, RawBytes, 0 );
77
- std::optional<TString> SpecialKeys;
76
+ std::optional<TString> SpecialKeysFull;
77
+ std::optional<TString> SpecialKeysPayload;
78
+
78
79
public:
79
80
size_t GetSize () const {
80
81
return Data.size ();
81
82
}
82
83
83
- const TString& GetSpecialKeysSafe () const {
84
- AFL_VERIFY (SpecialKeys);
85
- return *SpecialKeys;
84
+ const TString& GetSpecialKeysPayloadSafe () const {
85
+ AFL_VERIFY (SpecialKeysPayload);
86
+ return *SpecialKeysPayload;
87
+ }
88
+
89
+ const TString& GetSpecialKeysFullSafe () const {
90
+ AFL_VERIFY (SpecialKeysFull);
91
+ return *SpecialKeysFull;
86
92
}
87
93
88
94
bool HasSpecialKeys () const {
89
- return !!SpecialKeys ;
95
+ return !!SpecialKeysFull ;
90
96
}
91
97
92
98
TString DebugString () const ;
@@ -95,14 +101,14 @@ class TSerializedBatch {
95
101
static TConclusionStatus BuildWithLimit (std::shared_ptr<arrow::RecordBatch> batch, const TBatchSplitttingContext& context, std::optional<TSerializedBatch>& sbL, std::optional<TSerializedBatch>& sbR);
96
102
static TSerializedBatch Build (std::shared_ptr<arrow::RecordBatch> batch, const TBatchSplitttingContext& context);
97
103
98
- TSerializedBatch (TString&& schemaData, TString&& data, const ui32 rowsCount, const ui32 rawBytes, const std::optional<TString>& specialKeys)
99
- : SchemaData(schemaData )
100
- , Data(data)
104
+ TSerializedBatch (TString&& data, const ui32 rowsCount, const ui32 rawBytes,
105
+ const std::optional<TString>& specialKeysPayload, const std::optional<TString>& specialKeysFull )
106
+ : Data(data)
101
107
, RowsCount(rowsCount)
102
108
, RawBytes(rawBytes)
103
- , SpecialKeys(specialKeys )
104
- {
105
-
109
+ , SpecialKeysFull(specialKeysFull )
110
+ , SpecialKeysPayload(specialKeysPayload) {
111
+ AFL_VERIFY (!!SpecialKeysPayload == !!SpecialKeysFull);
106
112
}
107
113
};
108
114
0 commit comments