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
6 * @note: proxy methods are defined explicity to avoid interface errors
10 class DBConnRef
implements IDatabase
{
11 /** @var LoadBalancer */
14 /** @var DatabaseBase|null */
17 /** @var array|null */
21 * @param LoadBalancer $lb
22 * @param DatabaseBase|array $conn Connection or (server index, group, wiki ID) array
24 public function __construct( LoadBalancer
$lb, $conn ) {
26 if ( $conn instanceof DatabaseBase
) {
29 $this->params
= $conn;
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 );
39 return call_user_func_array( [ $this->conn
, $name ], $arguments );
42 public function getServerInfo() {
43 return $this->__call( __FUNCTION__
, func_get_args() );
46 public function bufferResults( $buffer = null ) {
47 return $this->__call( __FUNCTION__
, func_get_args() );
50 public function trxLevel() {
51 return $this->__call( __FUNCTION__
, func_get_args() );
54 public function trxTimestamp() {
55 return $this->__call( __FUNCTION__
, func_get_args() );
58 public function explicitTrxActive() {
59 return $this->__call( __FUNCTION__
, func_get_args() );
62 public function tablePrefix( $prefix = null ) {
63 return $this->__call( __FUNCTION__
, func_get_args() );
66 public function dbSchema( $schema = null ) {
67 return $this->__call( __FUNCTION__
, func_get_args() );
70 public function getLBInfo( $name = null ) {
71 return $this->__call( __FUNCTION__
, func_get_args() );
74 public function setLBInfo( $name, $value = null ) {
75 return $this->__call( __FUNCTION__
, func_get_args() );
78 public function implicitGroupby() {
79 return $this->__call( __FUNCTION__
, func_get_args() );
82 public function implicitOrderby() {
83 return $this->__call( __FUNCTION__
, func_get_args() );
86 public function lastQuery() {
87 return $this->__call( __FUNCTION__
, func_get_args() );
90 public function doneWrites() {
91 return $this->__call( __FUNCTION__
, func_get_args() );
94 public function lastDoneWrites() {
95 return $this->__call( __FUNCTION__
, func_get_args() );
98 public function writesPending() {
99 return $this->__call( __FUNCTION__
, func_get_args() );
102 public function writesOrCallbacksPending() {
103 return $this->__call( __FUNCTION__
, func_get_args() );
106 public function pendingWriteQueryDuration() {
107 return $this->__call( __FUNCTION__
, func_get_args() );
110 public function pendingWriteCallers() {
111 return $this->__call( __FUNCTION__
, func_get_args() );
114 public function isOpen() {
115 return $this->__call( __FUNCTION__
, func_get_args() );
118 public function setFlag( $flag ) {
119 return $this->__call( __FUNCTION__
, func_get_args() );
122 public function clearFlag( $flag ) {
123 return $this->__call( __FUNCTION__
, func_get_args() );
126 public function getFlag( $flag ) {
127 return $this->__call( __FUNCTION__
, func_get_args() );
130 public function getProperty( $name ) {
131 return $this->__call( __FUNCTION__
, func_get_args() );
134 public function getWikiID() {
135 return $this->__call( __FUNCTION__
, func_get_args() );
138 public function getType() {
139 return $this->__call( __FUNCTION__
, func_get_args() );
142 public function open( $server, $user, $password, $dbName ) {
143 return $this->__call( __FUNCTION__
, func_get_args() );
146 public function fetchObject( $res ) {
147 return $this->__call( __FUNCTION__
, func_get_args() );
150 public function fetchRow( $res ) {
151 return $this->__call( __FUNCTION__
, func_get_args() );
154 public function numRows( $res ) {
155 return $this->__call( __FUNCTION__
, func_get_args() );
158 public function numFields( $res ) {
159 return $this->__call( __FUNCTION__
, func_get_args() );
162 public function fieldName( $res, $n ) {
163 return $this->__call( __FUNCTION__
, func_get_args() );
166 public function insertId() {
167 return $this->__call( __FUNCTION__
, func_get_args() );
170 public function dataSeek( $res, $row ) {
171 return $this->__call( __FUNCTION__
, func_get_args() );
174 public function lastErrno() {
175 return $this->__call( __FUNCTION__
, func_get_args() );
178 public function lastError() {
179 return $this->__call( __FUNCTION__
, func_get_args() );
182 public function fieldInfo( $table, $field ) {
183 return $this->__call( __FUNCTION__
, func_get_args() );
186 public function affectedRows() {
187 return $this->__call( __FUNCTION__
, func_get_args() );
190 public function getSoftwareLink() {
191 return $this->__call( __FUNCTION__
, func_get_args() );
194 public function getServerVersion() {
195 return $this->__call( __FUNCTION__
, func_get_args() );
198 public function close() {
199 return $this->__call( __FUNCTION__
, func_get_args() );
202 public function reportConnectionError( $error = 'Unknown error' ) {
203 return $this->__call( __FUNCTION__
, func_get_args() );
206 public function query( $sql, $fname = __METHOD__
, $tempIgnore = false ) {
207 return $this->__call( __FUNCTION__
, func_get_args() );
210 public function reportQueryError( $error, $errno, $sql, $fname, $tempIgnore = false ) {
211 return $this->__call( __FUNCTION__
, func_get_args() );
214 public function freeResult( $res ) {
215 return $this->__call( __FUNCTION__
, func_get_args() );
218 public function selectField(
219 $table, $var, $cond = '', $fname = __METHOD__
, $options = []
221 return $this->__call( __FUNCTION__
, func_get_args() );
224 public function selectFieldValues(
225 $table, $var, $cond = '', $fname = __METHOD__
, $options = []
227 return $this->__call( __FUNCTION__
, func_get_args() );
230 public function select(
231 $table, $vars, $conds = '', $fname = __METHOD__
,
232 $options = [], $join_conds = []
234 return $this->__call( __FUNCTION__
, func_get_args() );
237 public function selectSQLText(
238 $table, $vars, $conds = '', $fname = __METHOD__
,
239 $options = [], $join_conds = []
241 return $this->__call( __FUNCTION__
, func_get_args() );
244 public function selectRow(
245 $table, $vars, $conds, $fname = __METHOD__
,
246 $options = [], $join_conds = []
248 return $this->__call( __FUNCTION__
, func_get_args() );
251 public function estimateRowCount(
252 $table, $vars = '*', $conds = '', $fname = __METHOD__
, $options = []
254 return $this->__call( __FUNCTION__
, func_get_args() );
257 public function selectRowCount(
258 $tables, $vars = '*', $conds = '', $fname = __METHOD__
, $options = [], $join_conds = []
260 return $this->__call( __FUNCTION__
, func_get_args() );
263 public function fieldExists( $table, $field, $fname = __METHOD__
) {
264 return $this->__call( __FUNCTION__
, func_get_args() );
267 public function indexExists( $table, $index, $fname = __METHOD__
) {
268 return $this->__call( __FUNCTION__
, func_get_args() );
271 public function tableExists( $table, $fname = __METHOD__
) {
272 return $this->__call( __FUNCTION__
, func_get_args() );
275 public function indexUnique( $table, $index ) {
276 return $this->__call( __FUNCTION__
, func_get_args() );
279 public function insert( $table, $a, $fname = __METHOD__
, $options = [] ) {
280 return $this->__call( __FUNCTION__
, func_get_args() );
283 public function update( $table, $values, $conds, $fname = __METHOD__
, $options = [] ) {
284 return $this->__call( __FUNCTION__
, func_get_args() );
287 public function makeList( $a, $mode = LIST_COMMA
) {
288 return $this->__call( __FUNCTION__
, func_get_args() );
291 public function makeWhereFrom2d( $data, $baseKey, $subKey ) {
292 return $this->__call( __FUNCTION__
, func_get_args() );
295 public function bitNot( $field ) {
296 return $this->__call( __FUNCTION__
, func_get_args() );
299 public function bitAnd( $fieldLeft, $fieldRight ) {
300 return $this->__call( __FUNCTION__
, func_get_args() );
303 public function bitOr( $fieldLeft, $fieldRight ) {
304 return $this->__call( __FUNCTION__
, func_get_args() );
307 public function buildConcat( $stringList ) {
308 return $this->__call( __FUNCTION__
, func_get_args() );
311 public function buildGroupConcatField(
312 $delim, $table, $field, $conds = '', $join_conds = []
314 return $this->__call( __FUNCTION__
, func_get_args() );
317 public function selectDB( $db ) {
318 return $this->__call( __FUNCTION__
, func_get_args() );
321 public function getDBname() {
322 return $this->__call( __FUNCTION__
, func_get_args() );
325 public function getServer() {
326 return $this->__call( __FUNCTION__
, func_get_args() );
329 public function addQuotes( $s ) {
330 return $this->__call( __FUNCTION__
, func_get_args() );
333 public function buildLike() {
334 return $this->__call( __FUNCTION__
, func_get_args() );
337 public function anyChar() {
338 return $this->__call( __FUNCTION__
, func_get_args() );
341 public function anyString() {
342 return $this->__call( __FUNCTION__
, func_get_args() );
345 public function nextSequenceValue( $seqName ) {
346 return $this->__call( __FUNCTION__
, func_get_args() );
349 public function replace( $table, $uniqueIndexes, $rows, $fname = __METHOD__
) {
350 return $this->__call( __FUNCTION__
, func_get_args() );
353 public function upsert(
354 $table, array $rows, array $uniqueIndexes, array $set, $fname = __METHOD__
356 return $this->__call( __FUNCTION__
, func_get_args() );
359 public function deleteJoin(
360 $delTable, $joinTable, $delVar, $joinVar, $conds, $fname = __METHOD__
362 return $this->__call( __FUNCTION__
, func_get_args() );
365 public function delete( $table, $conds, $fname = __METHOD__
) {
366 return $this->__call( __FUNCTION__
, func_get_args() );
369 public function insertSelect(
370 $destTable, $srcTable, $varMap, $conds,
371 $fname = __METHOD__
, $insertOptions = [], $selectOptions = []
373 return $this->__call( __FUNCTION__
, func_get_args() );
376 public function unionSupportsOrderAndLimit() {
377 return $this->__call( __FUNCTION__
, func_get_args() );
380 public function unionQueries( $sqls, $all ) {
381 return $this->__call( __FUNCTION__
, func_get_args() );
384 public function conditional( $cond, $trueVal, $falseVal ) {
385 return $this->__call( __FUNCTION__
, func_get_args() );
388 public function strreplace( $orig, $old, $new ) {
389 return $this->__call( __FUNCTION__
, func_get_args() );
392 public function getServerUptime() {
393 return $this->__call( __FUNCTION__
, func_get_args() );
396 public function wasDeadlock() {
397 return $this->__call( __FUNCTION__
, func_get_args() );
400 public function wasLockTimeout() {
401 return $this->__call( __FUNCTION__
, func_get_args() );
404 public function wasErrorReissuable() {
405 return $this->__call( __FUNCTION__
, func_get_args() );
408 public function wasReadOnlyError() {
409 return $this->__call( __FUNCTION__
, func_get_args() );
412 public function masterPosWait( DBMasterPos
$pos, $timeout ) {
413 return $this->__call( __FUNCTION__
, func_get_args() );
416 public function getSlavePos() {
417 return $this->__call( __FUNCTION__
, func_get_args() );
420 public function getMasterPos() {
421 return $this->__call( __FUNCTION__
, func_get_args() );
424 public function serverIsReadOnly() {
425 return $this->__call( __FUNCTION__
, func_get_args() );
428 public function onTransactionResolution( callable
$callback ) {
429 return $this->__call( __FUNCTION__
, func_get_args() );
432 public function onTransactionIdle( callable
$callback ) {
433 return $this->__call( __FUNCTION__
, func_get_args() );
436 public function onTransactionPreCommitOrIdle( callable
$callback ) {
437 return $this->__call( __FUNCTION__
, func_get_args() );
440 public function startAtomic( $fname = __METHOD__
) {
441 return $this->__call( __FUNCTION__
, func_get_args() );
444 public function endAtomic( $fname = __METHOD__
) {
445 return $this->__call( __FUNCTION__
, func_get_args() );
448 public function doAtomicSection( $fname, callable
$callback ) {
449 return $this->__call( __FUNCTION__
, func_get_args() );
452 public function begin( $fname = __METHOD__
, $mode = IDatabase
::TRANSACTION_EXPLICIT
) {
453 return $this->__call( __FUNCTION__
, func_get_args() );
456 public function commit( $fname = __METHOD__
, $flush = '' ) {
457 return $this->__call( __FUNCTION__
, func_get_args() );
460 public function rollback( $fname = __METHOD__
, $flush = '' ) {
461 return $this->__call( __FUNCTION__
, func_get_args() );
464 public function listTables( $prefix = null, $fname = __METHOD__
) {
465 return $this->__call( __FUNCTION__
, func_get_args() );
468 public function timestamp( $ts = 0 ) {
469 return $this->__call( __FUNCTION__
, func_get_args() );
472 public function timestampOrNull( $ts = null ) {
473 return $this->__call( __FUNCTION__
, func_get_args() );
476 public function ping() {
477 return $this->__call( __FUNCTION__
, func_get_args() );
480 public function getLag() {
481 return $this->__call( __FUNCTION__
, func_get_args() );
484 public function getSessionLagStatus() {
485 return $this->__call( __FUNCTION__
, func_get_args() );
488 public function maxListLen() {
489 return $this->__call( __FUNCTION__
, func_get_args() );
492 public function encodeBlob( $b ) {
493 return $this->__call( __FUNCTION__
, func_get_args() );
496 public function decodeBlob( $b ) {
497 return $this->__call( __FUNCTION__
, func_get_args() );
500 public function setSessionOptions( array $options ) {
501 return $this->__call( __FUNCTION__
, func_get_args() );
504 public function setSchemaVars( $vars ) {
505 return $this->__call( __FUNCTION__
, func_get_args() );
508 public function lockIsFree( $lockName, $method ) {
509 return $this->__call( __FUNCTION__
, func_get_args() );
512 public function lock( $lockName, $method, $timeout = 5 ) {
513 return $this->__call( __FUNCTION__
, func_get_args() );
516 public function unlock( $lockName, $method ) {
517 return $this->__call( __FUNCTION__
, func_get_args() );
520 public function getScopedLockAndFlush( $lockKey, $fname, $timeout ) {
521 return $this->__call( __FUNCTION__
, func_get_args() );
524 public function namedLocksEnqueue() {
525 return $this->__call( __FUNCTION__
, func_get_args() );
528 public function getInfinity() {
529 return $this->__call( __FUNCTION__
, func_get_args() );
532 public function encodeExpiry( $expiry ) {
533 return $this->__call( __FUNCTION__
, func_get_args() );
536 public function decodeExpiry( $expiry, $format = TS_MW
) {
537 return $this->__call( __FUNCTION__
, func_get_args() );
540 public function setBigSelects( $value = true ) {
541 return $this->__call( __FUNCTION__
, func_get_args() );
544 public function isReadOnly() {
545 return $this->__call( __FUNCTION__
, func_get_args() );
549 * Clean up the connection when out of scope
551 function __destruct() {
552 if ( $this->conn
!== null ) {
553 $this->lb
->reuseConnection( $this->conn
);