79 lines
2.0 KiB
Rust
79 lines
2.0 KiB
Rust
// 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::*;
|
|
use std::convert::*;
|
|
use ipc::IpcConfig;
|
|
use util::bytes::{FromRawBytes, BytesConvertable, FromBytesError};
|
|
|
|
pub struct Service {
|
|
pub commits: RwLock<usize>,
|
|
pub rollbacks: RwLock<usize>,
|
|
}
|
|
|
|
pub struct CustomData {
|
|
a: usize,
|
|
b: usize,
|
|
}
|
|
|
|
impl FromRawBytes for CustomData {
|
|
fn from_bytes(bytes: &[u8]) -> Result<CustomData, FromBytesError> {
|
|
Ok(CustomData { a: bytes[0] * 256 + bytes[1], b: bytes[2] * 256 + bytes[3]})
|
|
}
|
|
}
|
|
|
|
impl BytesConvertable for CustomData {
|
|
fn bytes(&self) -> &[u8] {
|
|
unsafe { &::std::mem::transmute::<[u8;8]>(self) }
|
|
}
|
|
}
|
|
|
|
#[derive(Ipc)]
|
|
impl Service {
|
|
fn commit(&self, f: u32) -> u32 {
|
|
let mut lock = self.commits.write().unwrap();
|
|
*lock = *lock + f as usize;
|
|
f
|
|
}
|
|
pub fn rollback(&self, a: Option<u32>, b: u32) -> i32 {
|
|
let a_0 = a.unwrap_or_else(|| 0);
|
|
let mut lock = self.rollbacks.write().unwrap();
|
|
*lock = *lock + a_0 as usize - b as usize;
|
|
(a_0 - b) as i32
|
|
}
|
|
pub fn push_custom(&self, data: CustomData) -> bool {
|
|
let clock = self.commits.write().unwrap();
|
|
let rlock = self.commits.write().unwrap();
|
|
|
|
*clock = data.a;
|
|
*rlock = data.b;
|
|
|
|
true
|
|
}
|
|
}
|
|
|
|
impl Service {
|
|
pub fn new() -> Service {
|
|
Service {
|
|
commits: RwLock::new(0usize),
|
|
rollbacks: RwLock::new(0usize),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl ::ipc::IpcConfig for Service {}
|