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

feat: speed up unmarshaller (#113)

Browse Source
This commit is contained in:
J. Nick Koston
2022-10-26 14:07:22 -05:00
committed by GitHub
parent 6094b85037
commit 8f7f982a75
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/dbus_fast/_private/unmarshaller.py
View File

@@ -365,8 +365,7 @@ class Unmarshaller:
signature_len = buf[self._pos] # byte signature_len = buf[self._pos] # byte
o = self._pos + 1 o = self._pos + 1
self._pos += signature_len + 2 # one for the byte, one for the '\0' self._pos += signature_len + 2 # one for the byte, one for the '\0'
type_ = get_signature_tree(buf[o : o + signature_len].decode()).types[0] token = buf[o : o + signature_len].decode()
token = type_.token
# Now that we have the token we can read the variant value # Now that we have the token we can read the variant value
key = HEADER_MESSAGE_ARG_NAME[field_0] key = HEADER_MESSAGE_ARG_NAME[field_0]
# Strings and signatures are the most common types # Strings and signatures are the most common types
@@ -378,7 +377,7 @@ class Unmarshaller:
else: else:
# There shouldn't be any other types in the header # There shouldn't be any other types in the header
# but just in case, we'll read it using the slow path # but just in case, we'll read it using the slow path
headers[key] = readers[type_.token](self, type_) headers[key] = readers[token](self, get_signature_tree(token).types[0])
return headers return headers
def _read_header(self) -> None: def _read_header(self) -> None: