about summary refs log tree commit diff
diff options
context:
space:
mode:
authorShea Levy <shea@shealevy.com>2017-04-26T16·38-0400
committerShea Levy <shea@shealevy.com>2017-04-26T18·15-0400
commit4bc00760f9bc36d5b4e8bba7de9bd71a30d7f31a (patch)
treec3b20ad676ed2a3e6f8cdfe4a2e4f04aad111a2b
parent98a2adb1359ec35ac8da7a52754d2290531f8cef (diff)
Add Store nesting to fix import-from-derivation within filterSource
-rw-r--r--src/libstore/remote-store.cc2
-rw-r--r--src/libutil/pool.hh16
2 files changed, 18 insertions, 0 deletions
diff --git a/src/libstore/remote-store.cc b/src/libstore/remote-store.cc
index bc9ef3d47c5e..af59d51106fc 100644
--- a/src/libstore/remote-store.cc
+++ b/src/libstore/remote-store.cc
@@ -414,7 +414,9 @@ Path RemoteStore::addToStore(const string & name, const Path & _srcPath,
     try {
         conn->to.written = 0;
         conn->to.warn = true;
+        connections->incCapacity();
         dumpPath(srcPath, conn->to, filter);
+        connections->decCapacity();
         conn->to.warn = false;
         conn->processStderr();
     } catch (SysError & e) {
diff --git a/src/libutil/pool.hh b/src/libutil/pool.hh
index 20df21948849..7033090020ef 100644
--- a/src/libutil/pool.hh
+++ b/src/libutil/pool.hh
@@ -68,6 +68,22 @@ public:
         state_->max = max;
     }
 
+    void incCapacity()
+    {
+        auto state_(state.lock());
+        state_->max++;
+        /* we could wakeup here, but this is only used when we're
+         * about to nest Pool usages, and we want to save the slot for
+         * the nested use if we can
+         */
+    }
+
+    void decCapacity()
+    {
+        auto state_(state.lock());
+        state_->max--;
+    }
+
     ~Pool()
     {
         auto state_(state.lock());