openethereum/db/src/lib.rs.in

// Copyright 2015-2017 Parity Technologies (UK) Ltd.
// This file is part of Parity.

// Parity is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.

// Parity is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with Parity.  If not, see <http://www.gnu.org/licenses/>.

extern crate ethcore_ipc as ipc;
extern crate rocksdb;
extern crate ethcore_devtools as devtools;
extern crate semver;
extern crate ethcore_ipc_nano as nanoipc;
extern crate nanomsg;
extern crate crossbeam;
extern crate ethcore_util as util;

pub mod database;
pub mod traits;

pub use traits::{DatabaseService, DBTransaction, Error};
pub use database::{Database, DatabaseClient, DatabaseIterator};

use std::sync::Arc;
use std::sync::atomic::*;
use std::path::PathBuf;

pub type DatabaseNanoClient = DatabaseClient<::nanomsg::Socket>;
pub type DatabaseConnection = nanoipc::GuardedSocket<DatabaseNanoClient>;

#[derive(Debug)]
pub enum ServiceError {
	Io(std::io::Error),
	Socket(nanoipc::SocketError),
}

impl std::convert::From<std::io::Error> for ServiceError {
	fn from(io_error: std::io::Error) -> ServiceError { ServiceError::Io(io_error) }
}

impl std::convert::From<nanoipc::SocketError> for ServiceError {
	fn from(socket_error: nanoipc::SocketError) -> ServiceError { ServiceError::Socket(socket_error) }
}

pub fn blocks_service_url(db_path: &str) -> Result<String, std::io::Error> {
	let mut path = PathBuf::from(db_path);
	::std::fs::create_dir_all(db_path)?;
	path.push("blocks.ipc");
	Ok(format!("ipc://{}", path.to_str().unwrap()))
}

pub fn extras_service_url(db_path: &str) -> Result<String, ::std::io::Error> {
	let mut path = PathBuf::from(db_path);
	::std::fs::create_dir_all(db_path)?;
	path.push("extras.ipc");
	Ok(format!("ipc://{}", path.to_str().unwrap()))
}

pub fn blocks_client(db_path: &str) -> Result<DatabaseConnection, ServiceError> {
	let url = blocks_service_url(db_path)?;
	let client = nanoipc::generic_client::<DatabaseClient<_>>(&url)?;
	Ok(client)
}

pub fn extras_client(db_path: &str) -> Result<DatabaseConnection, ServiceError> {
	let url = extras_service_url(db_path)?;
	let client = nanoipc::generic_client::<DatabaseClient<_>>(&url)?;
	Ok(client)
}

// for tests
pub fn run_worker(scope: &crossbeam::Scope, stop: Arc<AtomicBool>, socket_path: &str) {
	let socket_path = socket_path.to_owned();
	scope.spawn(move || {
		let mut worker = nanoipc::Worker::new(&Arc::new(Database::new()));
		worker.add_reqrep(&socket_path).unwrap();
		while !stop.load(Ordering::Relaxed) {
			worker.poll();
		}
	});
}