79 lines
2.7 KiB
Rust
79 lines
2.7 KiB
Rust
//! The `pubsub` module implements a threaded subscription service on client RPC request
|
|
|
|
use crate::rpc_pubsub::{RpcSolPubSub, RpcSolPubSubImpl};
|
|
use crate::rpc_subscriptions::RpcSubscriptions;
|
|
use jsonrpc_pubsub::{PubSubHandler, Session};
|
|
use jsonrpc_ws_server::{RequestContext, ServerBuilder};
|
|
use std::net::SocketAddr;
|
|
use std::sync::atomic::{AtomicBool, Ordering};
|
|
use std::sync::Arc;
|
|
use std::thread::{self, sleep, Builder, JoinHandle};
|
|
use std::time::Duration;
|
|
|
|
pub struct PubSubService {
|
|
thread_hdl: JoinHandle<()>,
|
|
}
|
|
|
|
impl PubSubService {
|
|
pub fn new(
|
|
subscriptions: &Arc<RpcSubscriptions>,
|
|
pubsub_addr: SocketAddr,
|
|
exit: &Arc<AtomicBool>,
|
|
) -> Self {
|
|
info!("rpc_pubsub bound to {:?}", pubsub_addr);
|
|
let rpc = RpcSolPubSubImpl::new(subscriptions.clone());
|
|
let exit_ = exit.clone();
|
|
let thread_hdl = Builder::new()
|
|
.name("solana-pubsub".to_string())
|
|
.spawn(move || {
|
|
let mut io = PubSubHandler::default();
|
|
io.extend_with(rpc.to_delegate());
|
|
|
|
let server = ServerBuilder::with_meta_extractor(io, |context: &RequestContext| {
|
|
info!("New pubsub connection");
|
|
let session = Arc::new(Session::new(context.sender().clone()));
|
|
session.on_drop(|| {
|
|
info!("Pubsub connection dropped");
|
|
});
|
|
session
|
|
})
|
|
.start(&pubsub_addr);
|
|
|
|
if let Err(e) = server {
|
|
warn!("Pubsub service unavailable error: {:?}. \nAlso, check that port {} is not already in use by another application", e, pubsub_addr.port());
|
|
return;
|
|
}
|
|
while !exit_.load(Ordering::Relaxed) {
|
|
sleep(Duration::from_millis(100));
|
|
}
|
|
server.unwrap().close();
|
|
})
|
|
.unwrap();
|
|
Self { thread_hdl }
|
|
}
|
|
|
|
pub fn close(self) -> thread::Result<()> {
|
|
self.join()
|
|
}
|
|
|
|
pub fn join(self) -> thread::Result<()> {
|
|
self.thread_hdl.join()
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
use std::net::{IpAddr, Ipv4Addr};
|
|
|
|
#[test]
|
|
fn test_pubsub_new() {
|
|
let subscriptions = Arc::new(RpcSubscriptions::default());
|
|
let pubsub_addr = SocketAddr::new(IpAddr::V4(Ipv4Addr::new(0, 0, 0, 0)), 0);
|
|
let exit = Arc::new(AtomicBool::new(false));
|
|
let pubsub_service = PubSubService::new(&subscriptions, pubsub_addr, &exit);
|
|
let thread = pubsub_service.thread_hdl.thread();
|
|
assert_eq!(thread.name().unwrap(), "solana-pubsub");
|
|
}
|
|
}
|