31 lines
502 B
Ruby
31 lines
502 B
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
class Mastodon::RackMiddleware
|
||
|
|
def initialize(app)
|
||
|
|
@app = app
|
||
|
|
end
|
||
|
|
|
||
|
|
def call(env)
|
||
|
|
@app.call(env)
|
||
|
|
ensure
|
||
|
|
clean_up_sockets!
|
||
|
|
end
|
||
|
|
|
||
|
|
private
|
||
|
|
|
||
|
|
def clean_up_sockets!
|
||
|
|
clean_up_redis_socket!
|
||
|
|
clean_up_statsd_socket!
|
||
|
|
end
|
||
|
|
|
||
|
|
def clean_up_redis_socket!
|
||
|
|
Thread.current[:redis]&.close
|
||
|
|
Thread.current[:redis] = nil
|
||
|
|
end
|
||
|
|
|
||
|
|
def clean_up_statsd_socket!
|
||
|
|
Thread.current[:statsd_socket]&.close
|
||
|
|
Thread.current[:statsd_socket] = nil
|
||
|
|
end
|
||
|
|
end
|