fix(pb): get new actor ids working e2e

Author: MasterPtato

packages/common/config/src/config/server/rivet/mod.rs

@@ -60,6 +60,11 @@ pub struct Rivet {
 	#[serde(default = "Rivet::default_namespace")]
 	pub namespace: String,
 
+	/// If specified, will use this as the default cluster ID.
+	///
+	/// This will have no effect if applied after the cluster has first ran.
+	pub instance_id: Option<Uuid>,
+
 	/// If specified, will use this as the default cluster ID.
 	///
 	/// This will have no effect if applied after the cluster has first ran.
@@ -139,6 +144,7 @@ impl Default for Rivet {
 	fn default() -> Rivet {
 		Self {
 			namespace: Self::default_namespace(),
+			instance_id: None,
 			default_cluster_id: None,
 			clusters: None,
 			provision: None,
@@ -171,7 +177,7 @@ impl Rivet {
 	}
 }
 
-impl Rivet {
+impl Rivet {	
 	pub fn default_cluster_id(&self) -> GlobalResult<Uuid> {
 		if let Some(default_cluster_id) = self.default_cluster_id {
 			ensure!(
@@ -184,9 +190,7 @@ impl Rivet {
 				// Return default development clusters
 				AccessKind::Development => Ok(default_dev_cluster::CLUSTER_ID),
 				// No cluster configured
-				AccessKind::Public | AccessKind::Private => {
-					bail!("`default_cluster_id` not configured")
-				}
+				AccessKind::Public | AccessKind::Private => bail!("`default_cluster_id` not configured"),
 			}
 		}
 	}

packages/core/services/cluster/src/ops/datacenter/get_for_label.rs

@@ -57,7 +57,7 @@ async fn get_dcs(ctx: OperationCtx, labels: Vec<u16>) -> GlobalResult<Vec<Datace
 		FROM db_cluster.datacenters@datacenter_label_idx
 		WHERE label = ANY($1)
 		",
-		labels.into_iter().map(|x| x as i64).collect::<Vec<_>>(),
+		labels.into_iter().map(|x| x.to_be_bytes()).collect::<Vec<_>>(),
 	)
 	.await?;
 

packages/core/services/cluster/src/workflows/server/install/install_scripts/components/mod.rs

@@ -203,7 +203,7 @@ pub mod lz4 {
 		indoc!(
 			r#"
 			echo 'Downloading lz4'
-			curl -L https://releases.rivet.gg/tools/lz4/1.10.0/debian11-amd64/lz4 -o /usr/local/bin/lz4
+			curl -Lfo /usr/local/bin/lz4 https://releases.rivet.gg/tools/lz4/1.10.0/debian11-amd64/lz4
 			chmod +x /usr/local/bin/lz4
 			"#
 		)
@@ -224,7 +224,7 @@ pub mod umoci {
 		indoc!(
 			r#"
 			echo 'Downloading umoci'
-			curl -Lf -o /usr/bin/umoci "https://github.com/opencontainers/umoci/releases/download/v0.4.7/umoci.amd64"
+			curl -Lfo /usr/bin/umoci "https://github.com/opencontainers/umoci/releases/download/v0.4.7/umoci.amd64"
 			chmod +x /usr/bin/umoci
 			"#
 		).to_string()
@@ -238,7 +238,7 @@ pub mod cni {
 		indoc!(
 			r#"
 			echo 'Downloading cnitool'
-			curl -Lf -o /usr/bin/cnitool "https://github.com/rivet-gg/cni/releases/download/v1.1.2-build3/cnitool"
+			curl -Lfo /usr/bin/cnitool "https://github.com/rivet-gg/cni/releases/download/v1.1.2-build3/cnitool"
 			chmod +x /usr/bin/cnitool
 			"#
 		).to_string()

packages/core/services/cluster/src/workflows/server/install/install_scripts/components/rivet/guard.rs

@@ -36,6 +36,7 @@ pub fn configure(config: &rivet_config::Config) -> GlobalResult<String> {
 			tls: server_config.tls.clone(),
 			rivet: Rivet {
 				namespace: server_config.rivet.namespace.clone(),
+				instance_id: server_config.rivet.instance_id,
 				auth: server_config.rivet.auth.clone(),
 				api_public: ApiPublic {
 					// NOTE: Templated later

packages/core/services/cluster/src/workflows/server/install/install_scripts/components/rivet/worker.rs

@@ -36,6 +36,7 @@ pub fn configure(config: &rivet_config::Config) -> GlobalResult<String> {
 			tls: server_config.tls.clone(),
 			rivet: Rivet {
 				namespace: server_config.rivet.namespace.clone(),
+				instance_id: server_config.rivet.instance_id,
 				clusters: Some({
 					let mut clusters = HashMap::new();
 

packages/core/services/dynamic-config/db/dynamic-config/migrations/20250523120501_instance_id.down.sql

@@ -0,0 +1,2 @@
+ALTER TABLE config
+	RENAME COLUMN cluster_id TO rivet_instance_id;

packages/core/services/dynamic-config/db/dynamic-config/migrations/20250523120501_instance_id.up.sql

@@ -1,57 +1,57 @@
 use chirp_workflow::prelude::*;
 use tokio::sync::OnceCell;
 
-// The cluster ID will never change, so store it in memory.
-static CLUSTER_ID_ONCE: OnceCell<Uuid> = OnceCell::const_new();
+// The instance ID will never change, so store it in memory.
+static INSTANCE_ID_ONCE: OnceCell<Uuid> = OnceCell::const_new();
 
 #[derive(Debug, Default)]
 pub struct Input {}
 
 #[derive(Debug)]
 pub struct Output {
-	pub cluster_id: Uuid,
+	pub instance_id: Uuid,
 }
 
 #[operation]
-pub async fn get_cluster_id(ctx: &OperationCtx, input: &Input) -> GlobalResult<Output> {
+pub async fn get_config(ctx: &OperationCtx, input: &Input) -> GlobalResult<Output> {
 	// IMPORTANT: This is not the same as the cluster ID from the `cluster` package. This is used
-	// for unqiuely identifying the entire Rivet cluster.
+	// for uniquely identifying the entire Rivet cluster.
 
-	// Pick a cluster ID to insert if none exists. If this is specified in the config. fall back to
+	// Pick an instance ID to insert if none exists. If this is specified in the config. fall back to
 	// this.
-	let default_cluster_id =
-		if let Some(cluster_id) = ctx.config().server()?.rivet.default_cluster_id {
-			cluster_id
+	let default_instance_id =
+		if let Some(instance_id) = ctx.config().server()?.rivet.instance_id {
+			instance_id
 		} else {
 			Uuid::new_v4()
 		};
 
-	let cluster_id = CLUSTER_ID_ONCE
+	let instance_id = INSTANCE_ID_ONCE
 		.get_or_try_init(|| async {
 			sql_fetch_one!(
 				[ctx, (Uuid,)]
 				"
 				WITH new_row AS (
-					INSERT INTO db_dynamic_config.config (id, cluster_id)
+					INSERT INTO db_dynamic_config.config (id, rivet_instance_id)
 					VALUES (1, $1)
 					ON CONFLICT (id) DO NOTHING
-					RETURNING cluster_id
+					RETURNING rivet_instance_id
 				)
-				SELECT cluster_id 
+				SELECT rivet_instance_id 
 				FROM new_row
 				UNION ALL
-				SELECT cluster_id 
+				SELECT rivet_instance_id 
 				FROM db_dynamic_config.config
 				WHERE NOT EXISTS (SELECT 1 FROM new_row)
 				",
-				default_cluster_id
+				default_instance_id
 			)
 			.await
 			.map(|x| x.0)
 		})
 		.await?;
 
 	Ok(Output {
-		cluster_id: *cluster_id,
+		instance_id: *instance_id,
 	})
 }

packages/core/services/dynamic-config/src/ops/get_config.rs

@@ -41,22 +41,22 @@ pub async fn run_from_env(
 		(),
 	);
 
-	// Get the cluster ID
-	let cluster_id = chirp_workflow::compat::op(&ctx, dynamic_config::ops::get_config::Input {})
+	// Get the rivet instance ID
+	let instance_id = chirp_workflow::compat::op(&ctx, dynamic_config::ops::get_config::Input {})
 		.await?
-		.cluster_id;
+		.instance_id;
 
 	// Build events
 	let mut events = Vec::new();
-	let distinct_id = format!("cluster:{cluster_id}");
+	let distinct_id = format!("instance:{instance_id}");
 
 	// Send beacon
-	let mut event = async_posthog::Event::new("cluster_beacon", &distinct_id);
-	event.insert_prop("$groups", json!({ "cluster": cluster_id }))?;
+	let mut event = async_posthog::Event::new("instance_beacon", &distinct_id);
+	event.insert_prop("$groups", json!({ "instance": instance_id }))?;
 	event.insert_prop(
 		"$set",
 		json!({
-			"cluster_id": cluster_id,
+			"instance_id": instance_id,
 			"os": os_report(),
 			"source_hash": rivet_env::source_hash(),
 			"config": get_config_data(&ctx)?,
@@ -66,10 +66,10 @@ pub async fn run_from_env(
 	)?;
 	events.push(event);
 
-	// Add cluster identification data
+	// Add instance identification data
 	let mut event = async_posthog::Event::new("$groupidentify", &distinct_id);
-	event.insert_prop("$group_type", "cluster")?;
-	event.insert_prop("$group_key", cluster_id)?;
+	event.insert_prop("$group_type", "instance")?;
+	event.insert_prop("$group_key", instance_id)?;
 	event.insert_prop(
 		"$group_set",
 		json!({
@@ -89,7 +89,7 @@ pub async fn run_from_env(
 	Ok(())
 }
 
-/// Returns information about the operating system running the cluster.
+/// Returns information about the operating system running the instance.
 ///
 /// This helps Rivet diagnose crash reports to easily pinpoint if issues are
 /// coming from a specific operating system.
@@ -190,9 +190,9 @@ struct ResourceAggregateRow {
 
 // /// Returns information about the pegboard configuration.
 // ///
-// /// This is helpful for diagnosing issues with the self-hosted clusters under
-// /// load. e.g. if a cluster is running on constraint resources (see os_report),
-// /// does the cluster configuration affect it?
+// /// This is helpful for diagnosing issues with the self-hosted instances under
+// /// load. e.g. if a instance is running on constraint resources (see os_report),
+// /// does the instance configuration affect it?
 // async fn get_pegboard_data(ctx: &OperationContext<()>) -> GlobalResult<serde_json::Value> {
 // 	use pegboard::protocol::ClientFlavor;
 

packages/core/services/telemetry/standalone/beacon/src/lib.rs

@@ -6,11 +6,13 @@ authors = ["Rivet Gaming, LLC <[email protected]>"]
 license = "Apache-2.0"
 
 [dependencies]
+anyhow = "1.0"
 bytes = "1.0"
 futures-util = "0.3"
 http = "0.2"
+serde = { version = "1.0", features = ["derive"] }
 serde_json = "1.0"
 tokio = { version = "1.40", features = ["full",] }
-tokio-tungstenite = "0.23.1"
+tokio-util = "0.7"
 uuid = { version = "1", features = ["v4", "serde"] }
 warp = "0.3.7"