Mark PyCoroutine, PyCoroutineWrapper as ThreadSafe

palaviv · palaviv · commit 58b9603e095a · 2020-05-08T10:34:15.000+03:00
diff --git a/vm/src/obj/objcoroutine.rs b/vm/src/obj/objcoroutine.rs
@@ -4,7 +4,7 @@ use super::objstr::PyStringRef;
 use super::objtype::PyClassRef;
 use crate::frame::FrameRef;
 use crate::function::OptionalArg;
-use crate::pyobject::{PyClassImpl, PyContext, PyObjectRef, PyRef, PyResult, PyValue};
+use crate::pyobject::{PyClassImpl, PyContext, PyObjectRef, PyRef, PyResult, PyValue, ThreadSafe};
 use crate::vm::VirtualMachine;
 
 pub type PyCoroutineRef = PyRef<PyCoroutine>;
@@ -15,6 +15,8 @@ pub struct PyCoroutine {
     inner: Coro,
 }
 
+impl ThreadSafe for PyCoroutine {}
+
 impl PyValue for PyCoroutine {
     fn class(vm: &VirtualMachine) -> PyClassRef {
         vm.ctx.types.coroutine_type.clone()
@@ -101,6 +103,8 @@ pub struct PyCoroutineWrapper {
     coro: PyCoroutineRef,
 }
 
+impl ThreadSafe for PyCoroutineWrapper {}
+
 impl PyValue for PyCoroutineWrapper {
     fn class(vm: &VirtualMachine) -> PyClassRef {
         vm.ctx.types.coroutine_wrapper_type.clone()
