includes/db/DBConnRef.php

   1 <?php
   2 /**
   3  * Helper class to handle automatically marking connections as reusable (via RAII pattern)
   4  * as well handling deferring the actual network connection until the handle is used
   5  *
   6  * @note: proxy methods are defined explicity to avoid interface errors
   7  * @ingroup Database
   8  * @since 1.22
   9  */
  10 class DBConnRef implements IDatabase {
  11         /** @var LoadBalancer */
  12         private $lb;
  13
  14         /** @var DatabaseBase|null */
  15         private $conn;
  16
  17         /** @var array|null */
  18         private $params;
  19
  20         /**
  21          * @param LoadBalancer $lb
  22          * @param DatabaseBase|array $conn Connection or (server index, group, wiki ID) array
  23          */
  24         public function __construct( LoadBalancer $lb, $conn ) {
  25                 $this->lb = $lb;
  26                 if ( $conn instanceof DatabaseBase ) {
  27                         $this->conn = $conn;
  28                 } else {
  29                         $this->params = $conn;
  30                 }
  31         }
  32
  33         function __call( $name, array $arguments ) {
  34                 if ( $this->conn === null ) {
  35                         list( $db, $groups, $wiki ) = $this->params;
  36                         $this->conn = $this->lb->getConnection( $db, $groups, $wiki );
  37                 }
  38
  39                 return call_user_func_array( array( $this->conn, $name ), $arguments );
  40         }
  41
  42         public function getServerInfo() {
  43                 return $this->__call( __FUNCTION__, func_get_args() );
  44         }
  45
  46         public function bufferResults( $buffer = null ) {
  47                 return $this->__call( __FUNCTION__, func_get_args() );
  48         }
  49
  50         public function trxLevel() {
  51                 return $this->__call( __FUNCTION__, func_get_args() );
  52         }
  53
  54         public function trxTimestamp() {
  55                 return $this->__call( __FUNCTION__, func_get_args() );
  56         }
  57
  58         public function tablePrefix( $prefix = null ) {
  59                 return $this->__call( __FUNCTION__, func_get_args() );
  60         }
  61
  62         public function dbSchema( $schema = null ) {
  63                 return $this->__call( __FUNCTION__, func_get_args() );
  64         }
  65
  66         public function getLBInfo( $name = null ) {
  67                 return $this->__call( __FUNCTION__, func_get_args() );
  68         }
  69
  70         public function setLBInfo( $name, $value = null ) {
  71                 return $this->__call( __FUNCTION__, func_get_args() );
  72         }
  73
  74         public function implicitGroupby() {
  75                 return $this->__call( __FUNCTION__, func_get_args() );
  76         }
  77
  78         public function implicitOrderby() {
  79                 return $this->__call( __FUNCTION__, func_get_args() );
  80         }
  81
  82         public function lastQuery() {
  83                 return $this->__call( __FUNCTION__, func_get_args() );
  84         }
  85
  86         public function doneWrites() {
  87                 return $this->__call( __FUNCTION__, func_get_args() );
  88         }
  89
  90         public function lastDoneWrites() {
  91                 return $this->__call( __FUNCTION__, func_get_args() );
  92         }
  93
  94         public function writesPending() {
  95                 return $this->__call( __FUNCTION__, func_get_args() );
  96         }
  97
  98         public function writesOrCallbacksPending() {
  99                 return $this->__call( __FUNCTION__, func_get_args() );
 100         }
 101
 102         public function pendingWriteQueryDuration() {
 103                 return $this->__call( __FUNCTION__, func_get_args() );
 104         }
 105
 106         public function isOpen() {
 107                 return $this->__call( __FUNCTION__, func_get_args() );
 108         }
 109
 110         public function setFlag( $flag ) {
 111                 return $this->__call( __FUNCTION__, func_get_args() );
 112         }
 113
 114         public function clearFlag( $flag ) {
 115                 return $this->__call( __FUNCTION__, func_get_args() );
 116         }
 117
 118         public function getFlag( $flag ) {
 119                 return $this->__call( __FUNCTION__, func_get_args() );
 120         }
 121
 122         public function getProperty( $name ) {
 123                 return $this->__call( __FUNCTION__, func_get_args() );
 124         }
 125
 126         public function getWikiID() {
 127                 return $this->__call( __FUNCTION__, func_get_args() );
 128         }
 129
 130         public function getType() {
 131                 return $this->__call( __FUNCTION__, func_get_args() );
 132         }
 133
 134         public function open( $server, $user, $password, $dbName ) {
 135                 return $this->__call( __FUNCTION__, func_get_args() );
 136         }
 137
 138         public function fetchObject( $res ) {
 139                 return $this->__call( __FUNCTION__, func_get_args() );
 140         }
 141
 142         public function fetchRow( $res ) {
 143                 return $this->__call( __FUNCTION__, func_get_args() );
 144         }
 145
 146         public function numRows( $res ) {
 147                 return $this->__call( __FUNCTION__, func_get_args() );
 148         }
 149
 150         public function numFields( $res ) {
 151                 return $this->__call( __FUNCTION__, func_get_args() );
 152         }
 153
 154         public function fieldName( $res, $n ) {
 155                 return $this->__call( __FUNCTION__, func_get_args() );
 156         }
 157
 158         public function insertId() {
 159                 return $this->__call( __FUNCTION__, func_get_args() );
 160         }
 161
 162         public function dataSeek( $res, $row ) {
 163                 return $this->__call( __FUNCTION__, func_get_args() );
 164         }
 165
 166         public function lastErrno() {
 167                 return $this->__call( __FUNCTION__, func_get_args() );
 168         }
 169
 170         public function lastError() {
 171                 return $this->__call( __FUNCTION__, func_get_args() );
 172         }
 173
 174         public function fieldInfo( $table, $field ) {
 175                 return $this->__call( __FUNCTION__, func_get_args() );
 176         }
 177
 178         public function affectedRows() {
 179                 return $this->__call( __FUNCTION__, func_get_args() );
 180         }
 181
 182         public function getSoftwareLink() {
 183                 return $this->__call( __FUNCTION__, func_get_args() );
 184         }
 185
 186         public function getServerVersion() {
 187                 return $this->__call( __FUNCTION__, func_get_args() );
 188         }
 189
 190         public function close() {
 191                 return $this->__call( __FUNCTION__, func_get_args() );
 192         }
 193
 194         public function reportConnectionError( $error = 'Unknown error' ) {
 195                 return $this->__call( __FUNCTION__, func_get_args() );
 196         }
 197
 198         public function query( $sql, $fname = __METHOD__, $tempIgnore = false ) {
 199                 return $this->__call( __FUNCTION__, func_get_args() );
 200         }
 201
 202         public function reportQueryError( $error, $errno, $sql, $fname, $tempIgnore = false ) {
 203                 return $this->__call( __FUNCTION__, func_get_args() );
 204         }
 205
 206         public function freeResult( $res ) {
 207                 return $this->__call( __FUNCTION__, func_get_args() );
 208         }
 209
 210         public function selectField(
 211                 $table, $var, $cond = '', $fname = __METHOD__, $options = array()
 212         ) {
 213                 return $this->__call( __FUNCTION__, func_get_args() );
 214         }
 215
 216         public function selectFieldValues(
 217                 $table, $var, $cond = '', $fname = __METHOD__, $options = array()
 218         ) {
 219                 return $this->__call( __FUNCTION__, func_get_args() );
 220         }
 221
 222         public function select(
 223                 $table, $vars, $conds = '', $fname = __METHOD__,
 224                 $options = array(), $join_conds = array()
 225         ) {
 226                 return $this->__call( __FUNCTION__, func_get_args() );
 227         }
 228
 229         public function selectSQLText(
 230                 $table, $vars, $conds = '', $fname = __METHOD__,
 231                 $options = array(), $join_conds = array()
 232         ) {
 233                 return $this->__call( __FUNCTION__, func_get_args() );
 234         }
 235
 236         public function selectRow(
 237                 $table, $vars, $conds, $fname = __METHOD__,
 238                 $options = array(), $join_conds = array()
 239         ) {
 240                 return $this->__call( __FUNCTION__, func_get_args() );
 241         }
 242
 243         public function estimateRowCount(
 244                 $table, $vars = '*', $conds = '', $fname = __METHOD__, $options = array()
 245         ) {
 246                 return $this->__call( __FUNCTION__, func_get_args() );
 247         }
 248
 249         public function selectRowCount(
 250                 $tables, $vars = '*', $conds = '', $fname = __METHOD__, $options = array(), $join_conds = array()
 251         ) {
 252                 return $this->__call( __FUNCTION__, func_get_args() );
 253         }
 254
 255         public function fieldExists( $table, $field, $fname = __METHOD__ ) {
 256                 return $this->__call( __FUNCTION__, func_get_args() );
 257         }
 258
 259         public function indexExists( $table, $index, $fname = __METHOD__ ) {
 260                 return $this->__call( __FUNCTION__, func_get_args() );
 261         }
 262
 263         public function tableExists( $table, $fname = __METHOD__ ) {
 264                 return $this->__call( __FUNCTION__, func_get_args() );
 265         }
 266
 267         public function indexUnique( $table, $index ) {
 268                 return $this->__call( __FUNCTION__, func_get_args() );
 269         }
 270
 271         public function insert( $table, $a, $fname = __METHOD__, $options = array() ) {
 272                 return $this->__call( __FUNCTION__, func_get_args() );
 273         }
 274
 275         public function update( $table, $values, $conds, $fname = __METHOD__, $options = array() ) {
 276                 return $this->__call( __FUNCTION__, func_get_args() );
 277         }
 278
 279         public function makeList( $a, $mode = LIST_COMMA ) {
 280                 return $this->__call( __FUNCTION__, func_get_args() );
 281         }
 282
 283         public function makeWhereFrom2d( $data, $baseKey, $subKey ) {
 284                 return $this->__call( __FUNCTION__, func_get_args() );
 285         }
 286
 287         public function bitNot( $field ) {
 288                 return $this->__call( __FUNCTION__, func_get_args() );
 289         }
 290
 291         public function bitAnd( $fieldLeft, $fieldRight ) {
 292                 return $this->__call( __FUNCTION__, func_get_args() );
 293         }
 294
 295         public function bitOr( $fieldLeft, $fieldRight ) {
 296                 return $this->__call( __FUNCTION__, func_get_args() );
 297         }
 298
 299         public function buildConcat( $stringList ) {
 300                 return $this->__call( __FUNCTION__, func_get_args() );
 301         }
 302
 303         public function buildGroupConcatField(
 304                 $delim, $table, $field, $conds = '', $join_conds = array()
 305         ) {
 306                 return $this->__call( __FUNCTION__, func_get_args() );
 307         }
 308
 309         public function selectDB( $db ) {
 310                 return $this->__call( __FUNCTION__, func_get_args() );
 311         }
 312
 313         public function getDBname() {
 314                 return $this->__call( __FUNCTION__, func_get_args() );
 315         }
 316
 317         public function getServer() {
 318                 return $this->__call( __FUNCTION__, func_get_args() );
 319         }
 320
 321         public function addQuotes( $s ) {
 322                 return $this->__call( __FUNCTION__, func_get_args() );
 323         }
 324
 325         public function buildLike() {
 326                 return $this->__call( __FUNCTION__, func_get_args() );
 327         }
 328
 329         public function anyChar() {
 330                 return $this->__call( __FUNCTION__, func_get_args() );
 331         }
 332
 333         public function anyString() {
 334                 return $this->__call( __FUNCTION__, func_get_args() );
 335         }
 336
 337         public function nextSequenceValue( $seqName ) {
 338                 return $this->__call( __FUNCTION__, func_get_args() );
 339         }
 340
 341         public function replace( $table, $uniqueIndexes, $rows, $fname = __METHOD__ ) {
 342                 return $this->__call( __FUNCTION__, func_get_args() );
 343         }
 344
 345         public function upsert(
 346                 $table, array $rows, array $uniqueIndexes, array $set, $fname = __METHOD__
 347         ) {
 348                 return $this->__call( __FUNCTION__, func_get_args() );
 349         }
 350
 351         public function deleteJoin(
 352                 $delTable, $joinTable, $delVar, $joinVar, $conds, $fname = __METHOD__
 353         ) {
 354                 return $this->__call( __FUNCTION__, func_get_args() );
 355         }
 356
 357         public function delete( $table, $conds, $fname = __METHOD__ ) {
 358                 return $this->__call( __FUNCTION__, func_get_args() );
 359         }
 360
 361         public function insertSelect(
 362                 $destTable, $srcTable, $varMap, $conds,
 363                 $fname = __METHOD__, $insertOptions = array(), $selectOptions = array()
 364         ) {
 365                 return $this->__call( __FUNCTION__, func_get_args() );
 366         }
 367
 368         public function unionSupportsOrderAndLimit() {
 369                 return $this->__call( __FUNCTION__, func_get_args() );
 370         }
 371
 372         public function unionQueries( $sqls, $all ) {
 373                 return $this->__call( __FUNCTION__, func_get_args() );
 374         }
 375
 376         public function conditional( $cond, $trueVal, $falseVal ) {
 377                 return $this->__call( __FUNCTION__, func_get_args() );
 378         }
 379
 380         public function strreplace( $orig, $old, $new ) {
 381                 return $this->__call( __FUNCTION__, func_get_args() );
 382         }
 383
 384         public function getServerUptime() {
 385                 return $this->__call( __FUNCTION__, func_get_args() );
 386         }
 387
 388         public function wasDeadlock() {
 389                 return $this->__call( __FUNCTION__, func_get_args() );
 390         }
 391
 392         public function wasLockTimeout() {
 393                 return $this->__call( __FUNCTION__, func_get_args() );
 394         }
 395
 396         public function wasErrorReissuable() {
 397                 return $this->__call( __FUNCTION__, func_get_args() );
 398         }
 399
 400         public function wasReadOnlyError() {
 401                 return $this->__call( __FUNCTION__, func_get_args() );
 402         }
 403
 404         public function masterPosWait( DBMasterPos $pos, $timeout ) {
 405                 return $this->__call( __FUNCTION__, func_get_args() );
 406         }
 407
 408         public function getSlavePos() {
 409                 return $this->__call( __FUNCTION__, func_get_args() );
 410         }
 411
 412         public function getMasterPos() {
 413                 return $this->__call( __FUNCTION__, func_get_args() );
 414         }
 415
 416         public function onTransactionIdle( $callback ) {
 417                 return $this->__call( __FUNCTION__, func_get_args() );
 418         }
 419
 420         public function onTransactionPreCommitOrIdle( $callback ) {
 421                 return $this->__call( __FUNCTION__, func_get_args() );
 422         }
 423
 424         public function startAtomic( $fname = __METHOD__ ) {
 425                 return $this->__call( __FUNCTION__, func_get_args() );
 426         }
 427
 428         public function endAtomic( $fname = __METHOD__ ) {
 429                 return $this->__call( __FUNCTION__, func_get_args() );
 430         }
 431
 432         public function begin( $fname = __METHOD__ ) {
 433                 return $this->__call( __FUNCTION__, func_get_args() );
 434         }
 435
 436         public function commit( $fname = __METHOD__, $flush = '' ) {
 437                 return $this->__call( __FUNCTION__, func_get_args() );
 438         }
 439
 440         public function rollback( $fname = __METHOD__, $flush = '' ) {
 441                 return $this->__call( __FUNCTION__, func_get_args() );
 442         }
 443
 444         public function listTables( $prefix = null, $fname = __METHOD__ ) {
 445                 return $this->__call( __FUNCTION__, func_get_args() );
 446         }
 447
 448         public function timestamp( $ts = 0 ) {
 449                 return $this->__call( __FUNCTION__, func_get_args() );
 450         }
 451
 452         public function timestampOrNull( $ts = null ) {
 453                 return $this->__call( __FUNCTION__, func_get_args() );
 454         }
 455
 456         public function ping() {
 457                 return $this->__call( __FUNCTION__, func_get_args() );
 458         }
 459
 460         public function getLag() {
 461                 return $this->__call( __FUNCTION__, func_get_args() );
 462         }
 463
 464         public function getSessionLagStatus() {
 465                 return $this->__call( __FUNCTION__, func_get_args() );
 466         }
 467
 468         public function maxListLen() {
 469                 return $this->__call( __FUNCTION__, func_get_args() );
 470         }
 471
 472         public function encodeBlob( $b ) {
 473                 return $this->__call( __FUNCTION__, func_get_args() );
 474         }
 475
 476         public function decodeBlob( $b ) {
 477                 return $this->__call( __FUNCTION__, func_get_args() );
 478         }
 479
 480         public function setSessionOptions( array $options ) {
 481                 return $this->__call( __FUNCTION__, func_get_args() );
 482         }
 483
 484         public function setSchemaVars( $vars ) {
 485                 return $this->__call( __FUNCTION__, func_get_args() );
 486         }
 487
 488         public function lockIsFree( $lockName, $method ) {
 489                 return $this->__call( __FUNCTION__, func_get_args() );
 490         }
 491
 492         public function lock( $lockName, $method, $timeout = 5 ) {
 493                 return $this->__call( __FUNCTION__, func_get_args() );
 494         }
 495
 496         public function unlock( $lockName, $method ) {
 497                 return $this->__call( __FUNCTION__, func_get_args() );
 498         }
 499
 500         public function namedLocksEnqueue() {
 501                 return $this->__call( __FUNCTION__, func_get_args() );
 502         }
 503
 504         public function getInfinity() {
 505                 return $this->__call( __FUNCTION__, func_get_args() );
 506         }
 507
 508         public function encodeExpiry( $expiry ) {
 509                 return $this->__call( __FUNCTION__, func_get_args() );
 510         }
 511
 512         public function decodeExpiry( $expiry, $format = TS_MW ) {
 513                 return $this->__call( __FUNCTION__, func_get_args() );
 514         }
 515
 516         public function setBigSelects( $value = true ) {
 517                 return $this->__call( __FUNCTION__, func_get_args() );
 518         }
 519
 520         public function isReadOnly() {
 521                 return $this->__call( __FUNCTION__, func_get_args() );
 522         }
 523
 524         /**
 525          * Clean up the connection when out of scope
 526          */
 527         function __destruct() {
 528                 if ( $this->conn !== null ) {
 529                         $this->lb->reuseConnection( $this->conn );
 530                 }
 531         }
 532 }