Implement a first pass at actors in the API. (#242)

* Implement actor field for tasks * Implement actor management in local scheduler. * initial python frontend for actors * import actors on worker * IPython code completion and tests * prepare creating actors through local schedulers * add actor id to PyTask * submit actor calls to local scheduler * starting to integrate * simple fix * Fixes from rebasing. * more work on python actors * Improve local scheduler actor handlers. * Pass actor ID to local scheduler when connecting a client. * first working version of actors * fixing actors * fix creating two copies of the same actor * fix actors * remove sleep * get rid of export synchronization * update * insert actor methods into the queue in the right order * remove print statements * make it compile again after rebase * Minor updates. * fix python actor ids * Pass actor_id to start_worker. * add test * Minor changes. * Update actor tests. * Temporary plan for import counter. * Temporarily fix import counters. * Fix some tests. * Fixes. * Make actor creation non-blocking. * Fix test? * Fix actors on Python 2. * fix rare case. * Fix python 2 test. * More tests. * Small fixes. * Linting. * Revert tensorflow version to 0.12.0 temporarily. * Small fix. * Enhance inheritance test.
2026-08-19 12:30:27 +08:00 · 2017-02-15 00:10:05 -08:00
parent 072eadd57f
commit 12a68e84d2
32 changed files with 1812 additions and 117 deletions
@@ -0,0 +1,47 @@
+#ifndef ACTOR_NOTIFICATION_TABLE_H
+#define ACTOR_NOTIFICATION_TABLE_H
+
+#include "task.h"
+#include "db.h"
+#include "table.h"
+
+typedef struct {
+  /** The ID of the actor. */
+  actor_id actor_id;
+  /** The ID of the local scheduler that is responsible for the actor. */
+  db_client_id local_scheduler_id;
+} actor_info;
+
+/*
+ *  ==== Subscribing to the actor notification table ====
+ */
+
+/* Callback for subscribing to the local scheduler table. */
+typedef void (*actor_notification_table_subscribe_callback)(actor_info info,
+                                                            void *user_context);
+
+/**
+ * Register a callback to process actor notification events.
+ *
+ * @param db_handle Database handle.
+ * @param subscribe_callback Callback that will be called when the local
+ *        scheduler event happens.
+ * @param subscribe_context Context that will be passed into the
+ *        subscribe_callback.
+ * @param retry Information about retrying the request to the database.
+ * @return Void.
+ */
+void actor_notification_table_subscribe(
+    db_handle *db_handle,
+    actor_notification_table_subscribe_callback subscribe_callback,
+    void *subscribe_context,
+    retry_info *retry);
+
+/* Data that is needed to register local scheduler table subscribe callbacks
+ * with the state database. */
+typedef struct {
+  actor_notification_table_subscribe_callback subscribe_callback;
+  void *subscribe_context;
+} actor_notification_table_subscribe_data;
+
+#endif /* ACTOR_NOTIFICATION_TABLE_H */