82 lines
1.8 KiB
Rust
82 lines
1.8 KiB
Rust
use std::sync::{Arc, RwLock};
|
|
|
|
use conduwuit::{utils, Result};
|
|
use database::{Database, Deserialized, Map};
|
|
|
|
pub struct Data {
|
|
global: Arc<Map>,
|
|
counter: RwLock<u64>,
|
|
pub(super) db: Arc<Database>,
|
|
}
|
|
|
|
const COUNTER: &[u8] = b"c";
|
|
|
|
impl Data {
|
|
pub(super) fn new(args: &crate::Args<'_>) -> Self {
|
|
let db = &args.db;
|
|
Self {
|
|
global: db["global"].clone(),
|
|
counter: RwLock::new(
|
|
Self::stored_count(&db["global"]).expect("initialized global counter"),
|
|
),
|
|
db: args.db.clone(),
|
|
}
|
|
}
|
|
|
|
pub fn next_count(&self) -> Result<u64> {
|
|
let _cork = self.db.cork();
|
|
let mut lock = self.counter.write().expect("locked");
|
|
let counter: &mut u64 = &mut lock;
|
|
debug_assert!(
|
|
*counter == Self::stored_count(&self.global).expect("database failure"),
|
|
"counter mismatch"
|
|
);
|
|
|
|
*counter = counter
|
|
.checked_add(1)
|
|
.expect("counter must not overflow u64");
|
|
|
|
self.global.insert(COUNTER, counter.to_be_bytes());
|
|
|
|
Ok(*counter)
|
|
}
|
|
|
|
#[inline]
|
|
pub fn current_count(&self) -> u64 {
|
|
let lock = self.counter.read().expect("locked");
|
|
let counter: &u64 = &lock;
|
|
debug_assert!(
|
|
*counter == Self::stored_count(&self.global).expect("database failure"),
|
|
"counter mismatch"
|
|
);
|
|
|
|
*counter
|
|
}
|
|
|
|
fn stored_count(global: &Arc<Map>) -> Result<u64> {
|
|
global
|
|
.get_blocking(COUNTER)
|
|
.as_deref()
|
|
.map_or(Ok(0_u64), utils::u64_from_bytes)
|
|
}
|
|
|
|
pub async fn database_version(&self) -> u64 {
|
|
self.global
|
|
.get(b"version")
|
|
.await
|
|
.deserialized()
|
|
.unwrap_or(0)
|
|
}
|
|
|
|
#[inline]
|
|
pub fn bump_database_version(&self, new_version: u64) -> Result<()> {
|
|
self.global.raw_put(b"version", new_version);
|
|
Ok(())
|
|
}
|
|
|
|
#[inline]
|
|
pub fn backup(&self) -> Result { self.db.db.backup() }
|
|
|
|
#[inline]
|
|
pub fn backup_list(&self) -> Result<String> { self.db.db.backup_list() }
|
|
}
|