MediaWiki  master
RedisConnRef.php
Go to the documentation of this file.
1 <?php
22 
31 class RedisConnRef implements LoggerAwareInterface {
33  protected $pool;
35  protected $conn;
37  protected $server;
39  protected $lastError;
40 
44  protected $logger;
45 
49  const AUTH_NO_ERROR = 200;
50 
54  const AUTH_ERROR_TEMPORARY = 201;
55 
59  const AUTH_ERROR_PERMANENT = 202;
60 
67  public function __construct(
68  RedisConnectionPool $pool, $server, Redis $conn, LoggerInterface $logger
69  ) {
70  $this->pool = $pool;
71  $this->server = $server;
72  $this->conn = $conn;
73  $this->logger = $logger;
74  }
75 
76  public function setLogger( LoggerInterface $logger ) {
77  $this->logger = $logger;
78  }
79 
84  public function getServer() {
85  return $this->server;
86  }
87 
88  public function getLastError() {
89  return $this->lastError;
90  }
91 
92  public function clearLastError() {
93  $this->lastError = null;
94  }
95 
104  public function __call( $name, $arguments ) {
105  // Work around https://github.com/nicolasff/phpredis/issues/70
106  $lname = strtolower( $name );
107  if (
108  ( $lname === 'blpop' || $lname === 'brpop' || $lname === 'brpoplpush' )
109  && count( $arguments ) > 1
110  ) {
111  // Get timeout off the end since it is always required and argument length can vary
112  $timeout = end( $arguments );
113  // Only give the additional one second buffer if not requesting an infinite timeout
114  $this->pool->resetTimeout( $this->conn, ( $timeout > 0 ? $timeout + 1 : $timeout ) );
115  }
116 
117  return $this->tryCall( $name, $arguments );
118  }
119 
128  private function tryCall( $method, $arguments ) {
129  $this->conn->clearLastError();
130  try {
131  $res = $this->conn->$method( ...$arguments );
132  $authError = $this->checkAuthentication();
133  if ( $authError === self::AUTH_ERROR_TEMPORARY ) {
134  $res = $this->conn->$method( ...$arguments );
135  }
136  if ( $authError === self::AUTH_ERROR_PERMANENT ) {
137  throw new RedisException( "Failure reauthenticating to Redis." );
138  }
139  } finally {
140  $this->postCallCleanup();
141  }
142 
143  return $res;
144  }
145 
156  public function scan( &$iterator, $pattern = null, $count = null ) {
157  return $this->tryCall( 'scan', [ &$iterator, $pattern, $count ] );
158  }
159 
171  public function sScan( $key, &$iterator, $pattern = null, $count = null ) {
172  return $this->tryCall( 'sScan', [ $key, &$iterator, $pattern, $count ] );
173  }
174 
186  public function hScan( $key, &$iterator, $pattern = null, $count = null ) {
187  return $this->tryCall( 'hScan', [ $key, &$iterator, $pattern, $count ] );
188  }
189 
201  public function zScan( $key, &$iterator, $pattern = null, $count = null ) {
202  return $this->tryCall( 'zScan', [ $key, &$iterator, $pattern, $count ] );
203  }
204 
210  private function checkAuthentication() {
211  if ( preg_match( '/^ERR operation not permitted\b/', $this->conn->getLastError() ) ) {
212  if ( !$this->pool->reauthenticateConnection( $this->server, $this->conn ) ) {
213  return self::AUTH_ERROR_PERMANENT;
214  }
215  $this->conn->clearLastError();
216  $this->logger->info(
217  "Used automatic re-authentication for Redis.",
218  [ 'redis_server' => $this->server ]
219  );
220  return self::AUTH_ERROR_TEMPORARY;
221  }
222  return self::AUTH_NO_ERROR;
223  }
224 
230  private function postCallCleanup() {
231  $this->lastError = $this->conn->getLastError() ?: $this->lastError;
232 
233  // Restore original timeout in the case of blocking calls.
234  $this->pool->resetTimeout( $this->conn );
235  }
236 
244  public function luaEval( $script, array $params, $numKeys ) {
245  $sha1 = sha1( $script ); // 40 char hex
246  $conn = $this->conn; // convenience
247  $server = $this->server; // convenience
248 
249  // Try to run the server-side cached copy of the script
250  $conn->clearLastError();
251  $res = $conn->evalSha( $sha1, $params, $numKeys );
252  // If we got a permission error reply that means that (a) we are not in
253  // multi()/pipeline() and (b) some connection problem likely occurred. If
254  // the password the client gave was just wrong, an exception should have
255  // been thrown back in getConnection() previously.
256  if ( preg_match( '/^ERR operation not permitted\b/', $conn->getLastError() ) ) {
257  $this->pool->reauthenticateConnection( $server, $conn );
258  $conn->clearLastError();
259  $res = $conn->eval( $script, $params, $numKeys );
260  $this->logger->info(
261  "Used automatic re-authentication for Lua script '$sha1'.",
262  [ 'redis_server' => $server ]
263  );
264  }
265  // If the script is not in cache, use eval() to retry and cache it
266  if ( preg_match( '/^NOSCRIPT/', $conn->getLastError() ) ) {
267  $conn->clearLastError();
268  $res = $conn->eval( $script, $params, $numKeys );
269  $this->logger->info(
270  "Used eval() for Lua script '$sha1'.",
271  [ 'redis_server' => $server ]
272  );
273  }
274 
275  if ( $conn->getLastError() ) { // script bug?
276  $this->logger->error(
277  'Lua script error on server "{redis_server}": {lua_error}',
278  [
279  'redis_server' => $server,
280  'lua_error' => $conn->getLastError()
281  ]
282  );
283  }
284 
285  $this->lastError = $conn->getLastError() ?: $this->lastError;
286 
287  return $res;
288  }
289 
294  public function isConnIdentical( Redis $conn ) {
295  return $this->conn === $conn;
296  }
297 
298  function __destruct() {
299  $this->pool->freeConnection( $this->server, $this->conn );
300  }
301 }
__construct(RedisConnectionPool $pool, $server, Redis $conn, LoggerInterface $logger)
const AUTH_ERROR_TEMPORARY
Temporary authentication error; recovered by reauthenticating.
sScan( $key, &$iterator, $pattern=null, $count=null)
Set Scan Handle this explicity due to needing the iterator passed by reference.
string null $lastError
luaEval( $script, array $params, $numKeys)
hScan( $key, &$iterator, $pattern=null, $count=null)
Hash Scan Handle this explicity due to needing the iterator passed by reference.
postCallCleanup()
Post Redis call cleanup.
tryCall( $method, $arguments)
Do the method call in the common try catch handler.
checkAuthentication()
Handle authentication errors and automatically reauthenticate.
const AUTH_ERROR_PERMANENT
Authentication error was permanent and could not be recovered.
scan(&$iterator, $pattern=null, $count=null)
Key Scan Handle this explicity due to needing the iterator passed by reference.
setLogger(LoggerInterface $logger)
const AUTH_NO_ERROR
No authentication errors.
RedisConnectionPool $pool
Helper class to manage Redis connections.
LoggerInterface $logger
Helper class to handle automatically marking connectons as reusable (via RAII pattern) ...
isConnIdentical(Redis $conn)
__call( $name, $arguments)
Magic __call handler for most Redis functions.
zScan( $key, &$iterator, $pattern=null, $count=null)
Sorted Set Scan Handle this explicity due to needing the iterator passed by reference.
string $server