jcoffey-dev
7dae9b29fd
Upstream commit: 474dd0229cb20cf513036619781ed97bd8073c3f Enterprise-only files removed or emptied: 63 Enterprise-only snippets removed: 117 in 50 files Dangling module declarations removed: 5 Cargo edits turning enterprise off: 14 Verification: clean Enterprise feature gates left for rebuilt features: 19 in 18 files Produced by tools/fork/strip.py. The full report is in docs/fork/strip-reports/ on main.
66 lines
2.4 KiB
Rust
66 lines
2.4 KiB
Rust
/*
|
|
* SPDX-FileCopyrightText: 2020 Stalwart Labs LLC <[email protected]>
|
|
*
|
|
* SPDX-License-Identifier: AGPL-3.0-only OR LicenseRef-SEL
|
|
*/
|
|
|
|
use super::FdbStore;
|
|
use crate::Store;
|
|
use foundationdb::{Database, api, api::NetworkAutoStop, options::DatabaseOption};
|
|
use parking_lot::Mutex;
|
|
use registry::schema::structs;
|
|
use std::sync::Arc;
|
|
|
|
static FDB_NETWORK: Mutex<Option<NetworkAutoStop>> = Mutex::new(None);
|
|
|
|
impl FdbStore {
|
|
pub async fn open(config: structs::FoundationDbStore) -> Result<Store, String> {
|
|
{
|
|
let mut guard = FDB_NETWORK.lock();
|
|
if guard.is_none() {
|
|
let network = unsafe {
|
|
api::FdbApiBuilder::default()
|
|
.build()
|
|
.map_err(|err| format!("Failed to boot FoundationDB: {err:?}"))?
|
|
.boot()
|
|
.map_err(|err| format!("Failed to boot FoundationDB: {err:?}"))?
|
|
};
|
|
*guard = Some(network);
|
|
}
|
|
}
|
|
|
|
let db = Database::new(config.cluster_file.as_deref())
|
|
.map_err(|err| format!("Failed to create FoundationDB database: {err:?}"))?;
|
|
|
|
if let Some(value) = config.transaction_timeout {
|
|
db.set_option(DatabaseOption::TransactionTimeout(
|
|
value.into_inner().as_millis() as i32,
|
|
))
|
|
.map_err(|err| format!("Failed to set option: {err:?}"))?;
|
|
}
|
|
if let Some(value) = config.transaction_retry_limit {
|
|
db.set_option(DatabaseOption::TransactionRetryLimit(value as i32))
|
|
.map_err(|err| format!("Failed to set option: {err:?}"))?;
|
|
}
|
|
if let Some(value) = config.transaction_retry_delay {
|
|
db.set_option(DatabaseOption::TransactionMaxRetryDelay(
|
|
value.into_inner().as_millis() as i32,
|
|
))
|
|
.map_err(|err| format!("Failed to set option: {err:?}"))?;
|
|
}
|
|
if let Some(value) = config.machine_id {
|
|
db.set_option(DatabaseOption::MachineId(value))
|
|
.map_err(|err| format!("Failed to set option: {err:?}"))?;
|
|
}
|
|
if let Some(value) = config.datacenter_id {
|
|
db.set_option(DatabaseOption::DatacenterId(value))
|
|
.map_err(|err| format!("Failed to set option: {err:?}"))?;
|
|
}
|
|
|
|
Ok(Store::FoundationDb(Arc::new(Self {
|
|
db,
|
|
version: Default::default(),
|
|
})))
|
|
}
|
|
}
|