From 71cf52430bd3ece01083309c5f0f5d934dca3d59 Mon Sep 17 00:00:00 2001
From: "J. Nick Koston" <nick@koston.org>
Date: Thu, 3 Nov 2022 10:09:43 +0100
Subject: [PATCH] feat: speed up marshall write_string and write_variant (#138)

---
 src/dbus_fast/_private/marshaller.pxd | 4 +++-
 src/dbus_fast/_private/marshaller.py  | 4 +++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/dbus_fast/_private/marshaller.pxd b/src/dbus_fast/_private/marshaller.pxd
index 6995d68..958b892 100644
--- a/src/dbus_fast/_private/marshaller.pxd
+++ b/src/dbus_fast/_private/marshaller.pxd
@@ -27,7 +27,7 @@ cdef class Marshaller:
         signature_len=cython.uint,
         written=cython.uint,
     )
-    cpdef write_string(self, object value, object _type)
+    cpdef write_string(self, str value, object _type)
 
     @cython.locals(
         signature_len=cython.uint,
@@ -59,6 +59,8 @@ cdef class Marshaller:
 
     @cython.locals(
         written=cython.uint,
+        signature=cython.str,
+        signature_bytes=cython.bytes,
     )
     cpdef write_variant(self, object variant, object type)
 
diff --git a/src/dbus_fast/_private/marshaller.py b/src/dbus_fast/_private/marshaller.py
index ecfa259..8ec71be 100644
--- a/src/dbus_fast/_private/marshaller.py
+++ b/src/dbus_fast/_private/marshaller.py
@@ -63,7 +63,9 @@ class Marshaller:
         return written
 
     def write_variant(self, variant: Variant, type_: SignatureType) -> int:
-        written = self._write_signature(variant.signature.encode())
+        signature = variant.signature
+        signature_bytes = signature.encode()
+        written = self._write_signature(signature_bytes)
         written += self._write_single(variant.type, variant.value)
         return written