sserver ETH: Avoid repeatedly starting DAG light calculation threads.

btccom · May 23, 2019 · 9250057 · 9250057
1 parent 0a93fc4
commit 9250057
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/src/eth/StratumServerEth.cc b/src/eth/StratumServerEth.cc
@@ -247,6 +247,14 @@ void EthashCalculator::buildDagCache(uint64_t height) {
   }
 
   auto buildLightWithLock = [this](uint64_t height, uint64_t epoch) {
+    {
+      ScopeLock sl(lock_);
+      if (buildingLightCaches_.find(epoch) != buildingLightCaches_.end()) {
+        return;
+      }
+      buildingLightCaches_.insert(epoch);
+    }
+
     LOG(INFO) << "building DAG cache for block height " << height << " (epoch "
               << epoch << ")";
     time_t beginTime = time(nullptr);
@@ -260,6 +268,8 @@ void EthashCalculator::buildDagCache(uint64_t height) {
               << ") built within " << (time(nullptr) - beginTime) << " seconds";
 
     ScopeLock sl(lock_);
+    buildingLightCaches_.erase(epoch);
+
     if (lightCaches_[epoch] != nullptr) {
       // Other threads have added the same light.
       ethash_light_delete(light);

diff --git a/src/eth/StratumServerEth.h b/src/eth/StratumServerEth.h
@@ -76,6 +76,7 @@ class EthashCalculator {
 
   std::mutex lock_;
   std::map<uint64_t /*epoch*/, ethash_light_t> lightCaches_;
+  std::set<uint64_t /*epoch*/> buildingLightCaches_;
   std::queue<uint64_t> lightEpochs_;
   string cacheFile_;