Fix Throttler reset (#1423)

nautechsystems · Dec 22, 2023 · ee5e34f · ee5e34f
1 parent 061e8f3
commit ee5e34f
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 0 deletions.
diff --git a/nautilus_trader/common/component.pxd b/nautilus_trader/common/component.pxd
@@ -183,6 +183,7 @@ cdef class Throttler:
     cdef readonly int sent_count
     """If count of messages sent from the throttler.\n\n:returns: `int`"""
 
+    cpdef void reset(self)
     cpdef double used(self)
     cpdef void send(self, msg)
     cdef int64_t _delta_next(self)

diff --git a/nautilus_trader/common/component.pyx b/nautilus_trader/common/component.pyx
@@ -1436,6 +1436,16 @@ cdef class Throttler:
         """
         return len(self._buffer)
 
+    cpdef void reset(self):
+        """
+        Reset the state of the throttler.
+        """
+        self._buffer.clear()
+        self._warm = False
+        self.recv_count = 0
+        self.sent_count = 0
+        self.is_limiting = False
+
     cpdef double used(self):
         """
         Return the percentage of maximum rate currently used.

diff --git a/nautilus_trader/risk/engine.pyx b/nautilus_trader/risk/engine.pyx
@@ -386,6 +386,8 @@ cdef class RiskEngine(Component):
     cpdef void _reset(self):
         self.command_count = 0
         self.event_count = 0
+        self._order_submit_throttler.reset()
+        self._order_modify_throttler.reset()
 
     cpdef void _dispose(self):
         pass