fix immediate

Author: Enjection

ydb/core/tx/schemeshard/schemeshard__operation_restore_backup_collection.cpp

@@ -333,6 +333,45 @@ class TCreateRestoreOpControlPlane: public TSubOperation {
         }
     }
 
+    TSubOperationState::TPtr SelectStateFunc(TTxState::ETxState state, TOperationContext& context) override {
+        switch(state) {
+        case TTxState::Waiting:
+        case TTxState::Propose:
+            return MakeHolder<TEmptyPropose>(OperationId);
+        case TTxState::CopyTableBarrier:
+            return MakeHolder<TWaitCopyTableBarrier>(OperationId);
+        case TTxState::Done: {
+            // Get target state from transaction and create TDone with it
+            // This will cause TDone::Process to apply the path state change
+            const auto* txState = context.SS->FindTx(OperationId);
+            if (txState && txState->TargetPathTargetState.Defined()) {
+                auto targetState = static_cast<TPathElement::EPathState>(*txState->TargetPathTargetState);
+                return MakeHolder<TDone>(OperationId, targetState);
+            }
+            return MakeHolder<TDone>(OperationId);
+        }
+        default:
+            return nullptr;
+        }
+    }
+
+    void StateDone(TOperationContext& context) override {
+        // When we reach Done state, don't try to advance to Invalid state
+        // Just complete the operation
+        if (GetState() == TTxState::Done) {
+            // Operation is complete, no need to advance state
+            return;
+        }
+        
+        // For other states, use normal state advancement
+        auto nextState = NextState(GetState());
+        SetState(nextState, context);
+        
+        if (nextState != TTxState::Invalid) {
+            context.OnComplete.ActivateTx(OperationId);
+        }
+    }
+
 public:
     using TSubOperation::TSubOperation;
 
@@ -355,6 +394,9 @@ class TCreateRestoreOpControlPlane: public TSubOperation {
 
         // Set the target path ID for coordinator communication
         txState.TargetPathId = bcPath.Base()->PathId;
+        
+        // Set the target state for the backup collection path to outgoing incremental restore
+        txState.TargetPathTargetState = NKikimrSchemeOp::EPathStateOutgoingIncrementalRestore;
 
         auto result = MakeHolder<TProposeResponse>(NKikimrScheme::StatusAccepted, ui64(OperationId.GetTxId()), ui64(schemeshardTabletId));
 
@@ -416,7 +458,7 @@ class TCreateRestoreOpControlPlane: public TSubOperation {
         context.DbChanges.PersistLongIncrementalRestoreOp(op);
 
         // Set initial operation state
-        SetState(NextState());
+        SetState(NextState(), context);
 
         return result;
     }