reifydb-sub-flow 0.4.5

Flow subsystem for stream processing and data flows
Documentation 
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

// SPDX-License-Identifier: Apache-2.0
// Copyright (c) 2025 ReifyDB

//! Tracks the latest CDC version where each primitive had changes.

use std::{collections::BTreeMap, sync::Arc};

use reifydb_core::{common::CommitVersion, interface::catalog::schema::SchemaId};
use reifydb_runtime::sync::rwlock::RwLock;

/// Tracks the latest CDC version for each primitive (table/view/flow).
///
/// This is used to compute flow lag by comparing a flow's current version
/// to the latest version where its sources had changes.
pub struct SchemaVersionTracker {
	versions: Arc<RwLock<BTreeMap<SchemaId, CommitVersion>>>,
}

impl SchemaVersionTracker {
	pub fn new() -> Self {
		Self {
			versions: Arc::new(RwLock::new(BTreeMap::new())),
		}
	}

	/// Update the latest version for a primitive.
	pub fn update(&self, object_id: SchemaId, version: CommitVersion) {
		let mut versions = self.versions.write();
		versions.entry(object_id)
			.and_modify(|v| {
				if version.0 > v.0 {
					*v = version;
				}
			})
			.or_insert(version);
	}

	/// Get all tracked primitive versions.
	pub fn all(&self) -> BTreeMap<SchemaId, CommitVersion> {
		let versions = self.versions.read();
		versions.clone()
	}
}

impl Default for SchemaVersionTracker {
	fn default() -> Self {
		Self::new()
	}
}