openethereum/ipc/tests/service.rs.in

89 lines
2.2 KiB
Rust
Raw Normal View History

2016-03-24 22:07:01 +01:00
// Copyright 2015, 2016 Ethcore (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/>.
use std::sync::RwLock;
use std::ops::*;
2016-04-13 02:46:36 +02:00
use std::convert::*;
2016-04-08 13:07:25 +02:00
use ipc::IpcConfig;
2016-04-13 02:46:36 +02:00
use util::bytes::{FromRawBytes, BytesConvertable, FromBytesError};
2016-03-24 22:07:01 +01:00
pub struct Service {
pub commits: RwLock<usize>,
pub rollbacks: RwLock<usize>,
}
2016-04-13 02:46:36 +02:00
pub struct CustomData {
2016-04-13 12:46:49 +02:00
pub a: usize,
pub b: usize,
2016-04-13 02:46:36 +02:00
}
impl FromRawBytes for CustomData {
fn from_bytes(bytes: &[u8]) -> Result<CustomData, FromBytesError> {
Ok(CustomData {
a: bytes[0] as usize * 256 + bytes[1] as usize,
b: bytes[2] as usize * 256 + bytes[3] as usize
})
2016-04-13 02:46:36 +02:00
}
}
impl BytesConvertable for CustomData {
fn bytes(&self) -> &[u8] {
let ip: *const CustomData = self;
let ptr: *const u8 = ip as *const _;
unsafe {
::std::slice::from_raw_parts(
ptr,
2016-04-13 12:46:49 +02:00
::std::mem::size_of::<CustomData>()
)
}
2016-04-13 02:46:36 +02:00
}
}
2016-03-24 22:07:01 +01:00
#[derive(Ipc)]
impl Service {
fn commit(&self, f: u32) -> u32 {
let mut lock = self.commits.write().unwrap();
*lock = *lock + f as usize;
f
}
2016-03-30 01:20:01 +02:00
pub fn rollback(&self, a: Option<u32>, b: u32) -> i32 {
let a_0 = a.unwrap_or_else(|| 0);
2016-03-24 22:07:01 +01:00
let mut lock = self.rollbacks.write().unwrap();
2016-03-30 01:20:01 +02:00
*lock = *lock + a_0 as usize - b as usize;
(a_0 - b) as i32
2016-03-24 22:07:01 +01:00
}
2016-04-13 02:46:36 +02:00
pub fn push_custom(&self, data: CustomData) -> bool {
let mut clock = self.commits.write().unwrap();
let mut rlock = self.commits.write().unwrap();
2016-04-13 02:46:36 +02:00
*clock = data.a;
*rlock = data.b;
true
}
2016-03-24 22:07:01 +01:00
}
2016-03-29 21:40:38 +02:00
impl Service {
2016-03-24 22:07:01 +01:00
pub fn new() -> Service {
Service {
commits: RwLock::new(0usize),
rollbacks: RwLock::new(0usize),
}
}
}
2016-04-07 22:18:48 +02:00
impl ::ipc::IpcConfig for Service {}