Mark PyByteInner and PyBytes as ThreadSafe

palaviv · palaviv · commit 6e89c6e014f4 · 2020-04-11T12:24:09.000+03:00
diff --git a/vm/src/obj/objbyteinner.rs b/vm/src/obj/objbyteinner.rs
@@ -18,7 +18,8 @@ use super::objtuple::PyTupleRef;
 use crate::function::OptionalArg;
 use crate::pyhash;
 use crate::pyobject::{
-    Either, PyComparisonValue, PyIterable, PyObjectRef, PyResult, TryFromObject, TypeProtocol,
+    Either, PyComparisonValue, PyIterable, PyObjectRef, PyResult, ThreadSafe, TryFromObject,
+    TypeProtocol,
 };
 use crate::vm::VirtualMachine;
 
@@ -27,6 +28,8 @@ pub struct PyByteInner {
     pub elements: Vec<u8>,
 }
 
+impl ThreadSafe for PyByteInner {}
+
 impl TryFromObject for PyByteInner {
     fn try_from_object(vm: &VirtualMachine, obj: PyObjectRef) -> PyResult<Self> {
         match_class!(match obj {
diff --git a/vm/src/obj/objbytes.rs b/vm/src/obj/objbytes.rs
@@ -21,7 +21,7 @@ use crate::pyobject::{
     Either, IntoPyObject,
     PyArithmaticValue::{self, *},
     PyClassImpl, PyComparisonValue, PyContext, PyIterable, PyObjectRef, PyRef, PyResult, PyValue,
-    TryFromObject, TypeProtocol,
+    ThreadSafe, TryFromObject, TypeProtocol,
 };
 use crate::vm::VirtualMachine;
 use std::str::FromStr;
@@ -40,6 +40,9 @@ use std::str::FromStr;
 pub struct PyBytes {
     inner: PyByteInner,
 }
+
+impl ThreadSafe for PyBytes {}
+
 pub type PyBytesRef = PyRef<PyBytes>;
 
 impl PyBytes {
