+ new API: ZEND_ASYNC_EV_HANDLER_REGISTER/ZEND_ASYNC_EV_HANDLER_UNREGISTER The API is designed to allow adding additional handlers to the reactor loop without modifying the reactor's code.

Author: EdmondDantes

Zend/zend_async_API.c

@@ -1602,3 +1602,54 @@ static void zend_async_main_handlers_shutdown(void)
 		vector->in_execution = false;
 	}
 }
+
+ZEND_API bool zend_async_ev_handler_register(const zend_async_ev_handler_fn handler)
+{
+	if (UNEXPECTED(!ZEND_ASYNC_IS_READY)) {
+		zend_throw_error(NULL, "True Async is no longer in the Ready state");
+		return false;
+	}
+	
+	if (UNEXPECTED(handler == NULL)) {
+		return false;
+	}
+	
+	/* Check if handler already exists */
+	for (int i = 0; i < ZEND_ASYNC_G(ev_handlers_count); i++) {
+		if (ZEND_ASYNC_G(ev_handlers)[i] == handler) {
+			return false; /* Already registered */
+		}
+	}
+	
+	/* Check if we have space */
+	if (ZEND_ASYNC_G(ev_handlers_count) >= 4) {
+		zend_throw_error(NULL, "Too many event handlers registered, maximum is 4");
+		return false;
+	}
+	
+	/* Add handler */
+	ZEND_ASYNC_G(ev_handlers)[ZEND_ASYNC_G(ev_handlers_count)] = handler;
+	ZEND_ASYNC_G(ev_handlers_count)++;
+	
+	return true;
+}
+
+ZEND_API bool zend_async_ev_handler_unregister(const zend_async_ev_handler_fn handler)
+{
+	if (UNEXPECTED(handler == NULL)) {
+		return false;
+	}
+	
+	for (int i = 0; i < ZEND_ASYNC_G(ev_handlers_count); i++) {
+		if (ZEND_ASYNC_G(ev_handlers)[i] == handler) {
+			/* Shift remaining handlers */
+			for (int j = i; j < ZEND_ASYNC_G(ev_handlers_count) - 1; j++) {
+				ZEND_ASYNC_G(ev_handlers)[j] = ZEND_ASYNC_G(ev_handlers)[j + 1];
+			}
+			ZEND_ASYNC_G(ev_handlers_count)--;
+			return true;
+		}
+	}
+	
+	return false;
+}

Zend/zend_async_API.h

@@ -139,6 +139,9 @@ typedef struct _zend_async_group_s zend_async_group_t;
 typedef struct _zend_fcall_s zend_fcall_t;
 typedef void (*zend_coroutine_entry_t)(void);
 
+/* The event loop additional handler function type */
+typedef void (*zend_async_ev_handler_fn)(bool no_wait);
+
 /* Channel method function types */
 typedef bool (*zend_channel_send_t)(zend_async_channel_t *channel, zval *value);
 typedef bool (*zend_channel_receive_t)(zend_async_channel_t *channel, zval *result);
@@ -1176,6 +1179,9 @@ typedef struct {
 	zend_coroutine_t *scheduler;
 	/* Exit exception object */
 	zend_object *exit_exception;
+	/* An array of additional handlers for the EventLoop that will be invoked by the reactor */
+	zend_async_ev_handler_fn ev_handlers[4];     /* Static array for event handlers */
+	int ev_handlers_count;                       /* Current number of handlers */
 } zend_async_globals_t;
 
 BEGIN_EXTERN_C()
@@ -1210,6 +1216,17 @@ END_EXTERN_C()
 #define ZEND_ASYNC_MAIN_SCOPE ZEND_ASYNC_G(main_scope)
 #define ZEND_ASYNC_SCHEDULER ZEND_ASYNC_G(scheduler)
 
+#define ZEND_ASYNC_EV_HANDLER_REGISTER(handler) zend_async_ev_handler_register(handler)
+#define ZEND_ASYNC_EV_HANDLER_UNREGISTER(handler) zend_async_ev_handler_unregister(handler)
+
+/* Call all registered event loop handlers */
+#define ZEND_ASYNC_EV_HANDLERS_CALL(no_wait) do { \
+    int _count = ZEND_ASYNC_G(ev_handlers_count); \
+    for (int _i = 0; _i < _count; _i++) { \
+        ZEND_ASYNC_G(ev_handlers)[_i](no_wait); \
+    } \
+} while (0)
+
 #define ZEND_ASYNC_INCREASE_EVENT_COUNT  if (ZEND_ASYNC_G(active_event_count) < UINT_MAX) { \
 		ZEND_ASYNC_G(active_event_count)++; \
 	} else { \
@@ -1245,6 +1262,10 @@ void zend_async_api_shutdown(void);
 void zend_async_globals_ctor(void);
 void zend_async_globals_dtor(void);
 
+/* LibUv event loop handlers API */
+ZEND_API bool zend_async_ev_handler_register(zend_async_ev_handler_fn handler);
+ZEND_API bool zend_async_ev_handler_unregister(zend_async_ev_handler_fn handler);
+
 ZEND_API const char * zend_async_get_api_version(void);
 ZEND_API int zend_async_get_api_version_number(void);