Use more effective lookup

1 files changed, 8 insertions, 22 deletions

diff --git a/src/fake_monitor.cc b/src/fake_monitor.cc
index 336ad9c..be775e1 100644
--- a/src/fake_monitor.cc
+++ b/src/fake_monitor.cc
@@ -46,26 +46,6 @@ public:
     return machines_.size();
   }
 
-  uint32_t id(size_t index) const override {
-    auto it = machines_.begin();
-    while (index--) ++it;
-    if (it == machines_.end()) {
-      assert(false);
-      return static_cast<uint32_t>(-1);
-    }
-    return it->first;
-  }
-
-  Machine machine_at(size_t index) const override {
-    auto it = machines_.begin();
-    while (index--) ++it;
-    if (it == machines_.end()) {
-      assert(false);
-      return Machine();
-    }
-    return it->second.data;
-  }
-
   Machine machine(uint32_t id) const override {
     auto it = machines_.find(id);
     if (it == machines_.end()) {
@@ -193,6 +173,12 @@ private:
     }
   }
 
+  uint32_t id_at(size_t index) const {
+    auto it = machines_.begin();
+    while (index--) ++it;
+    return it->first;
+  }
+
   void fiddle_jobs(Looper*, uint32_t) {
     timer_ = 0;
 
@@ -208,11 +194,11 @@ private:
       auto host = rand() % machines_.size();
       auto target = rand() % machines_.size();
       do {
-        auto tgtid = this->id(target);
+        auto tgtid = id_at(target);
         auto& entry = machines_[tgtid];
         if (entry.active < entry.data.max_jobs) {
           entry.active++;
-          add_job(id, host != target ? this->id(host) : tgtid, tgtid);
+          add_job(id, host != target ? id_at(host) : tgtid, tgtid);
           break;
         }
         target = rand() % machines_.size();