1 files changed, 15 insertions, 0 deletions

@@ -2,10 +2,14 @@
 
 #include <curlcpp/curl_easy.h>
 
+#include <chrono>
 #include <fstream>
 #include <sstream>
+#include <thread>
 
 Magick::Image imagestore::get(std::string key, std::string url) const {
+  std::lock_guard lock(mutex_);
+
   std::string filename = path_ + "/" + key;
 
   Magick::Image pic;
@@ -13,6 +17,9 @@ Magick::Image imagestore::get(std::string key, std::string url) const {
   if (std::ifstream(filename)) {
     pic.read(filename);
   } else {
+    // This is to avoid getting rate limited.
+    std::this_thread::sleep_for(std::chrono::milliseconds(50));
+
     std::ostringstream imgbuf;
     curl::curl_ios<std::ostringstream> imgios(imgbuf);
     curl::curl_easy imghandle(imgios);
@@ -50,3 +57,11 @@ Magick::Image imagestore::get(std::string key, std::string url) const {
 
   return pic;
 }
+
+void imagestore::cleanup() const {
+  std::lock_guard lock(mutex_);
+
+  for (const auto& entry : std::filesystem::directory_iterator(path_)) {
+    std::filesystem::remove_all(entry);
+  }
+}