50 lines
1.4 KiB
Rust
50 lines
1.4 KiB
Rust
use {
|
|
crate::{
|
|
entities::{market, ticker},
|
|
grpc::services,
|
|
},
|
|
eyre::Result,
|
|
lool::{cli::stylize::Stylize, logger::info},
|
|
sea_orm::DatabaseConnection,
|
|
std::net::SocketAddr,
|
|
tonic::transport::Server,
|
|
};
|
|
|
|
/// the environment variable to set the gRPC server address
|
|
const RUSTLER_GRPC_API_ADDR: &str = "RUSTLER_GRPC_API_ADDR";
|
|
|
|
/// 🐎 » starts the rustler gRPC server
|
|
pub async fn start(conn: DatabaseConnection) -> Result<()> {
|
|
fn get_default_addr() -> String {
|
|
let addr = "0.0.0.0:50051";
|
|
info!(
|
|
"`{}` not set, using default {}",
|
|
RUSTLER_GRPC_API_ADDR.italic(),
|
|
addr.green()
|
|
);
|
|
addr.to_owned()
|
|
}
|
|
|
|
let addr: SocketAddr =
|
|
std::env::var(RUSTLER_GRPC_API_ADDR).unwrap_or_else(|_| get_default_addr()).parse()?;
|
|
|
|
let market_db = market::Service::new(conn.clone()).await;
|
|
let ticker_db = ticker::Service::new(conn.clone()).await;
|
|
|
|
let market_grpc = services::market::GrpcServer { svc: market_db };
|
|
let ticker_grpc = services::ticker::GrpcServer { svc: ticker_db };
|
|
|
|
info!(
|
|
"🎉 gRPC server listening on {}",
|
|
addr.clone().to_string().green()
|
|
);
|
|
|
|
Server::builder()
|
|
.add_service(market_grpc.svc()) // add the market api
|
|
.add_service(ticker_grpc.svc()) // add the ticker api
|
|
.serve(addr)
|
|
.await?;
|
|
|
|
Ok(())
|
|
}
|