mirror of
https://github.com/mimblewimble/grin.git
synced 2025-02-08 12:21:09 +03:00
128 lines
3.5 KiB
Rust
128 lines
3.5 KiB
Rust
|
// Copyright 2018 The Grin Developers
|
||
|
|
//
|
||
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
// you may not use this file except in compliance with the License.
|
||
|
|
// You may obtain a copy of the License at
|
||
|
|
//
|
||
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
//
|
||
|
|
// Unless required by applicable law or agreed to in writing, software
|
||
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
// See the License for the specific language governing permissions and
|
||
|
|
// limitations under the License.
|
||
|
|
|
||
|
|
use super::utils::w;
|
||
|
|
use core::core::hash::Hashed;
|
||
|
|
use core::core::Transaction;
|
||
|
|
use core::ser;
|
||
|
|
use futures::future::ok;
|
||
|
|
use futures::Future;
|
||
|
|
use hyper::{Body, Request, StatusCode};
|
||
|
|
use pool;
|
||
|
|
use rest::*;
|
||
|
|
use router::{Handler, ResponseFuture};
|
||
|
|
use std::collections::HashMap;
|
||
|
|
use std::sync::Weak;
|
||
|
|
use types::*;
|
||
|
|
use url::form_urlencoded;
|
||
|
|
use util;
|
||
|
|
use util::RwLock;
|
||
|
|
use web::*;
|
||
|
|
|
||
|
|
/// Get basic information about the transaction pool.
|
||
|
|
/// GET /v1/pool
|
||
|
|
pub struct PoolInfoHandler {
|
||
|
|
pub tx_pool: Weak<RwLock<pool::TransactionPool>>,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Handler for PoolInfoHandler {
|
||
|
|
fn get(&self, _req: Request<Body>) -> ResponseFuture {
|
||
|
|
let pool_arc = w(&self.tx_pool);
|
||
|
|
let pool = pool_arc.read();
|
||
|
|
|
||
|
|
json_response(&PoolInfo {
|
||
|
|
pool_size: pool.total_size(),
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/// Dummy wrapper for the hex-encoded serialized transaction.
|
||
|
|
#[derive(Serialize, Deserialize)]
|
||
|
|
struct TxWrapper {
|
||
|
|
tx_hex: String,
|
||
|
|
}
|
||
|
|
|
||
|
|
/// Push new transaction to our local transaction pool.
|
||
|
|
/// POST /v1/pool/push
|
||
|
|
pub struct PoolPushHandler {
|
||
|
|
pub tx_pool: Weak<RwLock<pool::TransactionPool>>,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl PoolPushHandler {
|
||
|
|
fn update_pool(&self, req: Request<Body>) -> Box<Future<Item = (), Error = Error> + Send> {
|
||
|
|
let params = match req.uri().query() {
|
||
|
|
Some(query_string) => form_urlencoded::parse(query_string.as_bytes())
|
||
|
|
.into_owned()
|
||
|
|
.fold(HashMap::new(), |mut hm, (k, v)| {
|
||
|
|
hm.entry(k).or_insert(vec![]).push(v);
|
||
|
|
hm
|
||
|
|
}),
|
||
|
|
None => HashMap::new(),
|
||
|
|
};
|
||
|
|
|
||
|
|
let fluff = params.get("fluff").is_some();
|
||
|
|
let pool_arc = w(&self.tx_pool).clone();
|
||
|
|
|
||
|
|
Box::new(
|
||
|
|
parse_body(req)
|
||
|
|
.and_then(move |wrapper: TxWrapper| {
|
||
|
|
util::from_hex(wrapper.tx_hex)
|
||
|
|
.map_err(|e| ErrorKind::RequestError(format!("Bad request: {}", e)).into())
|
||
|
|
})
|
||
|
|
.and_then(move |tx_bin| {
|
||
|
|
ser::deserialize(&mut &tx_bin[..])
|
||
|
|
.map_err(|e| ErrorKind::RequestError(format!("Bad request: {}", e)).into())
|
||
|
|
})
|
||
|
|
.and_then(move |tx: Transaction| {
|
||
|
|
let source = pool::TxSource {
|
||
|
|
debug_name: "push-api".to_string(),
|
||
|
|
identifier: "?.?.?.?".to_string(),
|
||
|
|
};
|
||
|
|
info!(
|
||
|
|
"Pushing transaction {} to pool (inputs: {}, outputs: {}, kernels: {})",
|
||
|
|
tx.hash(),
|
||
|
|
tx.inputs().len(),
|
||
|
|
tx.outputs().len(),
|
||
|
|
tx.kernels().len(),
|
||
|
|
);
|
||
|
|
|
||
|
|
// Push to tx pool.
|
||
|
|
let mut tx_pool = pool_arc.write();
|
||
|
|
let header = tx_pool.blockchain.chain_head().unwrap();
|
||
|
|
tx_pool
|
||
|
|
.add_to_pool(source, tx, !fluff, &header)
|
||
|
|
.map_err(|e| {
|
||
|
|
error!("update_pool: failed with error: {:?}", e);
|
||
|
|
ErrorKind::Internal(format!("Failed to update pool: {:?}", e)).into()
|
||
|
|
})
|
||
|
|
}),
|
||
|
|
)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Handler for PoolPushHandler {
|
||
|
|
fn post(&self, req: Request<Body>) -> ResponseFuture {
|
||
|
|
Box::new(
|
||
|
|
self.update_pool(req)
|
||
|
|
.and_then(|_| ok(just_response(StatusCode::OK, "")))
|
||
|
|
.or_else(|e| {
|
||
|
|
ok(just_response(
|
||
|
|
StatusCode::INTERNAL_SERVER_ERROR,
|
||
|
|
format!("failed: {}", e),
|
||
|
|
))
|
||
|
|
}),
|
||
|
|
)
|
||
|
|
}
|
||
|
|
}
|