51 lines
1.7 KiB
Rust
51 lines
1.7 KiB
Rust
use beacon_chain::{BeaconChain, BeaconChainTypes};
|
|
use futures::Future;
|
|
use grpcio::{RpcContext, UnarySink};
|
|
use protos::services::{Empty, Fork, NodeInfoResponse};
|
|
use protos::services_grpc::BeaconNodeService;
|
|
use slog::{trace, warn};
|
|
use std::sync::Arc;
|
|
use types::EthSpec;
|
|
|
|
#[derive(Clone)]
|
|
pub struct BeaconNodeServiceInstance<T: BeaconChainTypes> {
|
|
pub chain: Arc<BeaconChain<T>>,
|
|
pub log: slog::Logger,
|
|
}
|
|
|
|
impl<T: BeaconChainTypes> BeaconNodeService for BeaconNodeServiceInstance<T> {
|
|
/// Provides basic node information.
|
|
fn info(&mut self, ctx: RpcContext, _req: Empty, sink: UnarySink<NodeInfoResponse>) {
|
|
trace!(self.log, "Node info requested via RPC");
|
|
|
|
// build the response
|
|
let mut node_info = NodeInfoResponse::new();
|
|
node_info.set_version(version::version());
|
|
|
|
// get the chain state
|
|
let state = &self.chain.head().beacon_state;
|
|
let state_fork = state.fork.clone();
|
|
let genesis_time = state.genesis_time;
|
|
|
|
// build the rpc fork struct
|
|
let mut fork = Fork::new();
|
|
fork.set_previous_version(state_fork.previous_version.to_vec());
|
|
fork.set_current_version(state_fork.current_version.to_vec());
|
|
fork.set_epoch(state_fork.epoch.into());
|
|
|
|
let spec = T::EthSpec::spec();
|
|
|
|
node_info.set_fork(fork);
|
|
node_info.set_genesis_time(genesis_time);
|
|
node_info.set_genesis_slot(spec.genesis_slot.as_u64());
|
|
node_info.set_chain_id(u32::from(spec.chain_id));
|
|
|
|
// send the node_info the requester
|
|
let error_log = self.log.clone();
|
|
let f = sink
|
|
.success(node_info)
|
|
.map_err(move |e| warn!(error_log, "failed to reply {:?}", e));
|
|
ctx.spawn(f)
|
|
}
|
|
}
|