DS proxy mirror-3-dc restoration strategy test

Author: alexvru

ydb/core/blobstorage/dsproxy/dsproxy_blackboard.h

@@ -108,9 +108,9 @@ struct TBlobState {
 };
 
 struct TDiskGetRequest {
-    const TLogoBlobID Id;
-    const ui32 Shift;
-    const ui32 Size;
+    TLogoBlobID Id;
+    ui32 Shift;
+    ui32 Size;
     ssize_t PartMapIndex = -1;
 
     TDiskGetRequest(const TLogoBlobID &id, const ui32 shift, const ui32 size)
@@ -127,7 +127,7 @@ struct TDiskPutRequest {
         ReasonInitial,
         ReasonAccelerate
     };
-    const TLogoBlobID Id;
+    TLogoBlobID Id;
     TRope Buffer;
     EPutReason Reason;
     bool IsHandoff;

ydb/core/blobstorage/dsproxy/ut_strategy/strategy_ut.cpp

@@ -1,7 +1,9 @@
 #include <ydb/core/blobstorage/dsproxy/dsproxy_blackboard.h>
 #include <ydb/core/blobstorage/dsproxy/dsproxy_strategy_restore.h>
+#include <ydb/core/blobstorage/dsproxy/dsproxy_strategy_get_m3dc_restore.h>
 #include <library/cpp/testing/unittest/registar.h>
 #include <util/stream/null.h>
+#include <util/generic/overloaded.h>
 
 using namespace NActors;
 using namespace NKikimr;
@@ -193,10 +195,153 @@ void RunStrategyTest(TBlobStorageGroupType type) {
     }
 }
 
+struct TGetQuery {
+    ui32 OrderNumber;
+    TLogoBlobID Id;
+    ui32 Shift;
+    ui32 Size;
+
+    friend bool operator ==(const TGetQuery& x, const TGetQuery& y) {
+        return x.OrderNumber == y.OrderNumber && x.Id == y.Id && x.Shift == y.Shift && x.Size == y.Size;
+    }
+};
+
+struct TPutQuery {
+    ui32 OrderNumber;
+    TLogoBlobID Id;
+
+    friend bool operator ==(const TPutQuery& x, const TPutQuery& y) {
+        return x.OrderNumber == y.OrderNumber && x.Id == y.Id;
+    }
+};
+
+using TOperation = std::variant<TGetQuery, TPutQuery>;
+
+void RunTestLevel(const TBlobStorageGroupInfo& info, TBlackboard& blackboard,
+        const std::function<EStrategyOutcome(TBlackboard&)>& runStrategies, const TLogoBlobID& id,
+        std::vector<TOperation>& stock, TSubgroupPartLayout presenceMask, bool nonWorkingDomain) {
+    // see which operations we can add to the stock
+    const size_t stockSizeOnEntry = stock.size();
+    auto& requests = blackboard.GroupDiskRequests.DiskRequestsForOrderNumber;
+    for (ui32 i = 0; i < info.GetTotalVDisksNum(); ++i) {
+        for (const auto& get : requests[i].GetsToSend) {
+            stock.push_back(TGetQuery{i, get.Id, get.Shift, get.Size});
+        }
+        requests[i].GetsToSend.clear();
+        for (const auto& put : requests[i].PutsToSend) {
+            stock.push_back(TPutQuery{i, put.Id});
+        }
+        requests[i].PutsToSend.clear();
+    }
+
+    // try every single operation in stock
+    for (size_t i = 0; i < stock.size(); ++i) {
+        std::swap(stock[i], stock.back());
+        TOperation operation = std::move(stock.back());
+        stock.pop_back();
+
+        TBlackboard branch(blackboard);
+        TSubgroupPartLayout myPresenceMask(presenceMask);
+
+        std::visit(TOverloaded{
+            [&](const TGetQuery& op) {
+                const ui32 idxInSubgroup = info.GetTopology().GetIdxInSubgroup(info.GetVDiskId(op.OrderNumber), id.Hash());
+                if (nonWorkingDomain && idxInSubgroup % 3 == 2) {
+                    branch.AddErrorResponse(op.Id, op.OrderNumber);
+                } else if (myPresenceMask.GetDisksWithPart(op.Id.PartId() - 1) >> idxInSubgroup & 1) {
+                    const ui32 blobSize = op.Id.BlobSize();
+                    const ui32 shift = Min(op.Shift, blobSize);
+                    const ui32 size = Min(op.Size ? op.Size : Max<ui32>(), blobSize - shift);
+                    branch.AddResponseData(op.Id, op.OrderNumber, shift, TRope(TString(size, 'X')));
+                } else {
+                    branch.AddNoDataResponse(op.Id, op.OrderNumber);
+                }
+            },
+            [&](const TPutQuery& op) {
+                const ui32 idxInSubgroup = info.GetTopology().GetIdxInSubgroup(info.GetVDiskId(op.OrderNumber), id.Hash());
+                if (nonWorkingDomain && idxInSubgroup % 3 == 2) {
+                    branch.AddErrorResponse(op.Id, op.OrderNumber);
+                } else {
+                    myPresenceMask.AddItem(idxInSubgroup, op.Id.PartId() - 1, info.Type);
+                    branch.AddPutOkResponse(op.Id, op.OrderNumber);
+                }
+            }
+        }, operation);
+
+        auto outcome = runStrategies(branch);
+        UNIT_ASSERT(outcome != EStrategyOutcome::ERROR);
+        if (outcome == EStrategyOutcome::DONE) {
+            TBlobStorageGroupInfo::TOrderNums nums;
+            info.GetTopology().PickSubgroup(id.Hash(), nums);
+            UNIT_ASSERT(info.GetQuorumChecker().GetBlobState(myPresenceMask, {&info.GetTopology()}) == TBlobStorageGroupInfo::EBS_FULL);
+        } else {
+            RunTestLevel(info, branch, runStrategies, id, stock, myPresenceMask, nonWorkingDomain);
+        }
+
+        stock.push_back(std::move(operation));
+        std::swap(stock[i], stock.back());
+    }
+
+    // revert stock
+    stock.resize(stockSizeOnEntry);
+}
+
 Y_UNIT_TEST_SUITE(DSProxyStrategyTest) {
 
     Y_UNIT_TEST(Restore_block42) {
         RunStrategyTest<TRestoreStrategy>(TBlobStorageGroupType::Erasure4Plus2Block);
     }
 
+    Y_UNIT_TEST(Restore_mirror3dc) {
+        THPTimer timer;
+        const TBlobStorageGroupType type(TBlobStorageGroupType::ErasureMirror3dc);
+
+        TBlobStorageGroupInfo info(type, 1, 3, 3);
+        info.Ref();
+        TGroupQueues groupQueues(info.GetTopology());
+        groupQueues.Ref();
+
+        std::vector<TOperation> stock;
+
+        TLogContext logCtx(NKikimrServices::BS_PROXY, false);
+        logCtx.SuppressLog = true;
+
+        auto runStrategies = [&](TBlackboard& blackboard) {
+            return blackboard.RunStrategy(logCtx, TMirror3dcGetWithRestoreStrategy());
+        };
+
+        const ui32 base = RandomNumber(512u);
+        for (ui32 i = 0; i < 512; ++i) {
+            const ui32 diskMask = (base + i) % 512;
+            for (bool nonWorkingDomain : {false, true}) {
+                Cerr << "diskMask# " << diskMask << " nonWorkingDomain# " << nonWorkingDomain << Endl;
+
+                TBlackboard blackboard(&info, &groupQueues, NKikimrBlobStorage::UserData, NKikimrBlobStorage::FastRead);
+
+                const TLogoBlobID id(1'000'000'000, 1, 1, 0, 1000, 0);
+                TSubgroupPartLayout presenceMask;
+                blackboard.AddNeeded(id, 0, id.BlobSize());
+                bool partsAvailable = false;
+                for (ui32 idxInSubgroup = 0; idxInSubgroup < 9; ++idxInSubgroup) {
+                    if (diskMask >> idxInSubgroup & 1 && (!nonWorkingDomain || idxInSubgroup % 3 != 2)) {
+                        presenceMask.AddItem(idxInSubgroup, idxInSubgroup % 3, info.Type);
+                        partsAvailable = true;
+                    }
+                }
+                if (!partsAvailable) {
+                    continue;
+                }
+
+                auto outcome = runStrategies(blackboard);
+                UNIT_ASSERT(outcome == EStrategyOutcome::IN_PROGRESS);
+
+                RunTestLevel(info, blackboard, runStrategies, id, stock, presenceMask, nonWorkingDomain);
+
+                if (TDuration::Seconds(timer.Passed()) >= TDuration::Minutes(5)) {
+                    break;
+                }
+            }
+        }
+    }
+
 }