Add forsake shortcut call to forcibly drop existing shortcut

commit_hash:5f1b87013c927f79e74d39da431010617d816f2d

Author: olegsidorkin

yt/yt/client/api/internal_client.h

@@ -139,6 +139,12 @@ struct TForsakeChaosCoordinatorOptions
 
 ////////////////////////////////////////////////////////////////////////////////
 
+struct TForsakeChaosShortcutOptions
+    : public TTimeoutOptions
+{ };
+
+////////////////////////////////////////////////////////////////////////////////
+
 struct TRemoveChaosCellMailboxOptions
     : public TTimeoutOptions
 { };
@@ -240,6 +246,11 @@ struct IInternalClient
         NHydra::TCellId coordinatorCellId,
         const TForsakeChaosCoordinatorOptions& options = {}) = 0;
 
+    virtual TFuture<void> ForsakeChaosShortcut(
+        NHydra::TCellId coordinatorCellId,
+        NChaosClient::TChaosObjectId chaosObjectId,
+        const TForsakeChaosShortcutOptions& options = {}) = 0;
+
     virtual TFuture<void> RemoveChaosCellMailbox(
         NHydra::TCellId chaosCellId,
         NHydra::TCellId destinationCellId,

yt/yt/client/driver/driver.cpp

@@ -443,6 +443,7 @@ class TDriver
             REGISTER_ALL(TReferenceLeaseCommand,            "reference_lease",                        Null,       Structured, true,  false);
             REGISTER_ALL(TUnreferenceLeaseCommand,          "unreference_lease",                      Null,       Structured, true,  false);
             REGISTER_ALL(TForsakeChaosCoordinator,          "forsake_chaos_coordinator",              Null,       Null,       true,  true );
+            REGISTER_ALL(TForsakeChaosShortcut,             "forsake_chaos_shortcut",                 Null,       Null,       true,  true );
             REGISTER_ALL(TRemoveChaosCellMailbox,           "remove_chaos_cell_mailbox",              Null,       Null,       true,  true );
             REGISTER_ALL(TGetOrderedTabletSafeTrimRowCount, "get_ordered_tablet_safe_trim_row_count", Null,       Structured, false, false);
             REGISTER_ALL(TGetConnectionOrchidValue,         "get_connection_orchid_value",            Null,       Structured, false, false);

yt/yt/client/driver/internal_commands.cpp

@@ -238,6 +238,20 @@ void TForsakeChaosCoordinator::DoExecute(ICommandContextPtr context)
 
 ////////////////////////////////////////////////////////////////////////////////
 
+void TForsakeChaosShortcut::Register(TRegistrar registrar)
+{
+    registrar.Parameter("coordinator_cell_id", &TThis::CoordinatorCellId_);
+        registrar.Parameter("chaos_object_id", &TThis::ChaosObjectId_);
+}
+
+void TForsakeChaosShortcut::DoExecute(ICommandContextPtr context)
+{
+    WaitFor(context->GetInternalClientOrThrow()->ForsakeChaosShortcut(CoordinatorCellId_, ChaosObjectId_))
+        .ThrowOnError();
+}
+
+////////////////////////////////////////////////////////////////////////////////
+
 void TRemoveChaosCellMailbox::Register(TRegistrar registrar)
 {
     registrar.Parameter("chaos_cell_id", &TThis::ChaosCellId_);

yt/yt/client/driver/internal_commands.h

@@ -192,6 +192,23 @@ class TForsakeChaosCoordinator
 
 ////////////////////////////////////////////////////////////////////////////////
 
+class TForsakeChaosShortcut
+    : public TTypedCommand<NApi::TForsakeChaosShortcutOptions>
+{
+public:
+    REGISTER_YSON_STRUCT_LITE(TForsakeChaosShortcut);
+
+    static void Register(TRegistrar registrar);
+
+private:
+    NHydra::TCellId CoordinatorCellId_;
+    NChaosClient::TChaosObjectId ChaosObjectId_;
+
+    void DoExecute(ICommandContextPtr context) override;
+};
+
+////////////////////////////////////////////////////////////////////////////////
+
 class TRemoveChaosCellMailbox
     : public TTypedCommand<NApi::TRemoveChaosCellMailboxOptions>
 {