wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-09-12 12:51:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

implement key value store for sharing reusable variables

Browse Source
This commit is contained in:
Robert Nishihara
2016-07-22 18:43:33 -07:00
parent baa4b7cae3
commit 03f1830cd0
10 changed files with 347 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/worker.h
+11 -1
View File
@@ -23,9 +23,16 @@ using grpc::Channel;
using grpc::ClientContext;
using grpc::ClientWriter;
struct ReusableVariable {
std::string variable_name;
std::string initializer;
std::string reinitializer;
};
struct WorkerMessage {
Task task;
std::string function;
std::string function; // Used for importing remote functions.
ReusableVariable reusable_variable; // Used for importing reusable variables.
};
class WorkerServiceImpl final : public WorkerService::Service {
@@ -34,6 +41,7 @@ public:
Status ExecuteTask(ServerContext* context, const ExecuteTaskRequest* request, ExecuteTaskReply* reply) override;
Status ImportFunction(ServerContext* context, const ImportFunctionRequest* request, ImportFunctionReply* reply) override;
Status Die(ServerContext* context, const DieRequest* request, DieReply* reply) override;
Status ImportReusableVariable(ServerContext* context, const ImportReusableVariableRequest* request, AckReply* reply) override;
private:
std::string worker_address_;
MessageQueue<WorkerMessage*> send_queue_;
@@ -100,6 +108,8 @@ class Worker {
void task_info(ClientContext &context, TaskInfoRequest &request, TaskInfoReply &reply);
// export function to workers
bool export_function(const std::string& function);
// export reusable variable to workers
void export_reusable_variable(const std::string& name, const std::string& initializer, const std::string& reinitializer);
private:
bool connected_;