Add force_commit_state

Author: snawaz

magicblock-committor-service/src/tasks/args_task.rs

@@ -88,9 +88,15 @@ impl BaseTask for ArgsTask {
         self: Box<Self>,
     ) -> Result<Box<dyn BaseTask>, Box<dyn BaseTask>> {
         match self.task_type {
-            ArgsTaskType::Commit(ref value) if value.is_commit_diff() => {
-                // We do not currently support executing CommitDiff as BufferTask
-                Err(self)
+            ArgsTaskType::Commit(mut value) if value.is_commit_diff() => {
+                //TODO (snawaz): We do not currently support executing CommitDiff as BufferTask, which is why we're forcing CommitTask to use CommitState before converting this task into BufferTask
+                // Once CommitDiff is supported by BufferTask, we do not have to
+                // force_commit_state.
+
+                value.force_commit_state();
+                Ok(Box::new(BufferTask::new_preparation_required(
+                    BufferTaskType::Commit(value),
+                )))
             }
             ArgsTaskType::Commit(value) => {
                 Ok(Box::new(BufferTask::new_preparation_required(

magicblock-committor-service/src/tasks/mod.rs

@@ -110,6 +110,7 @@ pub struct CommitTask {
     pub allow_undelegation: bool,
     pub committed_account: CommittedAccount,
     base_account: Option<Account>,
+    force_commit_state: bool,
 }
 
 impl CommitTask {
@@ -152,15 +153,21 @@ impl CommitTask {
             allow_undelegation,
             committed_account,
             base_account: fetched_account,
+            force_commit_state: false,
         }
     }
 
     pub fn is_commit_diff(&self) -> bool {
-        self.committed_account.account.data.len()
-            > CommitTask::COMMIT_STATE_SIZE_THRESHOLD
+        !self.force_commit_state
+            && self.committed_account.account.data.len()
+                > CommitTask::COMMIT_STATE_SIZE_THRESHOLD
             && self.base_account.is_some()
     }
 
+    pub fn force_commit_state(&mut self) {
+        self.force_commit_state = true;
+    }
+
     pub fn create_commit_ix(&self, validator: &Pubkey) -> Instruction {
         if let Some(fetched_account) = self.base_account.as_ref() {
             self.create_commit_diff_ix(validator, fetched_account)
@@ -190,6 +197,13 @@ impl CommitTask {
         validator: &Pubkey,
         fetched_account: &Account,
     ) -> Instruction {
+        if self.force_commit_state {
+            println!(
+                "force_commit_state is true, so call create_commit_state_ix"
+            );
+            return self.create_commit_state_ix(validator);
+        }
+
         let args = CommitDiffArgs {
             nonce: self.commit_id,
             lamports: self.committed_account.account.lamports,