wiki.techinc.nl/includes/libs/objectcache/WinCacheBagOStuff.php
Aaron Schulz bc79ce7fb7 objectcache: deprecate BagOStuff incr() and decr() methods
All callers in Wikimedia gerrit repos have since been migrated to
incrWithInit(), which is usually what is desired and is easier to
implement in various backends. Newer memcached versions of the
memcached protocol even support auto-initializing increments
(including an initial TTL) via the 'ma' command. The incr()/decr()
methods currently bloat the interface and subclasses (often with
slow and buggy implementations). These methods are also hard to
implement in a multi-DC store (either extremely slow or racey)
even though callers might assume they handle high concurrency.

Also, force each MediumSpecificBagOStuff subclass to implement
incrWithInit() instead of having some rely on the base method

In mcc.php, add incrWithInit() and simplify parameters, making
it possible to test non-memcached backends.

Change-Id: I53c9c2c839a1e71d5c104913fea0680c30d11108
2022-02-18 11:15:27 +11:00

189 lines
5.6 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 );
if ( PHP_SAPI === 'cli' ) {
$this->attrMap[self::ATTR_DURABILITY] = ini_get( 'wincache.enablecli' )
? self::QOS_DURABILITY_SCRIPT
: self::QOS_DURABILITY_NONE;
} else {
$this->attrMap[self::ATTR_DURABILITY] = self::QOS_DURABILITY_SERVICE;
}
}
protected function doGet( $key, $flags = 0, &$casToken = null ) {
$getToken = ( $casToken === self::PASS_BY_REF );
$casToken = null;
$blob = wincache_ucache_get( $key );
if ( !is_string( $blob ) && !is_int( $blob ) ) {
return false;
}
$value = $this->unserialize( $blob );
if ( $getToken && $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 = self::PASS_BY_REF; // 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 ) {
$ttl = $this->getExpirationAsTTL( $exptime );
$result = wincache_ucache_set( $key, $this->getSerialized( $value, $key ), $ttl );
// 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
}
$ttl = $this->getExpirationAsTTL( $exptime );
$result = wincache_ucache_add( $key, $this->getSerialized( $value, $key ), $ttl );
// 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, $components ) {
// 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( $components );
$components = array_map(
static function ( $component ) use ( &$charsLeft ) {
// 33 = 32 characters for the MD5 + 1 for the '#' prefix.
if ( $charsLeft > 33 && strlen( $component ) > $charsLeft ) {
$component = '#' . md5( $component );
}
$charsLeft -= strlen( $component );
return $component;
},
$components
);
if ( $charsLeft < 0 ) {
return $keyspace . ':BagOStuff-long-key:##' . md5( implode( ':', $components ) );
}
return $keyspace . ':' . implode( ':', $components );
}
public function doIncrWithInit( $key, $exptime, $step, $init, $flags ) {
if ( !wincache_lock( $key ) ) { // optimize with FIFO lock
return false;
}
$curValue = $this->doGet( $key );
if ( $curValue === false ) {
$newValue = $this->doSet( $key, $init, $exptime ) ? $init : false;
} elseif ( $this->isInteger( $curValue ) ) {
$sum = max( $curValue + $step, 0 );
$oldTTL = wincache_ucache_info( false, $key )["ucache_entries"][1]["ttl_seconds"];
$newValue = $this->doSet( $key, $sum, $oldTTL ) ? $sum : false;
} else {
$newValue = false;
}
wincache_unlock( $key );
return $newValue;
}
public function incr( $key, $value = 1, $flags = 0 ) {
return $this->doIncr( $key, $value, $flags );
}
public function decr( $key, $value = 1, $flags = 0 ) {
return $this->doIncr( $key, -$value, $flags );
}
private function doIncr( $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->doSet( $key, $n, $oldTTL );
} else {
$n = false;
}
wincache_unlock( $key );
return $n;
}
}