rivetkit-core 2.3.0-rc.12

Core runtime primitives for RivetKit actor hosts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92

use std::fmt;

use anyhow::Result;

use crate::ActorConfig;
use crate::actor::lifecycle_hooks::ActorStart;
use crate::runtime::RuntimeBoxFuture;

#[cfg(feature = "wasm-runtime")]
pub type ActorEntryFn = dyn Fn(ActorStart) -> RuntimeBoxFuture<Result<()>>;

#[cfg(not(feature = "wasm-runtime"))]
pub type ActorEntryFn = dyn Fn(ActorStart) -> RuntimeBoxFuture<Result<()>> + Send + Sync;

/// Runtime extension point for building actor receive loops.
pub struct ActorFactory {
	config: ActorConfig,
	entry: Box<ActorEntryFn>,
	manual_startup_ready: bool,
}

#[cfg(feature = "wasm-runtime")]
unsafe impl Send for ActorFactory {}

#[cfg(feature = "wasm-runtime")]
unsafe impl Sync for ActorFactory {}

impl ActorFactory {
	pub fn new<F>(config: ActorConfig, entry: F) -> Self
	where
		F: ActorEntry,
	{
		Self {
			config,
			entry: Box::new(entry),
			manual_startup_ready: false,
		}
	}

	/// Builds a factory whose runtime will explicitly signal `startup_ready`
	/// after its own startup preamble finishes.
	pub fn new_with_manual_startup_ready<F>(config: ActorConfig, entry: F) -> Self
	where
		F: ActorEntry,
	{
		Self {
			config,
			entry: Box::new(entry),
			manual_startup_ready: true,
		}
	}

	pub fn config(&self) -> &ActorConfig {
		&self.config
	}

	pub(crate) fn requires_manual_startup_ready(&self) -> bool {
		self.manual_startup_ready
	}

	pub async fn start(&self, start: ActorStart) -> Result<()> {
		(self.entry)(start).await
	}
}

#[cfg(feature = "wasm-runtime")]
pub trait ActorEntry: Fn(ActorStart) -> RuntimeBoxFuture<Result<()>> + 'static {}

#[cfg(feature = "wasm-runtime")]
impl<F> ActorEntry for F where F: Fn(ActorStart) -> RuntimeBoxFuture<Result<()>> + 'static {}

#[cfg(not(feature = "wasm-runtime"))]
pub trait ActorEntry:
	Fn(ActorStart) -> RuntimeBoxFuture<Result<()>> + Send + Sync + 'static
{
}

#[cfg(not(feature = "wasm-runtime"))]
impl<F> ActorEntry for F where
	F: Fn(ActorStart) -> RuntimeBoxFuture<Result<()>> + Send + Sync + 'static
{
}

impl fmt::Debug for ActorFactory {
	fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
		f.debug_struct("ActorFactory")
			.field("config", &self.config)
			.field("manual_startup_ready", &self.manual_startup_ready)
			.field("entry", &"<boxed entry>")
			.finish()
	}
}