This can be used by WANObjectCache or statsd metrics to estimate the size of objects and network commands without having to do an extra serialization round. Change-Id: I7b1b787bba39a21e2e04262e77ee7ae24e88f8ca
153 lines
4.4 KiB
PHP
153 lines
4.4 KiB
PHP
<?php
|
|
/**
|
|
* Object caching using WinCache.
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along
|
|
* with this program; if not, write to the Free Software Foundation, Inc.,
|
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
* http://www.gnu.org/copyleft/gpl.html
|
|
*
|
|
* @file
|
|
* @ingroup Cache
|
|
*/
|
|
|
|
/**
|
|
* Wrapper for WinCache object caching functions; identical interface
|
|
* to the APC wrapper
|
|
*
|
|
* @ingroup Cache
|
|
*/
|
|
class WinCacheBagOStuff extends MediumSpecificBagOStuff {
|
|
public function __construct( array $params = [] ) {
|
|
$params['segmentationSize'] = $params['segmentationSize'] ?? INF;
|
|
parent::__construct( $params );
|
|
}
|
|
|
|
protected function doGet( $key, $flags = 0, &$casToken = null ) {
|
|
$casToken = null;
|
|
|
|
$blob = wincache_ucache_get( $key );
|
|
if ( !is_string( $blob ) && !is_int( $blob ) ) {
|
|
return false;
|
|
}
|
|
|
|
$value = $this->unserialize( $blob );
|
|
if ( $value !== false ) {
|
|
$casToken = (string)$blob; // don't bother hashing this
|
|
}
|
|
|
|
return $value;
|
|
}
|
|
|
|
protected function doCas( $casToken, $key, $value, $exptime = 0, $flags = 0 ) {
|
|
if ( !wincache_lock( $key ) ) { // optimize with FIFO lock
|
|
return false;
|
|
}
|
|
|
|
$curCasToken = null; // passed by reference
|
|
$this->doGet( $key, self::READ_LATEST, $curCasToken );
|
|
if ( $casToken === $curCasToken ) {
|
|
$success = $this->set( $key, $value, $exptime, $flags );
|
|
} else {
|
|
$this->logger->info(
|
|
__METHOD__ . ' failed due to race condition for {key}.',
|
|
[ 'key' => $key ]
|
|
);
|
|
|
|
$success = false; // mismatched or failed
|
|
}
|
|
|
|
wincache_unlock( $key );
|
|
|
|
return $success;
|
|
}
|
|
|
|
protected function doSet( $key, $value, $exptime = 0, $flags = 0 ) {
|
|
$result = wincache_ucache_set( $key, $this->getSerialized( $value, $key ), $exptime );
|
|
|
|
// false positive, wincache_ucache_set returns an empty array
|
|
// in some circumstances.
|
|
// @phan-suppress-next-line PhanTypeComparisonToArray
|
|
return ( $result === [] || $result === true );
|
|
}
|
|
|
|
protected function doAdd( $key, $value, $exptime = 0, $flags = 0 ) {
|
|
if ( wincache_ucache_exists( $key ) ) {
|
|
return false; // avoid warnings
|
|
}
|
|
|
|
$result = wincache_ucache_add( $key, $this->getSerialized( $value, $key ), $exptime );
|
|
|
|
// false positive, wincache_ucache_add returns an empty array
|
|
// in some circumstances.
|
|
// @phan-suppress-next-line PhanTypeComparisonToArray
|
|
return ( $result === [] || $result === true );
|
|
}
|
|
|
|
protected function doDelete( $key, $flags = 0 ) {
|
|
wincache_ucache_delete( $key );
|
|
|
|
return true;
|
|
}
|
|
|
|
public function makeKeyInternal( $keyspace, $args ) {
|
|
// WinCache keys have a maximum length of 150 characters. From that,
|
|
// subtract the number of characters we need for the keyspace and for
|
|
// the separator character needed for each argument. To handle some
|
|
// custom prefixes used by thing like WANObjectCache, limit to 125.
|
|
// NOTE: Same as in memcached, except the max key length there is 255.
|
|
$charsLeft = 125 - strlen( $keyspace ) - count( $args );
|
|
|
|
$args = array_map(
|
|
function ( $arg ) use ( &$charsLeft ) {
|
|
// 33 = 32 characters for the MD5 + 1 for the '#' prefix.
|
|
if ( $charsLeft > 33 && strlen( $arg ) > $charsLeft ) {
|
|
$arg = '#' . md5( $arg );
|
|
}
|
|
|
|
$charsLeft -= strlen( $arg );
|
|
return $arg;
|
|
},
|
|
$args
|
|
);
|
|
|
|
if ( $charsLeft < 0 ) {
|
|
return $keyspace . ':BagOStuff-long-key:##' . md5( implode( ':', $args ) );
|
|
}
|
|
|
|
return $keyspace . ':' . implode( ':', $args );
|
|
}
|
|
|
|
public function incr( $key, $value = 1, $flags = 0 ) {
|
|
if ( !wincache_lock( $key ) ) { // optimize with FIFO lock
|
|
return false;
|
|
}
|
|
|
|
$n = $this->doGet( $key );
|
|
if ( $this->isInteger( $n ) ) {
|
|
$n = max( $n + (int)$value, 0 );
|
|
$oldTTL = wincache_ucache_info( false, $key )["ucache_entries"][1]["ttl_seconds"];
|
|
$this->set( $key, $n, $oldTTL );
|
|
} else {
|
|
$n = false;
|
|
}
|
|
|
|
wincache_unlock( $key );
|
|
|
|
return $n;
|
|
}
|
|
|
|
public function decr( $key, $value = 1, $flags = 0 ) {
|
|
return $this->incr( $key, -$value, $flags );
|
|
}
|
|
}
|