refactor(rivetkit-core): remove prevent_sleep field and deprecate set_prevent_sleep to no-op

Author: NathanFlurry

rivetkit-rust/packages/rivetkit-core/src/actor/context.rs

@@ -128,7 +128,6 @@ pub(crate) struct ActorContextInner {
 	pub(super) sleep: SleepState,
 	activity: ActivityState,
 	pending_disconnect_count: AtomicUsize,
-	prevent_sleep: AtomicBool,
 	sleep_requested: AtomicBool,
 	destroy_requested: AtomicBool,
 	destroy_completed: AtomicBool,
@@ -291,7 +290,6 @@ impl ActorContext {
 			sleep,
 			activity: ActivityState::default(),
 			pending_disconnect_count: AtomicUsize::new(0),
-			prevent_sleep: AtomicBool::new(false),
 			sleep_requested: AtomicBool::new(false),
 			destroy_requested: AtomicBool::new(false),
 			destroy_completed: AtomicBool::new(false),
@@ -447,20 +445,15 @@ impl ActorContext {
 		self.0.abort_signal.is_cancelled()
 	}
 
-	/// Prevents the actor from entering sleep while enabled.
-	///
-	/// Shutdown drain loops continue polling until this is cleared or the
-	/// configured grace deadline is reached.
-	pub fn set_prevent_sleep(&self, enabled: bool) {
-		let previous = self.0.prevent_sleep.swap(enabled, Ordering::SeqCst);
-		if previous != enabled {
-			self.notify_prevent_sleep_changed();
-			self.reset_sleep_timer();
-		}
-	}
+	/// Deprecated no-op. Use `keep_awake` to hold the actor awake for the
+	/// duration of a future, or `wait_until` to keep work alive across the
+	/// sleep grace period. Retained only for NAPI bridge compatibility.
+	#[deprecated(note = "no-op: use `keep_awake` or `wait_until` instead")]
+	pub fn set_prevent_sleep(&self, _enabled: bool) {}
 
+	#[deprecated(note = "no-op: always returns false")]
 	pub fn prevent_sleep(&self) -> bool {
-		self.0.prevent_sleep.load(Ordering::SeqCst)
+		false
 	}
 
 	pub fn wait_until(&self, future: impl Future<Output = ()> + Send + 'static) {

rivetkit-rust/packages/rivetkit-core/src/actor/sleep.rs

@@ -49,7 +49,6 @@ pub(crate) struct SleepState {
 pub(crate) enum CanSleep {
 	Yes,
 	NotReady,
-	PreventSleep,
 	NoSleep,
 	ActiveHttpRequests,
 	ActiveKeepAwake,
@@ -227,9 +226,6 @@ impl ActorContext {
 		{
 			return CanSleep::NotReady;
 		}
-		if self.prevent_sleep() {
-			return CanSleep::PreventSleep;
-		}
 		if config.no_sleep {
 			return CanSleep::NoSleep;
 		}
@@ -268,7 +264,6 @@ impl ActorContext {
 			&& self.active_http_request_count() == 0
 			&& self.websocket_callback_count() == 0
 			&& self.pending_disconnect_count() == 0
-			&& !self.prevent_sleep()
 	}
 
 	/// Spawn the fallback sleep timer used by `ActorContext`s that are not
@@ -379,7 +374,7 @@ impl ActorContext {
 
 			let shutdown_count = self.shutdown_task_count();
 			let websocket_count = self.websocket_callback_count();
-			if shutdown_count == 0 && websocket_count == 0 && !self.prevent_sleep() {
+			if shutdown_count == 0 && websocket_count == 0 {
 				return true;
 			}
 
@@ -520,11 +515,6 @@ impl ActorContext {
 			.unwrap_or(0)
 	}
 
-	pub(crate) fn notify_prevent_sleep_changed(&self) {
-		self.0.sleep.work.prevent_sleep_notify.notify_waiters();
-		self.reset_sleep_timer();
-	}
-
 	pub(crate) fn sleep_activity_notify(&self) -> Arc<Notify> {
 		self.0.sleep.work.activity_notify.clone()
 	}

rivetkit-rust/packages/rivetkit-core/src/actor/work_registry.rs

@@ -22,7 +22,6 @@ pub(crate) struct WorkRegistry {
 	/// counters like the envoy HTTP request counter whose increments happen
 	/// outside rivetkit-core.
 	pub(crate) activity_notify: Arc<Notify>,
-	pub(crate) prevent_sleep_notify: Arc<Notify>,
 	pub(crate) teardown_started: AtomicBool,
 }
 
@@ -45,7 +44,6 @@ impl WorkRegistry {
 			shutdown_tasks: Mutex::new(JoinSet::new()),
 			idle_notify,
 			activity_notify: Arc::new(Notify::new()),
-			prevent_sleep_notify: Arc::new(Notify::new()),
 			teardown_started: AtomicBool::new(false),
 		}
 	}

rivetkit-rust/packages/rivetkit/src/context.rs

@@ -95,11 +95,15 @@ impl<A: Actor> Ctx<A> {
 		self.inner.destroy();
 	}
 
+	#[deprecated(note = "no-op: use `keep_awake` or `wait_until` instead")]
 	pub fn set_prevent_sleep(&self, enabled: bool) {
+		#[allow(deprecated)]
 		self.inner.set_prevent_sleep(enabled);
 	}
 
+	#[deprecated(note = "no-op: always returns false")]
 	pub fn prevent_sleep(&self) -> bool {
+		#[allow(deprecated)]
 		self.inner.prevent_sleep()
 	}
 

rivetkit-typescript/packages/rivetkit-napi/src/actor_context.rs

@@ -451,11 +451,13 @@ impl ActorContext {
 	}
 
 	#[napi]
+	#[allow(deprecated)]
 	pub fn set_prevent_sleep(&self, prevent_sleep: bool) {
 		self.inner.set_prevent_sleep(prevent_sleep);
 	}
 
 	#[napi]
+	#[allow(deprecated)]
 	pub fn prevent_sleep(&self) -> bool {
 		self.inner.prevent_sleep()
 	}