tigerbeetle 0.16.28__py3-none-any.whl → 0.16.29__py3-none-any.whl

tigerbeetle/bindings.py

@@ -22,6 +22,7 @@ class Operation(enum.IntEnum):
     GET_ACCOUNT_BALANCES = 134
     QUERY_ACCOUNTS = 135
     QUERY_TRANSFERS = 136
+    GET_EVENTS = 137
 
 
 class PacketStatus(enum.IntEnum):
@@ -35,9 +36,7 @@ class PacketStatus(enum.IntEnum):
     INVALID_DATA_SIZE = 7
 
 
-Client = ctypes.c_void_p
-
-class Status(enum.IntEnum):
+class InitStatus(enum.IntEnum):
     SUCCESS = 0
     UNEXPECTED = 1
     OUT_OF_MEMORY = 2
@@ -47,6 +46,18 @@ class Status(enum.IntEnum):
     NETWORK_SUBSYSTEM = 6
 
 
+class ClientStatus(enum.IntEnum):
+    OK = 0
+    INVALID = 1
+
+
+class LogLevel(enum.IntEnum):
+    ERR = 0
+    WARN = 1
+    INFO = 2
+    DEBUG = 3
+
+
 class RegisterLogCallbackStatus(enum.IntEnum):
     SUCCESS = 0
     ALREADY_REGISTERED = 1
@@ -274,25 +285,38 @@ class CPacket(ctypes.Structure):
     @classmethod
     def from_param(cls, obj):
         validate_uint(bits=32, name="data_size", number=obj.data_size)
-        validate_uint(bits=16, name="tag", number=obj.tag)
+        validate_uint(bits=16, name="user_tag", number=obj.user_tag)
         validate_uint(bits=8, name="operation", number=obj.operation)
         return cls(
             user_data=obj.user_data,
             data=obj.data,
             data_size=obj.data_size,
-            tag=obj.tag,
+            user_tag=obj.user_tag,
             operation=obj.operation,
             status=obj.status,
+            opaque=obj.opaque,
         )
 
 CPacket._fields_ = [ # noqa: SLF001
     ("user_data", ctypes.c_void_p),
     ("data", ctypes.c_void_p),
     ("data_size", ctypes.c_uint32),
-    ("tag", ctypes.c_uint16),
+    ("user_tag", ctypes.c_uint16),
     ("operation", ctypes.c_uint8),
     ("status", ctypes.c_uint8),
-    ("reserved", ctypes.c_uint8 * 32),
+    ("opaque", ctypes.c_uint8 * 32),
+]
+
+
+class CClient(ctypes.Structure):
+    @classmethod
+    def from_param(cls, obj):
+        return cls(
+            opaque=obj.opaque,
+        )
+
+CClient._fields_ = [ # noqa: SLF001
+    ("opaque", ctypes.c_uint64 * 4),
 ]
 
 
@@ -608,22 +632,22 @@ CQueryFilter._fields_ = [ # noqa: SLF001
 
 
 # Don't be tempted to use c_char_p for bytes_ptr - it's for null terminated strings only.
-OnCompletion = ctypes.CFUNCTYPE(None, ctypes.c_void_p, Client, ctypes.POINTER(CPacket),
+OnCompletion = ctypes.CFUNCTYPE(None, ctypes.c_void_p, ctypes.POINTER(CPacket),
                                 ctypes.c_uint64, ctypes.c_void_p, ctypes.c_uint32)
-LogHandler = ctypes.CFUNCTYPE(None, ctypes.c_uint8, ctypes.c_void_p, ctypes.c_uint)
+LogHandler = ctypes.CFUNCTYPE(None, ctypes.c_uint, ctypes.c_void_p, ctypes.c_uint)
 
 # Initialize a new TigerBeetle client which connects to the addresses provided and
 # completes submitted packets by invoking the callback with the given context.
 tb_client_init = tbclient.tb_client_init
-tb_client_init.restype = Status
-tb_client_init.argtypes = [ctypes.POINTER(Client), ctypes.POINTER(ctypes.c_uint8 * 16),
+tb_client_init.restype = InitStatus
+tb_client_init.argtypes = [ctypes.POINTER(CClient), ctypes.POINTER(ctypes.c_uint8 * 16),
                            ctypes.c_char_p, ctypes.c_uint32, ctypes.c_void_p,
                            OnCompletion]
 
 # Initialize a new TigerBeetle client which echos back any data submitted.
 tb_client_init_echo = tbclient.tb_client_init_echo
-tb_client_init_echo.restype = Status
-tb_client_init_echo.argtypes = [ctypes.POINTER(Client), ctypes.POINTER(ctypes.c_uint8 * 16),
+tb_client_init_echo.restype = InitStatus
+tb_client_init_echo.argtypes = [ctypes.POINTER(CClient), ctypes.POINTER(ctypes.c_uint8 * 16),
                                 ctypes.c_char_p, ctypes.c_uint32, ctypes.c_void_p,
                                 OnCompletion]
 
@@ -631,15 +655,15 @@ tb_client_init_echo.argtypes = [ctypes.POINTER(Client), ctypes.POINTER(ctypes.c_
 # `TB_PACKET_CLIENT_SHUTDOWN` before freeing any allocated client resources from init.
 # It is undefined behavior to use any functions on the client once deinit is called.
 tb_client_deinit = tbclient.tb_client_deinit
-tb_client_deinit.restype = None
-tb_client_deinit.argtypes = [Client]
+tb_client_deinit.restype = ClientStatus
+tb_client_deinit.argtypes = [ctypes.POINTER(CClient)]
 
 # Submit a packet with its operation, data, and data_size fields set.
 # Once completed, `on_completion` will be invoked with `on_completion_ctx` and the given
 # packet on the `tb_client` thread (separate from caller's thread).
 tb_client_submit = tbclient.tb_client_submit
-tb_client_submit.restype = None
-tb_client_submit.argtypes = [Client, ctypes.POINTER(CPacket)]
+tb_client_submit.restype = ClientStatus
+tb_client_submit.argtypes = [ctypes.POINTER(CClient), ctypes.POINTER(CPacket)]
 
 tb_client_register_log_callback = tbclient.tb_client_register_log_callback
 tb_client_register_log_callback.restype = RegisterLogCallbackStatus

tigerbeetle/client.py

@@ -77,6 +77,9 @@ amount_max = (2 ** 128) - 1
 class InitError(Exception):
     pass
 
+class ClientClosedError(Exception):
+    pass
+
 class PacketError(Exception):
     pass
 
@@ -87,7 +90,7 @@ class Client:
 
     def __init__(self, cluster_id: int, replica_addresses: str):
         self._client_key = Client._counter.increment()
-        self._client = bindings.Client()
+        self._client = bindings.CClient()
 
         self._inflight_packets: dict[int, InflightPacket] = {}
 
@@ -104,7 +107,7 @@ class Client:
             self._client_key,
             self._c_on_completion
         )
-        if init_status != bindings.Status.SUCCESS:
+        if init_status != bindings.InitStatus.SUCCESS:
             raise InitError(init_status)
 
         Client._clients[self._client_key] = self
@@ -115,6 +118,7 @@ class Client:
         packet = bindings.CPacket()
         packet.next = None
         packet.user_data = Client._counter.increment()
+        packet.user_tag = 0
         packet.operation = operation
         packet.status = bindings.PacketStatus.OK
 
@@ -134,19 +138,18 @@ class Client:
             c_result_type=c_result_type)
 
     def close(self):
-        bindings.tb_client_deinit(self._client)
+        bindings.tb_client_deinit(ctypes.byref(self._client))
         tb_assert(self._client is not None)
         tb_assert(len(self._inflight_packets) == 0)
         del Client._clients[self._client_key]
 
     @staticmethod
     @bindings.OnCompletion
-    def _c_on_completion(completion_ctx, tb_client, packet, timestamp, bytes_ptr, len_):
+    def _c_on_completion(completion_ctx, packet, timestamp, bytes_ptr, len_):
         """
         Invoked in a separate thread
         """
         self: Client = Client._clients[completion_ctx]
-        tb_assert(self._client.value == tb_client)
 
         packet = ctypes.cast(packet, ctypes.POINTER(bindings.CPacket))
         inflight_packet = self._inflight_packets[packet[0].user_data]
@@ -193,11 +196,15 @@ class ClientSync(Client, bindings.StateMachineMixin):
         inflight_packet.on_completion = self._on_completion
         inflight_packet.on_completion_context = CompletionContextSync(event=threading.Event())
 
-        bindings.tb_client_submit(self._client, ctypes.byref(inflight_packet.packet))
-        inflight_packet.on_completion_context.event.wait()
+        client_state = bindings.tb_client_submit(ctypes.byref(self._client), ctypes.byref(inflight_packet.packet))
+        if client_state == bindings.ClientStatus.OK:
+            inflight_packet.on_completion_context.event.wait()
 
         del self._inflight_packets[inflight_packet.packet.user_data]
 
+        if client_state == bindings.ClientStatus.INVALID:
+            raise ClientClosedError()
+
         if isinstance(inflight_packet.response, Exception):
             raise inflight_packet.response
 
@@ -230,11 +237,15 @@ class ClientAsync(Client, bindings.AsyncStateMachineMixin):
             event=asyncio.Event()
         )
 
-        bindings.tb_client_submit(self._client, ctypes.byref(inflight_packet.packet))
-        await inflight_packet.on_completion_context.event.wait()
+        client_state = bindings.tb_client_submit(ctypes.byref(self._client), ctypes.byref(inflight_packet.packet))
+        if client_state == bindings.ClientStatus.OK:
+            await inflight_packet.on_completion_context.event.wait()
 
         del self._inflight_packets[inflight_packet.packet.user_data]
 
+        if client_state == bindings.ClientStatus.INVALID:
+            raise ClientClosedError()
+
         if isinstance(inflight_packet.response, Exception):
             raise inflight_packet.response
 
@@ -244,10 +255,10 @@ class ClientAsync(Client, bindings.AsyncStateMachineMixin):
 @bindings.LogHandler
 def log_handler(level_zig, message_ptr, message_len):
     level_python = {
-        0: logging.ERROR,
-        1: logging.WARNING,
-        2: logging.INFO,
-        3: logging.DEBUG,
+        bindings.LogLevel.ERR: logging.ERROR,
+        bindings.LogLevel.WARN: logging.WARNING,
+        bindings.LogLevel.INFO: logging.INFO,
+        bindings.LogLevel.DEBUG: logging.DEBUG,
     }[level_zig]
     logger.log(level_python, ctypes.string_at(message_ptr, message_len).decode("utf-8"))
 

{tigerbeetle-0.16.28.dist-info → tigerbeetle-0.16.29.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tigerbeetle
-Version: 0.16.28
+Version: 0.16.29
 Summary: The TigerBeetle client for Python.
 Project-URL: Homepage, https://github.com/tigerbeetle/tigerbeetle
 Project-URL: Issues, https://github.com/tigerbeetle/tigerbeetle/issues

tigerbeetle-0.16.29.dist-info/RECORD

@@ -0,0 +1,14 @@
+tigerbeetle/__init__.py,sha256=3cmov0HPkByJw1znLKoTqIuZUEeEQMEg1fflWY3ty9A,168
+tigerbeetle/bindings.py,sha256=8YfOv18KBGvLkkfhMiUFAuwCCzkhdgKg9-kuI80rW44,26460
+tigerbeetle/client.py,sha256=oGg1vAFqI8UZ5awI3sshZfxUdBsroUKwgp2rUa75lFc,9255
+tigerbeetle/lib.py,sha256=JKreDxdw_YejtH7F135_m6I6mNBSwjokaI6QicKm7cc,2425
+tigerbeetle/lib/aarch64-linux-gnu.2.27/libtb_client.so,sha256=9H4irv3j7WGsp97OFOESDYjkcrCAbAqavs0qUBJo36A,3882600
+tigerbeetle/lib/aarch64-linux-musl/libtb_client.so,sha256=fztP2HOQGv4e297kUhMJGfTr5iuBthij6L8Tqqe17Yk,3859344
+tigerbeetle/lib/aarch64-macos/libtb_client.dylib,sha256=WtlycdXHng6WURVstdZwwqBMtGLupSTacjq8mWeYNXw,534944
+tigerbeetle/lib/x86_64-linux-gnu.2.27/libtb_client.so,sha256=NiQMmn8L7ji5K_tkM84z8sib6IZoD88-S7AqXkyTHPk,3824296
+tigerbeetle/lib/x86_64-linux-musl/libtb_client.so,sha256=KPN4nl73s35aE7d9bWi9qSh2Z6Dv3ieQ8nax_96AVuA,3821328
+tigerbeetle/lib/x86_64-macos/libtb_client.dylib,sha256=leFoiEen_wjSCZr1Qrfxwf6mSGXyesGyqdmeQC4NgGQ,743167
+tigerbeetle/lib/x86_64-windows/tb_client.dll,sha256=M4uCwkVXA1r96CGNotF3gkaX7BoNNNBFK8cmLCUxeWM,829440
+tigerbeetle-0.16.29.dist-info/METADATA,sha256=ukDiaZOp_bowNIAa1bxJy5yIlcOIYW3BSdFD_AI8KVs,22957
+tigerbeetle-0.16.29.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+tigerbeetle-0.16.29.dist-info/RECORD,,

tigerbeetle-0.16.28.dist-info/RECORD

@@ -1,14 +0,0 @@
-tigerbeetle/__init__.py,sha256=3cmov0HPkByJw1znLKoTqIuZUEeEQMEg1fflWY3ty9A,168
-tigerbeetle/bindings.py,sha256=P7BDB1nq72D-fL6FhqBpCGfVX2FPore6x3T1d4JTiKk,25990
-tigerbeetle/client.py,sha256=sniZ6IATiAGWgtRrd60Xtfcixw4cMOaVZEUUP6sFfzM,8774
-tigerbeetle/lib.py,sha256=JKreDxdw_YejtH7F135_m6I6mNBSwjokaI6QicKm7cc,2425
-tigerbeetle/lib/aarch64-linux-gnu.2.27/libtb_client.so,sha256=-f3dgm0y557BKAXHjZyzFazHpeWr4pN3SVLNmGbWAjA,3751112
-tigerbeetle/lib/aarch64-linux-musl/libtb_client.so,sha256=8sBQU1R2la_AjZ7jh3p3CnafGh-i9J-dNxNx_elVYiY,3728360
-tigerbeetle/lib/aarch64-macos/libtb_client.dylib,sha256=HebPPhGsAYhZnTPHt9glx-EzFiOZjmeGc8rBsBjGA8k,516960
-tigerbeetle/lib/x86_64-linux-gnu.2.27/libtb_client.so,sha256=AnMz89Xbz2kDLWHh8srW4ZRRn77nJmxKc2NX8nqZB8I,3667392
-tigerbeetle/lib/x86_64-linux-musl/libtb_client.so,sha256=GzBbQn5CguA3uO_YJqYVG5Hw4QEE5g3RJfxoMaGzJ1M,3664384
-tigerbeetle/lib/x86_64-macos/libtb_client.dylib,sha256=zfEDtF2CIoFNjmZit9cpiNTZHtGfL3giGCTOU8HJO5I,729284
-tigerbeetle/lib/x86_64-windows/tb_client.dll,sha256=C2jKv4vz8-QKKg5KcUjTGJCes_3btEagnnvMsqsEJRc,820224
-tigerbeetle-0.16.28.dist-info/METADATA,sha256=dSXrE44EK9PIoQ1KU2PMuXW8Te9aDZ0_BeUon0OKr7c,22957
-tigerbeetle-0.16.28.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-tigerbeetle-0.16.28.dist-info/RECORD,,