BagOStuff.php

Go to the documentation of this file.

<?php
use Psr\Log\LoggerAwareInterface;
use Psr\Log\LoggerInterface;
use Psr\Log\NullLogger;
use Wikimedia\ScopedCallback;
use Wikimedia\WaitConditionLoop;
 
abstract class BagOStuff implements IExpiringStore, LoggerAwareInterface {
    protected $locks = [];
    protected $lastError = self::ERR_NONE;
    protected $keyspace = 'local';
    protected $logger;
    protected $asyncHandler;
    protected $syncTimeout;
 
    private $debugMode = false;
    private $duplicateKeyLookups = [];
    private $reportDupes = false;
    private $dupeTrackScheduled = false;
 
    protected $busyCallbacks = [];
 
    private $wallClockOverride;
 
    protected $attrMap = [];
 
    const ERR_NONE = 0; // no error
    const ERR_NO_RESPONSE = 1; // no response
    const ERR_UNREACHABLE = 2; // can't connect
    const ERR_UNEXPECTED = 3; // response gave some error
 
    const READ_LATEST = 1; // use latest data for replicated stores
    const READ_VERIFIED = 2; // promise that caller can tell when keys are stale
    const WRITE_SYNC = 1; // synchronously write to all locations for replicated stores
    const WRITE_CACHE_ONLY = 2; // Only change state of the in-memory cache
 
    public function __construct( array $params = [] ) {
        if ( isset( $params['logger'] ) ) {
            $this->setLogger( $params['logger'] );
        } else {
            $this->setLogger( new NullLogger() );
        }
 
        if ( isset( $params['keyspace'] ) ) {
            $this->keyspace = $params['keyspace'];
        }
 
        $this->asyncHandler = $params['asyncHandler'] ?? null;
 
        if ( !empty( $params['reportDupes'] ) && is_callable( $this->asyncHandler ) ) {
            $this->reportDupes = true;
        }
 
        $this->syncTimeout = $params['syncTimeout'] ?? 3;
    }
 
    public function setLogger( LoggerInterface $logger ) {
        $this->logger = $logger;
    }
 
    public function setDebug( $bool ) {
        $this->debugMode = $bool;
    }
 
    final public function getWithSetCallback( $key, $ttl, $callback, $flags = 0 ) {
        $value = $this->get( $key, $flags );
 
        if ( $value === false ) {
            if ( !is_callable( $callback ) ) {
                throw new InvalidArgumentException( "Invalid cache miss callback provided." );
            }
            $value = call_user_func( $callback );
            if ( $value !== false ) {
                $this->set( $key, $value, $ttl );
            }
        }
 
        return $value;
    }
 
    public function get( $key, $flags = 0, $oldFlags = null ) {
        // B/C for ( $key, &$casToken = null, $flags = 0 )
        $flags = is_int( $oldFlags ) ? $oldFlags : $flags;
 
        $this->trackDuplicateKeys( $key );
 
        return $this->doGet( $key, $flags );
    }
 
    private function trackDuplicateKeys( $key ) {
        if ( !$this->reportDupes ) {
            return;
        }
 
        if ( !isset( $this->duplicateKeyLookups[$key] ) ) {
            // Track that we have seen this key. This N-1 counting style allows
            // easy filtering with array_filter() later.
            $this->duplicateKeyLookups[$key] = 0;
        } else {
            $this->duplicateKeyLookups[$key] += 1;
 
            if ( $this->dupeTrackScheduled === false ) {
                $this->dupeTrackScheduled = true;
                // Schedule a callback that logs keys processed more than once by get().
                call_user_func( $this->asyncHandler, function () {
                    $dups = array_filter( $this->duplicateKeyLookups );
                    foreach ( $dups as $key => $count ) {
                        $this->logger->warning(
                            'Duplicate get(): "{key}" fetched {count} times',
                            // Count is N-1 of the actual lookup count
                            [ 'key' => $key, 'count' => $count + 1, ]
                        );
                    }
                } );
            }
        }
    }
 
    abstract protected function doGet( $key, $flags = 0 );
 
    protected function getWithToken( $key, &$casToken, $flags = 0 ) {
        throw new Exception( __METHOD__ . ' not implemented.' );
    }
 
    abstract public function set( $key, $value, $exptime = 0, $flags = 0 );
 
    abstract public function delete( $key );
 
    public function merge( $key, callable $callback, $exptime = 0, $attempts = 10, $flags = 0 ) {
        return $this->mergeViaLock( $key, $callback, $exptime, $attempts, $flags );
    }
 
    protected function mergeViaCas( $key, $callback, $exptime = 0, $attempts = 10 ) {
        do {
            $this->clearLastError();
            $reportDupes = $this->reportDupes;
            $this->reportDupes = false;
            $casToken = null; // passed by reference
            $currentValue = $this->getWithToken( $key, $casToken, self::READ_LATEST );
            $this->reportDupes = $reportDupes;
 
            if ( $this->getLastError() ) {
                $this->logger->warning(
                    __METHOD__ . ' failed due to I/O error on get() for {key}.',
                    [ 'key' => $key ]
                );
 
                return false; // don't spam retries (retry only on races)
            }
 
            // Derive the new value from the old value
            $value = call_user_func( $callback, $this, $key, $currentValue, $exptime );
 
            $this->clearLastError();
            if ( $value === false ) {
                $success = true; // do nothing
            } elseif ( $currentValue === false ) {
                // Try to create the key, failing if it gets created in the meantime
                $success = $this->add( $key, $value, $exptime );
            } else {
                // Try to update the key, failing if it gets changed in the meantime
                $success = $this->cas( $casToken, $key, $value, $exptime );
            }
            if ( $this->getLastError() ) {
                $this->logger->warning(
                    __METHOD__ . ' failed due to I/O error for {key}.',
                    [ 'key' => $key ]
                );
 
                return false; // IO error; don't spam retries
            }
        } while ( !$success && --$attempts );
 
        return $success;
    }
 
    protected function cas( $casToken, $key, $value, $exptime = 0 ) {
        if ( !$this->lock( $key, 0 ) ) {
            return false; // non-blocking
        }
 
        $curCasToken = null; // passed by reference
        $this->getWithToken( $key, $curCasToken, self::READ_LATEST );
        if ( $casToken === $curCasToken ) {
            $success = $this->set( $key, $value, $exptime );
        } else {
            $this->logger->info(
                __METHOD__ . ' failed due to race condition for {key}.',
                [ 'key' => $key ]
            );
 
            $success = false; // mismatched or failed
        }
 
        $this->unlock( $key );
 
        return $success;
    }
 
    protected function mergeViaLock( $key, $callback, $exptime = 0, $attempts = 10, $flags = 0 ) {
        if ( $attempts <= 1 ) {
            $timeout = 0; // clearly intended to be "non-blocking"
        } else {
            $timeout = 3;
        }
 
        if ( !$this->lock( $key, $timeout ) ) {
            return false;
        }
 
        $this->clearLastError();
        $reportDupes = $this->reportDupes;
        $this->reportDupes = false;
        $currentValue = $this->get( $key, self::READ_LATEST );
        $this->reportDupes = $reportDupes;
 
        if ( $this->getLastError() ) {
            $this->logger->warning(
                __METHOD__ . ' failed due to I/O error on get() for {key}.',
                [ 'key' => $key ]
            );
 
            $success = false;
        } else {
            // Derive the new value from the old value
            $value = call_user_func( $callback, $this, $key, $currentValue, $exptime );
            if ( $value === false ) {
                $success = true; // do nothing
            } else {
                $success = $this->set( $key, $value, $exptime, $flags ); // set the new value
            }
        }
 
        if ( !$this->unlock( $key ) ) {
            // this should never happen
            trigger_error( "Could not release lock for key '$key'." );
        }
 
        return $success;
    }
 
    public function changeTTL( $key, $expiry = 0 ) {
        $value = $this->get( $key );
 
        return ( $value === false ) ? false : $this->set( $key, $value, $expiry );
    }
 
    public function lock( $key, $timeout = 6, $expiry = 6, $rclass = '' ) {
        // Avoid deadlocks and allow lock reentry if specified
        if ( isset( $this->locks[$key] ) ) {
            if ( $rclass != '' && $this->locks[$key]['class'] === $rclass ) {
                ++$this->locks[$key]['depth'];
                return true;
            } else {
                return false;
            }
        }
 
        $fname = __METHOD__;
        $expiry = min( $expiry ?: INF, self::TTL_DAY );
        $loop = new WaitConditionLoop(
            function () use ( $key, $timeout, $expiry, $fname ) {
                $this->clearLastError();
                if ( $this->add( "{$key}:lock", 1, $expiry ) ) {
                    return true; // locked!
                } elseif ( $this->getLastError() ) {
                    $this->logger->warning(
                        $fname . ' failed due to I/O error for {key}.',
                        [ 'key' => $key ]
                    );
 
                    return WaitConditionLoop::CONDITION_ABORTED; // network partition?
                }
 
                return WaitConditionLoop::CONDITION_CONTINUE;
            },
            $timeout
        );
 
        $code = $loop->invoke();
        $locked = ( $code === $loop::CONDITION_REACHED );
        if ( $locked ) {
            $this->locks[$key] = [ 'class' => $rclass, 'depth' => 1 ];
        } elseif ( $code === $loop::CONDITION_TIMED_OUT ) {
            $this->logger->warning(
                "$fname failed due to timeout for {key}.",
                [ 'key' => $key, 'timeout' => $timeout ]
            );
        }
 
        return $locked;
    }
 
    public function unlock( $key ) {
        if ( isset( $this->locks[$key] ) && --$this->locks[$key]['depth'] <= 0 ) {
            unset( $this->locks[$key] );
 
            $ok = $this->delete( "{$key}:lock" );
            if ( !$ok ) {
                $this->logger->warning(
                    __METHOD__ . ' failed to release lock for {key}.',
                    [ 'key' => $key ]
                );
            }
 
            return $ok;
        }
 
        return true;
    }
 
    final public function getScopedLock( $key, $timeout = 6, $expiry = 30, $rclass = '' ) {
        $expiry = min( $expiry ?: INF, self::TTL_DAY );
 
        if ( !$this->lock( $key, $timeout, $expiry, $rclass ) ) {
            return null;
        }
 
        $lSince = $this->getCurrentTime(); // lock timestamp
 
        return new ScopedCallback( function () use ( $key, $lSince, $expiry ) {
            $latency = 0.050; // latency skew (err towards keeping lock present)
            $age = ( $this->getCurrentTime() - $lSince + $latency );
            if ( ( $age + $latency ) >= $expiry ) {
                $this->logger->warning(
                    "Lock for {key} held too long ({age} sec).",
                    [ 'key' => $key, 'age' => $age ]
                );
                return; // expired; it's not "safe" to delete the key
            }
            $this->unlock( $key );
        } );
    }
 
    public function deleteObjectsExpiringBefore( $date, $progressCallback = false ) {
        // stub
        return false;
    }
 
    public function getMulti( array $keys, $flags = 0 ) {
        $res = [];
        foreach ( $keys as $key ) {
            $val = $this->get( $key );
            if ( $val !== false ) {
                $res[$key] = $val;
            }
        }
        return $res;
    }
 
    public function setMulti( array $data, $exptime = 0 ) {
        $res = true;
        foreach ( $data as $key => $value ) {
            if ( !$this->set( $key, $value, $exptime ) ) {
                $res = false;
            }
        }
        return $res;
    }
 
    public function add( $key, $value, $exptime = 0 ) {
        // @note: avoid lock() here since that method uses *this* method by default
        if ( $this->get( $key ) === false ) {
            return $this->set( $key, $value, $exptime );
        }
        return false; // key already set
    }
 
    public function incr( $key, $value = 1 ) {
        if ( !$this->lock( $key, 1 ) ) {
            return false;
        }
        $n = $this->get( $key );
        if ( $this->isInteger( $n ) ) { // key exists?
            $n += intval( $value );
            $this->set( $key, max( 0, $n ) ); // exptime?
        } else {
            $n = false;
        }
        $this->unlock( $key );
 
        return $n;
    }
 
    public function decr( $key, $value = 1 ) {
        return $this->incr( $key, - $value );
    }
 
    public function incrWithInit( $key, $ttl, $value = 1, $init = 1 ) {
        $this->clearLastError();
        $newValue = $this->incr( $key, $value );
        if ( $newValue === false && !$this->getLastError() ) {
            // No key set; initialize
            $newValue = $this->add( $key, (int)$init, $ttl ) ? $init : false;
            if ( $newValue === false && !$this->getLastError() ) {
                // Raced out initializing; increment
                $newValue = $this->incr( $key, $value );
            }
        }
 
        return $newValue;
    }
 
    public function getLastError() {
        return $this->lastError;
    }
 
    public function clearLastError() {
        $this->lastError = self::ERR_NONE;
    }
 
    protected function setLastError( $err ) {
        $this->lastError = $err;
    }
 
    public function addBusyCallback( callable $workCallback ) {
        $this->busyCallbacks[] = $workCallback;
    }
 
    public function modifySimpleRelayEvent( array $event ) {
        return $event;
    }
 
    protected function debug( $text ) {
        if ( $this->debugMode ) {
            $this->logger->debug( "{class} debug: $text", [
                'class' => static::class,
            ] );
        }
    }
 
    protected function convertExpiry( $exptime ) {
        if ( $exptime != 0 && $exptime < ( 10 * self::TTL_YEAR ) ) {
            return (int)$this->getCurrentTime() + $exptime;
        } else {
            return $exptime;
        }
    }
 
    protected function convertToRelative( $exptime ) {
        if ( $exptime >= ( 10 * self::TTL_YEAR ) ) {
            $exptime -= (int)$this->getCurrentTime();
            if ( $exptime <= 0 ) {
                $exptime = 1;
            }
            return $exptime;
        } else {
            return $exptime;
        }
    }
 
    protected function isInteger( $value ) {
        return ( is_int( $value ) || ctype_digit( $value ) );
    }
 
    public function makeKeyInternal( $keyspace, $args ) {
        $key = $keyspace;
        foreach ( $args as $arg ) {
            $arg = str_replace( ':', '%3A', $arg );
            $key = $key . ':' . $arg;
        }
        return strtr( $key, ' ', '_' );
    }
 
    public function makeGlobalKey( $class, $component = null ) {
        return $this->makeKeyInternal( 'global', func_get_args() );
    }
 
    public function makeKey( $class, $component = null ) {
        return $this->makeKeyInternal( $this->keyspace, func_get_args() );
    }
 
    public function getQoS( $flag ) {
        return $this->attrMap[$flag] ?? self::QOS_UNKNOWN;
    }
 
    protected function mergeFlagMaps( array $bags ) {
        $map = [];
        foreach ( $bags as $bag ) {
            foreach ( $bag->attrMap as $attr => $rank ) {
                if ( isset( $map[$attr] ) ) {
                    $map[$attr] = min( $map[$attr], $rank );
                } else {
                    $map[$attr] = $rank;
                }
            }
        }
 
        return $map;
    }
 
    protected function getCurrentTime() {
        return $this->wallClockOverride ?: microtime( true );
    }
 
    public function setMockTime( &$time ) {
        $this->wallClockOverride =& $time;
    }
}