3 files changed, 25 insertions, 2 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);
+  }
+}
@@ -3,6 +3,7 @@
 
 #include <Magick++.h>
 
+#include <mutex>
 #include <string>
 
 class imagestore {
@@ -11,8 +12,12 @@ class imagestore {
 
   Magick::Image get(std::string key, std::string url) const;
 
+  void cleanup() const;
+
  private:
   std::string path_;
+
+  mutable std::mutex mutex_;
 };
 
 #endif /* end of include guard: IMAGESTORE_H_80B1E49F */
@@ -36,7 +36,8 @@ class server {
     socket_.listen(port_);
     socket_.start_accept();
 
-    asio::post(std::bind(&server::cleanup_thread, this));
+    std::thread cleanup(std::bind(&server::cleanup_thread, this));
+    cleanup.detach();
 
     std::cout << "Listening on port " << port_ << "..." << std::endl;
 
@@ -145,7 +146,9 @@ class server {
 
   void cleanup_thread() {
     for (;;) {
-      // sleep
+      std::this_thread::sleep_for(std::chrono::hours(24));
+
+      images_.cleanup();
     }
   }