|
1 |
| -use crate::compiler::plugin::proc_macro::ProcMacroHost; |
2 |
| -use anyhow::Result; |
| 1 | +use std::num::NonZero; |
| 2 | +use std::thread::available_parallelism; |
| 3 | + |
| 4 | +use anyhow::{anyhow, Result}; |
3 | 5 | use connection::Connection;
|
| 6 | +use crossbeam_channel::{Receiver, Sender}; |
| 7 | +use scarb_proc_macro_server_types::jsonrpc::{ResponseError, RpcRequest, RpcResponse}; |
| 8 | +use serde_json::Value; |
| 9 | + |
| 10 | +use crate::compiler::plugin::proc_macro::ProcMacroHost; |
4 | 11 |
|
5 | 12 | mod connection;
|
6 | 13 |
|
7 | 14 | pub fn start_proc_macro_server(_proc_macros: ProcMacroHost) -> Result<()> {
|
8 | 15 | let connection = Connection::new();
|
| 16 | + let available_parallelism = available_parallelism().map(NonZero::get).unwrap_or(4); |
| 17 | + |
| 18 | + for i in 0..available_parallelism { |
| 19 | + let receiver = connection.receiver.clone(); |
| 20 | + let sender = connection.sender.clone(); |
9 | 21 |
|
10 |
| - //TODO |
| 22 | + std::thread::Builder::new() |
| 23 | + .name(format!("proc-macro-server-worker-thread-{i}")) |
| 24 | + .spawn(move || { |
| 25 | + handle_requests(receiver, sender); |
| 26 | + }) |
| 27 | + .expect("failed to spawn thread"); |
| 28 | + } |
11 | 29 |
|
12 | 30 | connection.join();
|
13 | 31 |
|
14 | 32 | Ok(())
|
15 | 33 | }
|
| 34 | + |
| 35 | +fn handle_requests(receiver: Receiver<RpcRequest>, sender: Sender<RpcResponse>) { |
| 36 | + for request in receiver { |
| 37 | + let id = request.id; |
| 38 | + let response = route_request(request); |
| 39 | + |
| 40 | + let response = match response { |
| 41 | + Ok(result) => RpcResponse { |
| 42 | + id, |
| 43 | + result: Some(result), |
| 44 | + error: None, |
| 45 | + }, |
| 46 | + Err(error) => RpcResponse { |
| 47 | + id, |
| 48 | + result: None, |
| 49 | + error: Some(ResponseError { |
| 50 | + message: error.to_string(), |
| 51 | + }), |
| 52 | + }, |
| 53 | + }; |
| 54 | + |
| 55 | + sender.send(response).unwrap(); |
| 56 | + } |
| 57 | +} |
| 58 | + |
| 59 | +fn route_request(request: RpcRequest) -> Result<Value> { |
| 60 | + #[allow(clippy::match_single_binding)] |
| 61 | + match request.method.as_str() { |
| 62 | + //TODO add method handlers |
| 63 | + _ => Err(anyhow!("method not found")), |
| 64 | + } |
| 65 | +} |
0 commit comments