chore(pegboard): allow configuring reschedule_backoff_max_exponent

Author: NathanFlurry

engine/packages/config/src/config/pegboard.rs

@@ -29,6 +29,12 @@ pub struct Pegboard {
 	///
 	/// **Experimental**
 	pub retry_reset_duration: Option<i64>,
+	/// Maximum exponent for the reschedule backoff calculation.
+	///
+	/// This controls the maximum backoff duration when rescheduling actors.
+	///
+	/// **Experimental**
+	pub reschedule_backoff_max_exponent: Option<usize>,
 }
 
 impl Pegboard {
@@ -47,4 +53,8 @@ impl Pegboard {
 	pub fn retry_reset_duration(&self) -> i64 {
 		self.retry_reset_duration.unwrap_or(10 * 60 * 1000)
 	}
+
+	pub fn reschedule_backoff_max_exponent(&self) -> usize {
+		self.reschedule_backoff_max_exponent.unwrap_or(8)
+	}
 }

engine/packages/pegboard/src/workflows/actor/runtime.rs

@@ -619,6 +619,7 @@ pub async fn reschedule_actor(
 	let mut backoff = reschedule_backoff(
 		state.reschedule_state.retry_count,
 		ctx.config().pegboard().base_retry_timeout(),
+		ctx.config().pegboard().reschedule_backoff_max_exponent(),
 	);
 
 	let (now, reset) = ctx
@@ -736,6 +737,7 @@ async fn compare_retry(ctx: &ActivityCtx, input: &CompareRetryInput) -> Result<(
 		let backoff = reschedule_backoff(
 			input.retry_count,
 			ctx.config().pegboard().base_retry_timeout(),
+			ctx.config().pegboard().reschedule_backoff_max_exponent(),
 		);
 		state.reschedule_ts = Some(now + i64::try_from(backoff.current_duration())?);
 	}
@@ -815,6 +817,10 @@ pub async fn set_complete(ctx: &ActivityCtx, input: &SetCompleteInput) -> Result
 	Ok(())
 }
 
-fn reschedule_backoff(retry_count: usize, base_retry_timeout: usize) -> util::backoff::Backoff {
-	util::backoff::Backoff::new_at(8, None, base_retry_timeout, 500, retry_count)
+fn reschedule_backoff(
+	retry_count: usize,
+	base_retry_timeout: usize,
+	max_exponent: usize,
+) -> util::backoff::Backoff {
+	util::backoff::Backoff::new_at(max_exponent, None, base_retry_timeout, 500, retry_count)
 }

rivetkit-typescript/packages/rivetkit/src/engine-process/mod.ts

@@ -97,6 +97,8 @@ export async function ensureEngineProcess(
 			// order to account for this.
 			RIVET__PEGBOARD__RETRY_RESET_DURATION: "100",
 			RIVET__PEGBOARD__BASE_RETRY_TIMEOUT: "100",
+			// Set max exponent to 1 to have a maximum of base_retry_timeout
+			RIVET__PEGBOARD__RESCHEDULE_BACKOFF_MAX_EXPONENT: "1",
 		},
 	});