Make S3 downloads slightly more interruptable

DeterminateSystems · Oct 11, 2024 · d38f62f · d38f62f
1 parent 0500fba
commit d38f62f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/libstore/s3-binary-cache-store.cc b/src/libstore/s3-binary-cache-store.cc
@@ -9,6 +9,7 @@
 #include "globals.hh"
 #include "compression.hh"
 #include "filetransfer.hh"
+#include "signals.hh"
 
 #include <aws/core/Aws.h>
 #include <aws/core/VersionConfig.h>
@@ -117,6 +118,7 @@ class RetryStrategy : public Aws::Client::DefaultRetryStrategy
 {
  bool ShouldRetry(const Aws::Client::AWSError<Aws::Client::CoreErrors>& error, long attemptedRetries) const override
  {
+ checkInterrupt();
  auto retry = Aws::Client::DefaultRetryStrategy::ShouldRetry(error, attemptedRetries);
  if (retry)
  printError("AWS error '%s' (%s), will retry in %d ms",