Class: Async::WebDriver::Bridge::Pool

Inherits:
Object
  • Object
show all
Defined in:
lib/async/webdriver/bridge/pool.rb

Overview

A pool of sessions.

begin
bridge = Async::WebDriver::Bridge::Pool.start(Async::WebDriver::Bridge::Chrome.new)
session = bridge.session
ensure
bridge&.close
end

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(bridge, capabilities: bridge.default_capabilities, minimum: 2) ⇒ Pool

Initialize the session pool.



34
35
36
37
38
39
40
41
42
43
 
# File 'lib/async/webdriver/bridge/pool.rb', line 34

def initialize(bridge, capabilities: bridge.default_capabilities, minimum: 2)
	@bridge = bridge
	@capabilities = capabilities
	@minimum = minimum
	
	@thread = nil
	
	@waiting = Thread::Queue.new
	@sessions = Thread::Queue.new
end

Class Method Details

.start(bridge, **options) ⇒ Object

Create a new session pool and start it.



24
25
26
27
28
 
# File 'lib/async/webdriver/bridge/pool.rb', line 24

def self.start(bridge, **options)
	self.new(bridge, **options).tap do |pool|
		pool.start
	end
end

Instance Method Details

#close ⇒ Object

Close the session pool.



46
47
48
49
50
51
52
53
54
55
56
57
58
59
 
# File 'lib/async/webdriver/bridge/pool.rb', line 46

def close
	if @waiting
		@waiting.close
	end
	
	if @thread
		@thread.join
		@thread = nil
	end
	
	if @sessions
		@sessions.close
	end
end

#reuse(session) ⇒ Object 



109
110
111
112
113
 
# File 'lib/async/webdriver/bridge/pool.rb', line 109

def reuse(session)
	session.reset!
	
	@sessions << {"sessionId" => session.id, "capabilities" => session.capabilities}
end

#session(&block) ⇒ Object

Open a session.



89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
 
# File 'lib/async/webdriver/bridge/pool.rb', line 89

def session(&block)
	@waiting << true
	
	reply = @sessions.pop
	
	session = Session.open(@bridge.endpoint, reply["sessionId"], reply["capabilities"])
	
	return session unless block_given?
	
	begin
		yield session
		
		# Try to reuse the session for extreme performance:
		reuse(session)
		session = nil
	ensure
		session&.close
	end
end

#start ⇒ Object

Start the session pool.



66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
 
# File 'lib/async/webdriver/bridge/pool.rb', line 66

def start
	@thread ||= Thread.new do
		Sync do
			@bridge.start
			
			client = Client.open(@bridge.endpoint)
			
			@minimum.times do
				@waiting << true
			end
			
			while @waiting.pop
				session = prepare_session(client)
				@sessions << session
			end
		ensure
			client&.close
			@bridge.close
		end
	end
end