Refactored operation id

ydb-platform · Aug 29, 2024 · e51b937 · e51b937
1 parent 66fc8f0
commit e51b937
Show file tree

Hide file tree

Showing 3 changed files with 35 additions and 9 deletions.
diff --git a/include/ydb-cpp-sdk/library/operation_id/operation_id.h b/include/ydb-cpp-sdk/library/operation_id/operation_id.h
@@ -35,8 +35,17 @@ class TOperationId {
 
     TOperationId();
     explicit TOperationId(const std::string& string, bool allowEmpty = false);
+
+    TOperationId(const TOperationId& other);
+    TOperationId(TOperationId&& other) = default;
+
+    TOperationId& operator=(const TOperationId& other);
+    TOperationId& operator=(TOperationId&& other) = default;
+
+    ~TOperationId() = default;
+
     EKind GetKind() const;
-    EKind& GetMutableKind();
+    void SetKind(const EKind& kind);
 
     const TDataList& GetData() const;
     TDataList& GetMutableData();
@@ -47,6 +56,7 @@ class TOperationId {
 
 private:
     bool IsValidKind(int kind);
+    void CopyData(const TOperationId::TDataList& otherData);
 
     EKind Kind;
     TDataList Data;

diff --git a/include/ydb-cpp-sdk/library/yql_common/issue/yql_issue.h b/include/ydb-cpp-sdk/library/yql_common/issue/yql_issue.h
@@ -112,7 +112,7 @@ class TIssue;
 using TIssuePtr = TIntrusivePtr<TIssue>;
 class TIssue: public TThrRefBase {
     std::vector<TIntrusivePtr<TIssue>> Children_;
-    std::string Message;
+    TString Message;
 public:
     TPosition Position;
     TPosition EndPosition;
@@ -209,7 +209,7 @@ class TIssue: public TThrRefBase {
     }
 
     // Unsafe method. Doesn't call SanitizeNonAscii(Message)
-    std::string* MutableMessage() {
+    TString* MutableMessage() {
         return &Message;
     }
 

diff --git a/src/library/operation_id/operation_id.cpp b/src/library/operation_id/operation_id.cpp
@@ -120,19 +120,35 @@ TOperationId::TOperationId(const std::string &string, bool allowEmpty) {
     }
 }
 
-TOperationId::EKind TOperationId::GetKind() const {
-    return Kind;
+TOperationId::TOperationId(const TOperationId& other) {
+    Kind = other.Kind;
+    CopyData(other.Data);
+}
+
+TOperationId& TOperationId::operator=(const TOperationId& other) {
+    Kind = other.Kind;
+    Data.clear();
+    Index.clear();
+    CopyData(other.Data);
+    return *this;
 }
 
-TOperationId::EKind& TOperationId::GetMutableKind() {
+void TOperationId::CopyData(const TOperationId::TDataList& otherData) {
+    for (const auto& data : otherData) {
+        Data.push_back(std::make_unique<TData>(data->Key, data->Value));
+        Index[data->Key].push_back(&Data.back()->Value);
+    }
+}
+
+TOperationId::EKind TOperationId::GetKind() const {
     return Kind;
 }
 
-const TOperationId::TDataList& TOperationId::GetData() const {
-    return Data;
+void TOperationId::SetKind(const EKind& kind) {
+    Kind = kind;
 }
 
-TOperationId::TDataList& TOperationId::GetMutableData() {
+const TOperationId::TDataList& TOperationId::GetData() const {
     return Data;
 }