use {
crate::{
database_call,
Parameters,
rdfox_api::{
CServer_createFirstLocalServerRole,
CServer_getNumberOfLocalServerRoles,
CServer_startLocalServer,
CServer_stopLocalServer,
CServerConnection,
CServerConnection_newServerConnection,
},
RoleCreds,
server_connection::ServerConnection,
},
ekg_namespace::consts::LOG_TARGET_DATABASE,
std::{
ffi::CString,
ptr,
sync::{
Arc,
atomic::{AtomicBool, Ordering},
},
},
};
#[derive(Debug)]
pub struct Server {
default_role_creds: RoleCreds,
running: AtomicBool,
}
impl Drop for Server {
fn drop(&mut self) { self.stop(); }
}
impl std::fmt::Display for Server {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
write!(f, "server {self:p}", )
}
}
impl Server {
pub fn is_running(&self) -> bool { self.running.load(Ordering::Relaxed) }
pub fn start(role_creds: RoleCreds) -> Result<Arc<Self>, ekg_error::Error> {
Self::start_with_parameters(role_creds, None)
}
pub fn start_with_parameters(
role_creds: RoleCreds,
params: Option<Parameters>,
) -> Result<Arc<Self>, ekg_error::Error> {
if let Some(params) = params {
#[cfg(feature = "rdfox-7-0")]
{
let mut number_of_data_stores_in_server: usize = 0;
database_call!(
"Starting a local RDFFox server",
CServer_startLocalServer(params.inner.cast_const(), &mut number_of_data_stores_in_server)
)?;
}
#[cfg(not(feature = "rdfox-7-0"))]
database_call!(
"Starting a local RDFFox server",
CServer_startLocalServer(params.inner.cast_const())
)?;
} else {
let params = Parameters::empty()?;
#[cfg(feature = "rdfox-7-0")]
{
let mut number_of_data_stores_in_server = 0usize;
database_call!(
"Starting a local RDFFox server with default parameters",
CServer_startLocalServer(params.inner.cast_const(), &mut number_of_data_stores_in_server)
)?;
}
#[cfg(not(feature = "rdfox-7-0"))]
database_call!(
"Starting a local RDFFox server with default parameters",
CServer_startLocalServer(params.inner.cast_const())
)?;
};
let server = Server {
default_role_creds: role_creds,
running: AtomicBool::new(true),
};
if server.get_number_of_local_server_roles()? == 0 {
server.create_role(&server.default_role_creds)?;
}
tracing::debug!(
target: LOG_TARGET_DATABASE,
"Local RDFox server has been started"
);
Ok(Arc::new(server))
}
pub fn create_role(&self, role_creds: &RoleCreds) -> Result<(), ekg_error::Error> {
let c_role_name = CString::new(role_creds.role_name.as_str()).unwrap();
let c_password = CString::new(role_creds.password.as_str()).unwrap();
let msg = format!(
"Creating server role named [{}]",
role_creds.role_name
);
database_call!(
msg.as_str(),
CServer_createFirstLocalServerRole(c_role_name.as_ptr(), c_password.as_ptr())
)
}
pub fn get_number_of_local_server_roles(&self) -> Result<u16, ekg_error::Error> {
let mut number_of_roles = 0_usize;
database_call!(
"Getting the number of local server roles",
CServer_getNumberOfLocalServerRoles(&mut number_of_roles)
)?;
Ok(number_of_roles as u16)
}
pub fn connection_with_default_role(
self: &Arc<Self>,
) -> Result<Arc<ServerConnection>, ekg_error::Error> {
let role_creds = &self.default_role_creds;
self.connection(role_creds.clone())
}
pub fn connection(
self: &Arc<Self>,
role_creds: RoleCreds,
) -> Result<Arc<ServerConnection>, ekg_error::Error> {
let c_role_name = CString::new(role_creds.role_name.as_str()).unwrap();
let c_password = CString::new(role_creds.password.as_str()).unwrap();
let mut server_connection_ptr: *mut CServerConnection = ptr::null_mut();
database_call!(
"Creating a server connection",
CServerConnection_newServerConnection(
c_role_name.as_ptr(),
c_password.as_ptr(),
&mut server_connection_ptr,
)
)?;
if server_connection_ptr.is_null() {
tracing::error!(
target: LOG_TARGET_DATABASE,
"Could not establish connection to {self}"
);
return Err(ekg_error::Error::CouldNotConnectToServer);
}
Ok(Arc::new(ServerConnection::new(
role_creds,
self.clone(),
server_connection_ptr,
)))
}
pub fn stop(&mut self) {
*self.running.get_mut() = false;
tracing::trace!(
target: LOG_TARGET_DATABASE,
server = format!("{self:p}"),
"Stopping local RDFox server"
);
unsafe {
CServer_stopLocalServer();
}
tracing::trace!(
target: LOG_TARGET_DATABASE,
server = format!("{self:p}"),
"Stopped local RDFox server"
);
}
}