reifydb-runtime 0.10.0

Runtime infrastructure for ReifyDB
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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70

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

use std::fmt;

use reifydb_value::{
	Result,
	error::{Diagnostic, Error},
	fragment::Fragment,
};

pub trait Shutdown {
	fn shutdown(&self) -> Result<()>;
}

pub fn shutdown_failed(subsystem: impl fmt::Display, reason: impl fmt::Display) -> Diagnostic {
	Diagnostic {
		code: "SUB_004".to_string(),
		rql: None,
		message: format!("{} subsystem shutdown failed: {}", subsystem, reason),
		column: None,
		fragment: Fragment::None,
		label: None,
		help: None,
		notes: vec![],
		cause: None,
		operator_chain: None,
	}
}

pub struct ShutdownErrors {
	owner: &'static str,
	failed: Vec<String>,
}

impl ShutdownErrors {
	pub fn new(owner: &'static str) -> Self {
		Self {
			owner,
			failed: Vec::new(),
		}
	}

	pub fn step(&mut self, part: &'static str, result: Result<()>) {
		let Err(err) = result else {
			return;
		};
		let diagnostic = err.diagnostic();
		if diagnostic.code == "SUB_004" {
			self.failed.extend(diagnostic.notes.into_iter().map(|note| format!("{part}/{note}")));
		} else {
			self.failed.push(format!("{part}: {}", diagnostic.message));
		}
	}

	pub fn fail(&mut self, part: &'static str, reason: impl fmt::Display) {
		self.failed.push(format!("{part}: {reason}"));
	}

	pub fn finish(self) -> Result<()> {
		if self.failed.is_empty() {
			return Ok(());
		}
		let reason = format!("{} parts failed", self.failed.len());
		Err(Error(Box::new(Diagnostic {
			notes: self.failed,
			..shutdown_failed(self.owner, reason)
		})))
	}
}