Simple idea for a mirror site

1 files changed, 44 insertions, 0 deletions

diff --git a/lib/active_file/sites/mirror_site.rb b/lib/active_file/sites/mirror_site.rb
new file mode 100644
index 0000000000..86dd906be7
--- /dev/null
+++ b/lib/active_file/sites/mirror_site.rb
@@ -0,0 +1,44 @@
+class ActiveFile::Sites::MirrorSite < ActiveFile::Site
+  attr_reader :sites
+
+  def initialize(sites:)
+    @sites = sites
+  end
+
+  def upload(key, data)
+    perform_across_sites :upload, key, data
+  end
+
+  def download(key)
+    sites.detect { |site| site.exist?(key) }.download(key)
+  end
+
+  def delete(key)
+    perform_across_sites :delete, key
+  end
+
+  def exists?(key)
+    perform_across_sites(:exists?, key).any?
+  end
+
+
+  def byte_size(key)
+    primary_site.byte_size(key)
+  end
+
+  def checksum(key)
+    primary_site.checksum(key)
+  end
+
+  private
+    def primary_site
+      sites.first
+    end
+
+    def perform_across_sites(method, **args)
+      # FIXME: Convert to be threaded
+      sites.collect do |site|
+        site.send method, **args
+      end   
+    end
+end