initializers/z_dalli.rb: initialize Rails.cache.

config/initializers/z_dalli.rb

@@ -1,9 +1,14 @@
-unless defined?(MEMCACHE)
+Rails.application.configure do
-  MEMCACHE = Dalli::Client.new(Danbooru.config.memcached_servers, :namespace => Danbooru.config.app_name.gsub(/[^A-Za-z0-9]/, "_"))
-end
-
   begin
-  MEMCACHE.get("x")
+    if Rails.env.test?
+      config.cache_store = :memory_store, { size: 32.megabytes }
+      Rails.cache = ActiveSupport::Cache.lookup_store(Rails.application.config.cache_store)
+    else
+      config.cache_store = :dalli_store, Danbooru.config.memcached_servers, { namespace: Danbooru.config.safe_app_name }
+      Rails.cache = ActiveSupport::Cache.lookup_store(Rails.application.config.cache_store)
+
+      Rails.cache.dalli.alive!
+    end
   rescue Dalli::RingError => e
     puts "-" * 40
     puts "WARNING! MEMCACHE SERVER NOT FOUND! You will experience performance degradation."
@@ -14,11 +19,8 @@ rescue Dalli::RingError => e
     end
     puts "-- END STACKTRACE --"
     puts "-" * 40
-  MEMCACHE = MemcacheMock.new
-end
 
-if Rails.env.production?
+    config.cache_store = :memory_store, { size: 32.megabytes }
-  Rails.application.configure do
+    Rails.cache = ActiveSupport::Cache.lookup_store(Rails.application.config.cache_store)
-    config.cache_store = :dalli_store, Danbooru.config.memcached_servers
   end
 end