ExternalStoreDB.php

Go to the documentation of this file.

<?php
use MediaWiki\MediaWikiServices;
use Wikimedia\Rdbms\LoadBalancer;
use Wikimedia\Rdbms\IDatabase;
use Wikimedia\Rdbms\DBConnRef;
use Wikimedia\Rdbms\MaintainableDBConnRef;
 
class ExternalStoreDB extends ExternalStoreMedium {
    public function fetchFromURL( $url ) {
        list( $cluster, $id, $itemID ) = $this->parseURL( $url );
        $ret = $this->fetchBlob( $cluster, $id, $itemID );
 
        if ( $itemID !== false && $ret !== false ) {
            return $ret->getItem( $itemID );
        }
 
        return $ret;
    }
 
    public function batchFetchFromURLs( array $urls ) {
        $batched = $inverseUrlMap = [];
        foreach ( $urls as $url ) {
            list( $cluster, $id, $itemID ) = $this->parseURL( $url );
            $batched[$cluster][$id][] = $itemID;
            // false $itemID gets cast to int, but should be ok
            // since we do === from the $itemID in $batched
            $inverseUrlMap[$cluster][$id][$itemID] = $url;
        }
        $ret = [];
        foreach ( $batched as $cluster => $batchByCluster ) {
            $res = $this->batchFetchBlobs( $cluster, $batchByCluster );
            foreach ( $res as $id => $blob ) {
                foreach ( $batchByCluster[$id] as $itemID ) {
                    $url = $inverseUrlMap[$cluster][$id][$itemID];
                    if ( $itemID === false ) {
                        $ret[$url] = $blob;
                    } else {
                        $ret[$url] = $blob->getItem( $itemID );
                    }
                }
            }
        }
 
        return $ret;
    }
 
    public function store( $location, $data ) {
        $dbw = $this->getMaster( $location );
        $dbw->insert( $this->getTable( $dbw ),
            [ 'blob_text' => $data ],
            __METHOD__ );
        $id = $dbw->insertId();
        if ( !$id ) {
            throw new MWException( __METHOD__ . ': no insert ID' );
        }
 
        return "DB://$location/$id";
    }
 
    public function isReadOnly( $location ) {
        return ( $this->getLoadBalancer( $location )->getReadOnlyReason() !== false );
    }
 
    private function getLoadBalancer( $cluster ) {
        $lbFactory = MediaWikiServices::getInstance()->getDBLoadBalancerFactory();
        return $lbFactory->getExternalLB( $cluster );
    }
 
    public function getSlave( $cluster ) {
        global $wgDefaultExternalStore;
 
        $wiki = isset( $this->params['wiki'] ) ? $this->params['wiki'] : false;
        $lb = $this->getLoadBalancer( $cluster );
 
        if ( !in_array( "DB://" . $cluster, (array)$wgDefaultExternalStore ) ) {
            wfDebug( "read only external store\n" );
            $lb->allowLagged( true );
        } else {
            wfDebug( "writable external store\n" );
        }
 
        $db = $lb->getConnectionRef( DB_REPLICA, [], $wiki );
        $db->clearFlag( DBO_TRX ); // sanity
 
        return $db;
    }
 
    public function getMaster( $cluster ) {
        $wiki = isset( $this->params['wiki'] ) ? $this->params['wiki'] : false;
        $lb = $this->getLoadBalancer( $cluster );
 
        $db = $lb->getMaintenanceConnectionRef( DB_MASTER, [], $wiki );
        $db->clearFlag( DBO_TRX ); // sanity
 
        return $db;
    }
 
    public function getTable( $db ) {
        $table = $db->getLBInfo( 'blobs table' );
        if ( is_null( $table ) ) {
            $table = 'blobs';
        }
 
        return $table;
    }
 
    private function fetchBlob( $cluster, $id, $itemID ) {
        static $externalBlobCache = [];
 
        $cacheID = ( $itemID === false ) ? "$cluster/$id" : "$cluster/$id/";
        if ( isset( $externalBlobCache[$cacheID] ) ) {
            wfDebugLog( 'ExternalStoreDB-cache',
                "ExternalStoreDB::fetchBlob cache hit on $cacheID" );
 
            return $externalBlobCache[$cacheID];
        }
 
        wfDebugLog( 'ExternalStoreDB-cache',
            "ExternalStoreDB::fetchBlob cache miss on $cacheID" );
 
        $dbr = $this->getSlave( $cluster );
        $ret = $dbr->selectField( $this->getTable( $dbr ),
            'blob_text', [ 'blob_id' => $id ], __METHOD__ );
        if ( $ret === false ) {
            wfDebugLog( 'ExternalStoreDB',
                "ExternalStoreDB::fetchBlob master fallback on $cacheID" );
            // Try the master
            $dbw = $this->getMaster( $cluster );
            $ret = $dbw->selectField( $this->getTable( $dbw ),
                'blob_text', [ 'blob_id' => $id ], __METHOD__ );
            if ( $ret === false ) {
                wfDebugLog( 'ExternalStoreDB',
                    "ExternalStoreDB::fetchBlob master failed to find $cacheID" );
            }
        }
        if ( $itemID !== false && $ret !== false ) {
            // Unserialise object; caller extracts item
            $ret = unserialize( $ret );
        }
 
        $externalBlobCache = [ $cacheID => $ret ];
 
        return $ret;
    }
 
    private function batchFetchBlobs( $cluster, array $ids ) {
        $dbr = $this->getSlave( $cluster );
        $res = $dbr->select( $this->getTable( $dbr ),
            [ 'blob_id', 'blob_text' ], [ 'blob_id' => array_keys( $ids ) ], __METHOD__ );
        $ret = [];
        if ( $res !== false ) {
            $this->mergeBatchResult( $ret, $ids, $res );
        }
        if ( $ids ) {
            wfDebugLog( __CLASS__, __METHOD__ .
                " master fallback on '$cluster' for: " .
                implode( ',', array_keys( $ids ) ) );
            // Try the master
            $dbw = $this->getMaster( $cluster );
            $res = $dbw->select( $this->getTable( $dbr ),
                [ 'blob_id', 'blob_text' ],
                [ 'blob_id' => array_keys( $ids ) ],
                __METHOD__ );
            if ( $res === false ) {
                wfDebugLog( __CLASS__, __METHOD__ . " master failed on '$cluster'" );
            } else {
                $this->mergeBatchResult( $ret, $ids, $res );
            }
        }
        if ( $ids ) {
            wfDebugLog( __CLASS__, __METHOD__ .
                " master on '$cluster' failed locating items: " .
                implode( ',', array_keys( $ids ) ) );
        }
 
        return $ret;
    }
 
    private function mergeBatchResult( array &$ret, array &$ids, $res ) {
        foreach ( $res as $row ) {
            $id = $row->blob_id;
            $itemIDs = $ids[$id];
            unset( $ids[$id] ); // to track if everything is found
            if ( count( $itemIDs ) === 1 && reset( $itemIDs ) === false ) {
                // single result stored per blob
                $ret[$id] = $row->blob_text;
            } else {
                // multi result stored per blob
                $ret[$id] = unserialize( $row->blob_text );
            }
        }
    }
 
    protected function parseURL( $url ) {
        $path = explode( '/', $url );
 
        return [
            $path[2], // cluster
            $path[3], // id
            isset( $path[4] ) ? $path[4] : false // itemID
        ];
    }
}