Add event queue overflow handling

1 files changed, 32 insertions, 19 deletions

diff --git a/mpv.py b/mpv.py
index 6469ebb..94d99c3 100644
--- a/mpv.py
+++ b/mpv.py
@@ -58,6 +58,9 @@ else:
 class ShutdownError(SystemError):
     pass
 
+class EventOverflowError(SystemError):
+    pass
+
 class MpvHandle(c_void_p):
     pass
 
@@ -886,8 +889,6 @@ class MPV(object):
         for event in _event_generator(self._event_handle):
             try:
                 eid = event.event_id.value
-                if not eid == MpvEventID.LOG_MESSAGE:
-                    print(event)
 
                 with self._event_handler_lock:
                     if eid == MpvEventID.SHUTDOWN:
@@ -919,8 +920,15 @@ class MPV(object):
                     if callback:
                         callback(ErrorCode.exception_for_ec(event.error), event.data)
 
+                if eid == MpvEventID.QUEUE_OVERFLOW:
+                    # cache list, since error handlers will unregister themselves
+                    for cb in list(self._command_reply_callbacks.values()):
+                        cb(EventOverflowError('libmpv event queue has flown over because events have not been processed fast enough'), None)
+
                 if eid == MpvEventID.SHUTDOWN:
                     _mpv_destroy(self._event_handle)
+                    for cb in list(self._command_reply_callbacks.values()):
+                        cb(ShutdownError('libmpv core has been shutdown'), None)
                     return
 
             except Exception as e:
@@ -971,6 +979,19 @@ class MPV(object):
         except ShutdownError:
             return
 
+    def _set_error_handler(self, future):
+        @self.event_callback('shutdown', 'queue-overflow')
+        def shutdown_handler(event):
+            nonlocal future
+            try:
+                if event.event_id.value == MpvEventID.SHUTDOWN:
+                    future.set_exception(ShutdownError('libmpv core has been shutdown'))
+                else:
+                    future.set_exception(EventOverflowError('libmpv event queue has flown over because events have not been processed fast enough'))
+            except InvalidStateError:
+                pass
+        return shutdown_handler.unregister_mpv_events
+
     @contextmanager
     def prepare_and_wait_for_property(self, name, cond=lambda val: val, level_sensitive=True, timeout=None):
         """Context manager that waits until ``cond`` evaluates to a truthy value on the named property. See
@@ -992,13 +1013,7 @@ class MPV(object):
             except InvalidStateError:
                 pass
         self.observe_property(name, observer)
-
-        @self.event_callback('shutdown')
-        def shutdown_handler(event):
-            try:
-                result.set_exception(ShutdownError('libmpv core has been shutdown'))
-            except InvalidStateError:
-                pass
+        err_unregister = self._set_error_handler(result)
 
         try:
             result.set_running_or_notify_cancel()
@@ -1012,7 +1027,7 @@ class MPV(object):
                 self.check_core_alive()
                 result.result(timeout)
         finally:
-            shutdown_handler.unregister_mpv_events()
+            err_unregister()
             self.unobserve_property(name, observer)
 
     def wait_for_event(self, *event_types, cond=lambda evt: True, timeout=None):
@@ -1041,13 +1056,6 @@ class MPV(object):
         """
         result = Future()
 
-        @self.event_callback('shutdown')
-        def shutdown_handler(event):
-            try:
-                result.set_exception(ShutdownError('libmpv core has been shutdown'))
-            except InvalidStateError:
-                pass
-
         @self.event_callback(*event_types)
         def target_handler(evt):
 
@@ -1063,6 +1071,8 @@ class MPV(object):
             except InvalidStateError:
                 pass
 
+        err_unregister = self._set_error_handler(result)
+
         try:
             result.set_running_or_notify_cancel()
             yield result
@@ -1070,7 +1080,7 @@ class MPV(object):
             result.result(timeout)
 
         finally:
-            shutdown_handler.unregister_mpv_events()
+            err_unregister()
             target_handler.unregister_mpv_events()
 
     def __del__(self):
@@ -1138,7 +1148,10 @@ class MPV(object):
                 result = result.unpack(decoder)
                 future.set_result(callback(error, result))
             except Exception as e:
-                future.set_exception(e)
+                try:
+                    future.set_exception(e)
+                except InvalidStateError:
+                    pass
 
         def abort():
             _mpv_abort_async_command(self._event_handle, id(future))