Simplify TSubOperation (#10211)

Author: Enjection

ydb/core/tx/schemeshard/schemeshard__operation_part.cpp

@@ -89,7 +89,7 @@ static TString LogMessage(const TString& ev, TOperationContext& context, bool ig
     } \
     \
     bool TSubOperation::HandleReply(TEvType::TPtr& ev, TOperationContext& context) { \
-        return Progress(context, &ISubOperationState::HandleReply, ev, context);     \
+        return Progress(context, &ISubOperationState::HandleReply, ev);     \
     }
 
     SCHEMESHARD_INCOMING_EVENTS(DefaultHandleReply)

ydb/core/tx/schemeshard/schemeshard__operation_part.h

@@ -278,7 +278,7 @@ class TSubOperation: public TSubOperationBase {
     }
 
     bool ProgressState(TOperationContext& context) override {
-        return Progress(context, &ISubOperationState::ProgressState, context);
+        return Progress(context, &ISubOperationState::ProgressState);
     }
 
     #define DefaultHandleReply(TEvType, ...) \
@@ -292,9 +292,9 @@ class TSubOperation: public TSubOperationBase {
     using TFunc = bool(ISubOperationState::*)(Args...);
 
     template <typename... Args>
-    bool Progress(TOperationContext& context, TFunc<Args...> func, Args&&... args) {
+    bool Progress(TOperationContext& context, TFunc<Args..., TOperationContext&> func, Args&&... args) {
         Y_ABORT_UNLESS(StateFunc);
-        const bool isDone = std::invoke(func, StateFunc.Get(), std::forward<Args>(args)...);
+        const bool isDone = std::invoke(func, StateFunc.Get(), std::forward<Args>(args)..., context);
         if (isDone) {
             StateDone(context);
         }