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
123
124
125
126
127
128
129
130
131
132
133
134
135
|
<?php
namespace SMW\Elastic\Connection;
use Elasticsearch\ClientBuilder;
use SMW\Elastic\Exception\ClientBuilderNotFoundException;
use SMW\ApplicationFactory;
use SMW\Connection\ConnectionProvider as IConnectionProvider;
use SMW\Options;
use Psr\Log\LoggerAwareTrait;
use Onoi\Cache\Cache;
/**
* @private
*
* @license GNU GPL v2+
* @since 3.0
*
* @author mwjames
*/
class ConnectionProvider implements IConnectionProvider {
use LoggerAwareTrait;
/**
* @var Options
*/
private $options;
/**
* @var Cache
*/
private $cache;
/**
* @var ElasticClient
*/
private $connection;
/**
* @since 3.0
*
* @param Options $options
* @param Cache $cache
*/
public function __construct( Options $options, Cache $cache ) {
$this->options = $options;
$this->cache = $cache;
}
/**
* @see ConnectionProvider::getConnection
*
* @since 3.0
*
* @return Connection
*/
public function getConnection() {
if ( $this->connection !== null ) {
return $this->connection;
}
$params = [
'hosts' => $this->options->get( 'endpoints' ),
'retries' => $this->options->dotGet( 'connection.retries', 1 ),
'client' => [
// controls the request timeout
'timeout' => $this->options->dotGet( 'connection.timeout', 30 ),
// controls the original connection timeout duration
'connect_timeout' => $this->options->dotGet( 'connection.connect_timeout', 30 )
]
// Use `singleHandler` if you know you will never need async capabilities,
// since it will save a small amount of overhead by reducing indirection
// 'handler' => ClientBuilder::singleHandler()
];
if ( $this->hasClientBuilder() ) {
$this->connection = new Client(
ClientBuilder::fromConfig( $params, true ),
$this->cache,
$this->options
);
} else {
$this->connection = new DummyClient();
}
$this->connection->setLogger(
$this->logger
);
$this->logger->info(
[
'Connection',
'{provider} : {hosts}'
],
[
'role' => 'developer',
'provider' => 'elastic',
'hosts' => $params['hosts']
]
);
return $this->connection;
}
/**
* @see ConnectionProvider::releaseConnection
*
* @since 3.0
*/
public function releaseConnection() {
$this->connection = null;
}
private function hasClientBuilder() {
if ( $this->options->dotGet( 'is.elasticstore', false ) === false ) {
return false;
}
// Fail hard because someone selected the ElasticStore but forgot to install
// the elastic interface!
if ( !class_exists( '\Elasticsearch\ClientBuilder' ) ) {
throw new ClientBuilderNotFoundException();
}
return true;
}
}
|
