Skip to content

Commit 56d4674

Browse files
committed
Moved commit "Add few auto partitioning fields to describe and SDK" from ydb repo
1 parent 566ac7a commit 56d4674

File tree

2 files changed

+43
-6
lines changed

2 files changed

+43
-6
lines changed

include/ydb-cpp-sdk/client/topic/control_plane.h

Lines changed: 18 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -145,14 +145,21 @@ class TPartitionInfo {
145145
const std::optional<TPartitionConsumerStats>& GetPartitionConsumerStats() const;
146146
const std::optional<TPartitionLocation>& GetPartitionLocation() const;
147147

148+
const std::optional<std::string>& GetFromBound() const;
149+
const std::optional<std::string>& GetToBound() const;
150+
148151
private:
149152
uint64_t PartitionId_;
150153
bool Active_;
151154
std::vector<uint64_t> ChildPartitionIds_;
152155
std::vector<uint64_t> ParentPartitionIds_;
156+
153157
std::optional<TPartitionStats> PartitionStats_;
154158
std::optional<TPartitionConsumerStats> PartitionConsumerStats_;
155159
std::optional<TPartitionLocation> PartitionLocation_;
160+
161+
std::optional<std::string> FromBound_;
162+
std::optional<std::string> ToBound_;
156163
};
157164

158165
struct TAlterPartitioningSettings;
@@ -187,8 +194,8 @@ friend struct TAutoPartitioningSettingsBuilder;
187194

188195
struct TAlterAutoPartitioningSettings {
189196
using TSelf = TAlterAutoPartitioningSettings;
190-
public:
191-
TAlterAutoPartitioningSettings(TAlterPartitioningSettings& parent): Parent_(parent) {}
197+
public:
198+
TAlterAutoPartitioningSettings(TAlterPartitioningSettings& parent): Parent_(parent) {}
192199

193200
FLUENT_SETTING_OPTIONAL(EAutoPartitioningStrategy, Strategy);
194201
FLUENT_SETTING_OPTIONAL(TDuration, StabilizationWindow);
@@ -197,8 +204,8 @@ struct TAlterAutoPartitioningSettings {
197204

198205
TAlterPartitioningSettings& EndAlterAutoPartitioningSettings() { return Parent_; };
199206

200-
private:
201-
TAlterPartitioningSettings& Parent_;
207+
private:
208+
TAlterPartitioningSettings& Parent_;
202209
};
203210

204211
class TPartitioningSettings {
@@ -207,11 +214,11 @@ class TPartitioningSettings {
207214
public:
208215
TPartitioningSettings() : MinActivePartitions_(0), MaxActivePartitions_(0), PartitionCountLimit_(0), AutoPartitioningSettings_(){}
209216
TPartitioningSettings(const Ydb::Topic::PartitioningSettings& settings);
210-
TPartitioningSettings(ui64 minActivePartitions, ui64 maxActivePartitions, TAutoPartitioningSettings autoscalingSettings = {})
217+
TPartitioningSettings(uint64_t minActivePartitions, uint64_t maxActivePartitions, TAutoPartitioningSettings autoPartitioning = {})
211218
: MinActivePartitions_(minActivePartitions)
212219
, MaxActivePartitions_(maxActivePartitions)
213220
, PartitionCountLimit_(0)
214-
, AutoPartitioningSettings_(autoscalingSettings)
221+
, AutoPartitioningSettings_(autoPartitioning)
215222
{
216223
}
217224

@@ -460,6 +467,11 @@ struct TConsumerSettings {
460467
return *this;
461468
}
462469

470+
TConsumerSettings& SetImportant(bool isImportant) {
471+
Important_ = isImportant;
472+
return *this;
473+
}
474+
463475
TSettings& EndAddConsumer() { return Parent_; };
464476

465477
private:

src/client/topic/impl/topic.cpp

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -395,13 +395,22 @@ TPartitionInfo::TPartitionInfo(const Ydb::Topic::DescribeTopicResult::PartitionI
395395
for (const auto& partId : partitionInfo.parent_partition_ids()) {
396396
ParentPartitionIds_.push_back(partId);
397397
}
398+
398399
if (partitionInfo.has_partition_stats()) {
399400
PartitionStats_ = TPartitionStats{partitionInfo.partition_stats()};
400401
}
401402

402403
if (partitionInfo.has_partition_location()) {
403404
PartitionLocation_ = TPartitionLocation{partitionInfo.partition_location()};
404405
}
406+
407+
if (partitionInfo.has_key_range() && partitionInfo.key_range().has_from_bound()) {
408+
FromBound_ = std::string{partitionInfo.key_range().from_bound()};
409+
}
410+
411+
if (partitionInfo.has_key_range() && partitionInfo.key_range().has_to_bound()) {
412+
ToBound_ = std::string{partitionInfo.key_range().to_bound()};
413+
}
405414
}
406415

407416
TPartitionInfo::TPartitionInfo(const Ydb::Topic::DescribeConsumerResult::PartitionInfo& partitionInfo)
@@ -437,6 +446,14 @@ const std::optional<TPartitionLocation>& TPartitionInfo::GetPartitionLocation()
437446
return PartitionLocation_;
438447
}
439448

449+
const std::vector<uint64_t> TPartitionInfo::GetChildPartitionIds() const {
450+
return ChildPartitionIds_;
451+
}
452+
453+
const std::vector<uint64_t> TPartitionInfo::GetParentPartitionIds() const {
454+
return ParentPartitionIds_;
455+
}
456+
440457
bool TPartitionInfo::GetActive() const {
441458
return Active_;
442459
}
@@ -445,6 +462,14 @@ uint64_t TPartitionInfo::GetPartitionId() const {
445462
return PartitionId_;
446463
}
447464

465+
const std::optional<std::string>& TPartitionInfo::GetFromBound() const {
466+
return FromBound_;
467+
}
468+
469+
const std::optional<std::string>& TPartitionInfo::GetToBound() const {
470+
return ToBound_;
471+
}
472+
448473
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
449474
// TTopicClient
450475

0 commit comments

Comments
 (0)