Hacky way to solve things

Author: tli2

src/include/brain/brain.h

@@ -15,6 +15,8 @@
 #include <string>
 #include <cstdio>
 #include <utility>
+#include "capnp/ez-rpc.h"
+#include "peloton/capnp/peloton_service.capnp.h"
 #include "common/notifiable_task.h"
 
 namespace peloton {
@@ -28,7 +30,19 @@ class BrainEnvironment {
 class BrainJob {
  public:
   // TODO(tianyu): Extend this interface for richer interaction
-  virtual void RunJob(BrainEnvironment &env) { printf("foo"); };
+  virtual void RunJob(BrainEnvironment *) = 0;
+};
+
+class ExampleBrainJob: public BrainJob {
+ public:
+   void RunJob(BrainEnvironment *) override {
+     // TODO(tianyu): Replace with real address
+     capnp::EzRpcClient client("localhost:15445");
+     PelotonService::Client peloton_service = client.getMain<PelotonService>();
+     auto request = peloton_service.createIndexRequest();
+     request.getRequest().setIndexKeys(42);
+     auto response = request.send().wait(client.getWaitScope());
+   }
 };
 
 class Brain {
@@ -39,11 +53,14 @@ class Brain {
   inline void RegisterJob(const struct timeval *period,
                           std::string name,
                           BrainJob &job) {
-    auto callback = [&](int, short, void *env) {
-      job.RunJob(*reinterpret_cast<BrainEnvironment *>(env));
+    auto callback = [](int, short, void *pair) {
+      auto *job_env_pair = reinterpret_cast<std::pair<BrainJob *, BrainEnvironment *> *>(pair);
+      job_env_pair->first->RunJob(job_env_pair->second);
     };
+    // TODO(tianyu) Deal with this memory
+    auto *pair = new std::pair<BrainJob *, BrainEnvironment *>(&job, &env_);
     job_handles_[name] =
-        scheduler_.RegisterPeriodicEvent(period, callback, &env);
+        scheduler_.RegisterPeriodicEvent(period, callback, pair);
   }
 
   inline void Run() {
@@ -57,8 +74,7 @@ class Brain {
  private:
   NotifiableTask scheduler_;
   std::unordered_map<std::string, struct event *> job_handles_;
-  // TODO(tianyu): May need to have multiple env instead of shared
-  BrainEnvironment env;
+  BrainEnvironment env_;
 };
 }
 }

src/include/capnp/peloton_service.capnp

@@ -12,7 +12,7 @@ struct CreateIndexRequest {
 }
 
 struct CreateIndexResponse {
-
+  message @0 :Text;
 }
 
 interface PelotonService {

src/main/peloton/peloton.cpp

@@ -47,9 +47,10 @@ void RunPelotonBrain() {
   peloton::brain::Brain brain;
   evthread_use_pthreads();
   // TODO(tianyu): register jobs here
-  struct timeval *one_minute;
-  one_minute->tv_sec = 60;
-  brain.RegisterJob(one_minute, "test", peloton::brain::BrainJob());
+  struct timeval one_second;
+  one_second.tv_sec = 1;
+  peloton::brain::ExampleBrainJob job;
+  brain.RegisterJob(&one_second, "test", job);
   brain.Run();
 }