ezri/dbus-fast
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

feat: speed up marshalling variants (#86)

Browse Source
This commit is contained in:
J. Nick Koston 2022-10-08 13:16:03 -10:00 committed by GitHub
parent f56c41722e
commit 7847e26e6e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 16 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/dbus_fast/_private/marshaller.pxd
View File

@ -23,6 +23,12 @@ cdef class Marshaller:
)
cpdef write_string(self, object value, _ = *)
@cython.locals(
signature_bytes=cython.bytes,
signature_len=cython.uint,
)
cdef _write_signature(self, str signature)
@cython.locals(
array_len=cython.uint,
written=cython.uint,
@ -40,7 +46,7 @@ cdef class Marshaller:
@cython.locals(
written=cython.uint,
)
cpdef write_single(self, object type_, object body)
cdef _write_single(self, object type_, object body)
@cython.locals(
written=cython.uint,

15
src/dbus_fast/_private/marshaller.py
View File

@ -37,6 +37,9 @@ class Marshaller:
return written + 4
def write_signature(self, signature: str, _=None) -> int:
return self._write_signature(signature)
def _write_signature(self, signature) -> int:
signature_bytes = signature.encode()
signature_len = len(signature)
self._buf.append(signature_len)
@ -56,8 +59,8 @@ class Marshaller:
return written
def write_variant(self, variant: Variant, _=None) -> int:
written = self.write_signature(variant.signature)
written += self.write_single(variant.type, variant.value)
written = self._write_signature(variant.signature)
written += self._write_single(variant.type, variant.value)
return written
def write_array(self, array: Iterable[Any], type_: SignatureType) -> int:
@ -103,16 +106,16 @@ class Marshaller:
def write_struct(self, array: List[Any], type_: SignatureType) -> int:
written = self._align(8)
for i, value in enumerate(array):
written += self.write_single(type_.children[i], value)
written += self._write_single(type_.children[i], value)
return written
def write_dict_entry(self, dict_entry: List[Any], type_: SignatureType) -> int:
written = self._align(8)
written += self.write_single(type_.children[0], dict_entry[0])
written += self.write_single(type_.children[1], dict_entry[1])
written += self._write_single(type_.children[0], dict_entry[0])
written += self._write_single(type_.children[1], dict_entry[1])
return written
def write_single(self, type_: SignatureType, body: Any) -> int:
def _write_single(self, type_: SignatureType, body: Any) -> int:
t = type_.token
if t not in self._writers: