debian-mirror-gitlab/config/initializers/fix_local_cache_middleware.rb

25 lines
905 B
Ruby
Raw Normal View History

2017-08-17 22:00:37 +05:30
module LocalCacheRegistryCleanupWithEnsure
LocalCacheRegistry =
ActiveSupport::Cache::Strategy::LocalCache::LocalCacheRegistry
LocalStore =
ActiveSupport::Cache::Strategy::LocalCache::LocalStore
def call(env)
LocalCacheRegistry.set_cache_for(local_cache_key, LocalStore.new)
2018-03-17 18:26:18 +05:30
response = @app.call(env) # rubocop:disable Gitlab/ModuleWithInstanceVariables
2017-08-17 22:00:37 +05:30
response[2] = ::Rack::BodyProxy.new(response[2]) do
LocalCacheRegistry.set_cache_for(local_cache_key, nil)
end
cleanup_after_response = true # ADDED THIS LINE
response
rescue Rack::Utils::InvalidParameterError
[400, {}, []]
ensure # ADDED ensure CLAUSE to cleanup when something is thrown
LocalCacheRegistry.set_cache_for(local_cache_key, nil) unless
cleanup_after_response
end
end
ActiveSupport::Cache::Strategy::LocalCache::Middleware
.prepend(LocalCacheRegistryCleanupWithEnsure)