FEATURE: added method for draining idle connections

This commit is contained in:
Sam
2014-08-11 15:13:37 +10:00
parent eeff092ead
commit 9fc7c145b3
2 changed files with 80 additions and 0 deletions

View File

@ -0,0 +1,33 @@
class ActiveRecord::ConnectionAdapters::ConnectionPool
# drain all idle connections
# if idle_time is specified only connections idle for N seconds will be drained
def drain(idle_time=nil)
synchronize do
@available.clear
@connections.delete_if do |conn|
try_drain?(conn, idle_time)
end
@connections.each do |conn|
@available.add conn if !conn.in_use?
end
end
end
private
def try_drain?(conn, idle_time)
if !conn.in_use?
if !idle_time || conn.last_use < idle_time.seconds.ago
conn.disconnect!
return true
end
end
false
end
end