blob: b5fa681de3e8f00459c32bdbb806342359fc8bc6 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
<?php
namespace Elastica\Connection;
use Elastica\Client;
use Elastica\Connection;
use Elastica\Connection\Strategy\StrategyInterface;
use Exception;
/**
* Description of ConnectionPool.
*
* @author chabior
*/
class ConnectionPool
{
/**
* @var array|\Elastica\Connection[] Connections array
*/
protected $_connections;
/**
* @var \Elastica\Connection\Strategy\StrategyInterface Strategy for connection
*/
protected $_strategy;
/**
* @var callback Function called on connection fail
*/
protected $_callback;
/**
* @param array $connections
* @param \Elastica\Connection\Strategy\StrategyInterface $strategy
* @param callback $callback
*/
public function __construct(array $connections, StrategyInterface $strategy, $callback = null)
{
$this->_connections = $connections;
$this->_strategy = $strategy;
$this->_callback = $callback;
}
/**
* @param \Elastica\Connection $connection
*
* @return $this
*/
public function addConnection(Connection $connection)
{
$this->_connections[] = $connection;
return $this;
}
/**
* @param array|\Elastica\Connection[] $connections
*
* @return $this
*/
public function setConnections(array $connections)
{
$this->_connections = $connections;
return $this;
}
/**
* @return bool
*/
public function hasConnection()
{
foreach ($this->_connections as $connection) {
if ($connection->isEnabled()) {
return true;
}
}
return false;
}
/**
* @return array
*/
public function getConnections()
{
return $this->_connections;
}
/**
* @throws \Elastica\Exception\ClientException
*
* @return \Elastica\Connection
*/
public function getConnection()
{
return $this->_strategy->getConnection($this->getConnections());
}
/**
* @param \Elastica\Connection $connection
* @param \Exception $e
* @param Client $client
*/
public function onFail(Connection $connection, Exception $e, Client $client)
{
$connection->setEnabled(false);
if ($this->_callback) {
call_user_func($this->_callback, $connection, $e, $client);
}
}
/**
* @return \Elastica\Connection\Strategy\StrategyInterface
*/
public function getStrategy()
{
return $this->_strategy;
}
}
|