reifydb-sub-flow 0.5.0

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

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

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

use reifydb_core::{common::CommitVersion, interface::catalog::shape::ShapeId};
use reifydb_runtime::sync::rwlock::RwLock;

pub struct ShapeVersionTracker {
	versions: Arc<RwLock<BTreeMap<ShapeId, CommitVersion>>>,
}

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

	pub fn update(&self, object_id: ShapeId, 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);
	}

	pub fn all(&self) -> BTreeMap<ShapeId, CommitVersion> {
		let versions = self.versions.read();
		versions.clone()
	}
}

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