includes/installer/NetworkedDatabaseCreator.php

   1 <?php
   2
   3 namespace MediaWiki\Installer;
   4
   5 use MediaWiki\Installer\Task\ITaskContext;
   6 use MediaWiki\Status\Status;
   7 use Wikimedia\Rdbms\DatabaseDomain;
   8 use Wikimedia\Rdbms\IDatabase;
   9 use Wikimedia\Rdbms\ILoadBalancer;
  10
  11 /**
  12  * Common code for MySQL and PostgreSQL, which are similar due to the fact that
  13  * databases can be created by getting a connection to a network service.
  14  *
  15  * SQLite does not inherit this because its connection handles can't operate on
  16  * databases other than the one you originally connected to.
  17  */
  18 abstract class NetworkedDatabaseCreator extends DatabaseCreator {
  19         public function existsLocally( $database ) {
  20                 $connStatus = $this->context->getConnection( ITaskContext::CONN_CREATE_DATABASE );
  21                 if ( !$connStatus->isOK() ) {
  22                         return false;
  23                 }
  24                 return $this->existsInConnection( $connStatus->getDB(), $database );
  25         }
  26
  27         public function existsInLoadBalancer( ILoadBalancer $loadBalancer, $database ) {
  28                 $conn = $loadBalancer->getConnection( DB_PRIMARY, [], DatabaseDomain::newUnspecified()->getId() );
  29                 return $this->existsInConnection( $conn, $database );
  30         }
  31
  32         public function createLocally( $database ): Status {
  33                 $connStatus = $this->context->getConnection( ITaskContext::CONN_CREATE_DATABASE );
  34                 if ( !$connStatus->isOK() ) {
  35                         return $connStatus;
  36                 }
  37                 $conn = $connStatus->getDB();
  38                 return $this->createInConnection( $conn, $database );
  39         }
  40
  41         public function createInLoadBalancer( ILoadBalancer $loadBalancer, $database ): Status {
  42                 $conn = $loadBalancer->getConnection( DB_PRIMARY, [], DatabaseDomain::newUnspecified()->getId() );
  43                 return $this->createInConnection( $conn, $database );
  44         }
  45
  46         /**
  47          * Determine whether a database exists on a connection
  48          *
  49          * @param IDatabase $conn
  50          * @param string $database
  51          * @return bool
  52          */
  53         abstract protected function existsInConnection( IDatabase $conn, $database );
  54
  55         /**
  56          * Create a database on a connection
  57          *
  58          * @param IDatabase $conn
  59          * @param string $database
  60          * @return Status
  61          */
  62         abstract protected function createInConnection( IDatabase $conn, $database ): Status;
  63 }