SessionConsistentConnectionManager.php

Go to the documentation of this file.

<?php
namespace Wikimedia\Rdbms;
 
class SessionConsistentConnectionManager extends ConnectionManager {
 
    private $forceWriteConnection = false;
 
    public function prepareForUpdates() {
        $this->forceWriteConnection = true;
    }
 
    public function getReadConnection( ?array $groups = null, int $flags = 0 ) {
        if ( $this->forceWriteConnection ) {
            return parent::getWriteConnection( $flags );
        }
 
        return parent::getReadConnection( $groups, $flags );
    }
 
    public function getWriteConnection( int $flags = 0 ) {
        $this->prepareForUpdates();
        return parent::getWriteConnection( $flags );
    }
 
    public function getReadConnectionRef( array $groups = null ) {
        if ( $this->forceWriteConnection ) {
            return parent::getWriteConnectionRef();
        }
 
        return parent::getReadConnectionRef( $groups );
    }
 
    public function getWriteConnectionRef() {
        $this->prepareForUpdates();
        return parent::getWriteConnectionRef();
    }
 
}