Moved commit "CDC Initial Scan progress" from ydb repo

Author: CyberROFL

include/ydb-cpp-sdk/client/table/table.h

@@ -264,10 +264,24 @@ class TBuildIndexOperation : public TOperation {
 
 ////////////////////////////////////////////////////////////////////////////////
 
-//! Represents index description
+//! Represents changefeed description
 class TChangefeedDescription {
     friend class NYdb::TProtoAccessor;
 
+public:
+    class TInitialScanProgress {
+    public:
+        explicit TInitialScanProgress(uint32_t total, uint32_t completed);
+
+        uint32_t GetPartsTotal() const;
+        uint32_t GetPartsCompleted() const;
+        float GetProgress() const; // percentage
+
+    private:
+        uint32_t PartsTotal;
+        uint32_t PartsCompleted;
+    };
+
 public:
     TChangefeedDescription(const std::string& name, EChangefeedMode mode, EChangefeedFormat format);
 
@@ -295,6 +309,7 @@ class TChangefeedDescription {
     bool GetInitialScan() const;
     const std::unordered_map<std::string, std::string>& GetAttributes() const;
     const std::string& GetAwsRegion() const;
+    const std::optional<TInitialScanProgress>& GetInitialScanProgress() const;
 
     void SerializeTo(Ydb::Table::Changefeed& proto) const;
     std::string ToString() const;
@@ -318,6 +333,7 @@ class TChangefeedDescription {
     bool InitialScan_ = false;
     std::unordered_map<std::string, std::string> Attributes_;
     std::string AwsRegion_;
+    std::optional<TInitialScanProgress> InitialScanProgress_;
 };
 
 bool operator==(const TChangefeedDescription& lhs, const TChangefeedDescription& rhs);

src/client/table/table.cpp

@@ -2398,6 +2398,27 @@ TChangefeedDescription::TChangefeedDescription(const Ydb::Table::ChangefeedDescr
     : TChangefeedDescription(FromProto(proto))
 {}
 
+TChangefeedDescription::TInitialScanProgress::TInitialScanProgress(uint32_t total, uint32_t completed)
+    : PartsTotal(total)
+    , PartsCompleted(completed)
+{}
+
+uint32_t TChangefeedDescription::TInitialScanProgress::GetPartsTotal() const {
+    return PartsTotal;
+}
+
+uint32_t TChangefeedDescription::TInitialScanProgress::GetPartsCompleted() const {
+    return PartsCompleted;
+}
+
+float TChangefeedDescription::TInitialScanProgress::GetProgress() const {
+    if (PartsTotal == 0) {
+        return 0;
+    }
+
+    return 100 * float(PartsCompleted) / float(PartsTotal);
+}
+
 TChangefeedDescription& TChangefeedDescription::WithVirtualTimestamps() {
     VirtualTimestamps_ = true;
     return *this;
@@ -2474,6 +2495,10 @@ const std::string& TChangefeedDescription::GetAwsRegion() const {
     return AwsRegion_;
 }
 
+const std::optional<TChangefeedDescription::TInitialScanProgress>& TChangefeedDescription::GetInitialScanProgress() const {
+    return InitialScanProgress_;
+}
+
 template <typename TProto>
 TChangefeedDescription TChangefeedDescription::FromProto(const TProto& proto) {
     EChangefeedMode mode;
@@ -2541,6 +2566,13 @@ TChangefeedDescription TChangefeedDescription::FromProto(const TProto& proto) {
             ret.State_ = EChangefeedState::Unknown;
             break;
         }
+
+        if (proto.has_initial_scan_progress()) {
+            ret.InitialScanProgress_ = std::make_optional<TInitialScanProgress>(
+                proto.initial_scan_progress().parts_total(),
+                proto.initial_scan_progress().parts_completed()
+            );
+        }
     }
 
     for (const auto& [key, value] : proto.attributes()) {
@@ -2628,6 +2660,10 @@ void TChangefeedDescription::Out(IOutputStream& o) const {
         o << ", aws_region: " << AwsRegion_;
     }
 
+    if (InitialScanProgress_) {
+        o << ", initial_scan_progress: " << InitialScanProgress_->GetProgress() << "%";
+    }
+
     o << " }";
 }