This commit is contained in:
@@ -0,0 +1,70 @@
|
||||
require 'thread'
|
||||
|
||||
module Typhoeus
|
||||
|
||||
# The easy pool stores already initialized
|
||||
# easy handles for future use. This is useful
|
||||
# because creating them is expensive.
|
||||
#
|
||||
# @api private
|
||||
module Pool
|
||||
@mutex = Mutex.new
|
||||
@pid = Process.pid
|
||||
|
||||
# Releases easy into the pool. The easy handle is
|
||||
# reset before it gets back in.
|
||||
#
|
||||
# @example Release easy.
|
||||
# Typhoeus::Pool.release(easy)
|
||||
def self.release(easy)
|
||||
easy.cookielist = "flush" # dump all known cookies to 'cookiejar'
|
||||
easy.cookielist = "all" # remove all cookies from memory for this handle
|
||||
easy.reset
|
||||
@mutex.synchronize { easies << easy }
|
||||
end
|
||||
|
||||
# Return an easy from the pool.
|
||||
#
|
||||
# @example Return easy.
|
||||
# Typhoeus::Pool.get
|
||||
#
|
||||
# @return [ Ethon::Easy ] The easy.
|
||||
def self.get
|
||||
@mutex.synchronize do
|
||||
if @pid == Process.pid
|
||||
easies.pop
|
||||
else
|
||||
# Process has forked. Clear all easies to avoid sockets being
|
||||
# shared between processes.
|
||||
@pid = Process.pid
|
||||
easies.clear
|
||||
nil
|
||||
end
|
||||
end || Ethon::Easy.new
|
||||
end
|
||||
|
||||
# Clear the pool
|
||||
def self.clear
|
||||
@mutex.synchronize { easies.clear }
|
||||
end
|
||||
|
||||
# Use yielded easy, will be released automatically afterwards.
|
||||
#
|
||||
# @example Use easy.
|
||||
# Typhoeus::Pool.with_easy do |easy|
|
||||
# # use easy
|
||||
# end
|
||||
def self.with_easy(&block)
|
||||
easy = get
|
||||
yield easy
|
||||
ensure
|
||||
release(easy) if easy
|
||||
end
|
||||
|
||||
private
|
||||
|
||||
def self.easies
|
||||
@easies ||= []
|
||||
end
|
||||
end
|
||||
end
|
||||
Reference in New Issue
Block a user