Core/grin-wallet
0
0
Fork 0
mirror of https://github.com/mimblewimble/grin-wallet.git synced 2025-01-20 19:11:09 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix panic when calling certain API endpoints (#589)

Browse source 
* Detect if already in runtime context

* Add comment

Co-authored-by: jaspervdm <j@sper.dev>
This commit is contained in:
Quentin Le Sceller 2021-03-05 13:29:08 -05:00 committed by GitHub
parent 712101bba2
commit 563b456e47
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 17 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
impls/src/client_utils/client.rs
View file

@ -28,7 +28,7 @@ use std::fmt::{self, Display};
use std::net::SocketAddr;
use std::sync::{Arc, Mutex};
use std::time::Duration;
use tokio::runtime::{Builder, Runtime};
use tokio::runtime::{Builder, Handle, Runtime};
// Global Tokio runtime.
// Needs a `Mutex` because `Runtime::block_on` requires mutable access.
@ -101,6 +101,7 @@ impl From<Context<ErrorKind>> for Error {
}
}
#[derive(Clone)]
pub struct Client {
/// Whether to use socks proxy
pub use_socks: bool,
@ -339,9 +340,20 @@ impl Client {
}
pub fn send_request(&self, req: Request<Body>) -> Result<String, Error> {
RUNTIME
.lock()
.unwrap()
.block_on(self.send_request_async(req))
// This client is currently used both outside and inside of a tokio runtime
// context. In the latter case we are not allowed to do a blocking call to
// our global runtime, which unfortunately means we have to spawn a new thread
if Handle::try_current().is_ok() {
let rt = RUNTIME.clone();
let client = self.clone();
std::thread::spawn(move || rt.lock().unwrap().block_on(client.send_request_async(req)))
.join()
.unwrap()
} else {
RUNTIME
.lock()
.unwrap()
.block_on(self.send_request_async(req))
}
}
}