MediaWiki REL1_30
DatabaseMssql.php
Go to the documentation of this file.
1<?php
28namespace Wikimedia\Rdbms;
29
30use MediaWiki;
31use Exception;
32use stdClass;
33
37class DatabaseMssql extends Database {
38 protected $mPort;
39 protected $mUseWindowsAuth = false;
40
41 protected $mInsertId = null;
42 protected $mLastResult = null;
43 protected $mAffectedRows = null;
44 protected $mSubqueryId = 0;
45 protected $mScrollableCursor = true;
46 protected $mPrepareStatements = true;
47 protected $mBinaryColumnCache = null;
48 protected $mBitColumnCache = null;
49 protected $mIgnoreDupKeyErrors = false;
50 protected $mIgnoreErrors = [];
51
52 public function implicitGroupby() {
53 return false;
54 }
55
56 public function implicitOrderby() {
57 return false;
58 }
59
60 public function unionSupportsOrderAndLimit() {
61 return false;
62 }
63
64 public function __construct( array $params ) {
65 $this->mPort = $params['port'];
66 $this->mUseWindowsAuth = $params['UseWindowsAuth'];
67
68 parent::__construct( $params );
69 }
70
80 public function open( $server, $user, $password, $dbName ) {
81 # Test for driver support, to avoid suppressed fatal error
82 if ( !function_exists( 'sqlsrv_connect' ) ) {
83 throw new DBConnectionError(
84 $this,
85 "Microsoft SQL Server Native (sqlsrv) functions missing.
86 You can download the driver from: http://go.microsoft.com/fwlink/?LinkId=123470\n"
87 );
88 }
89
90 # e.g. the class is being loaded
91 if ( !strlen( $user ) ) {
92 return null;
93 }
94
95 $this->close();
96 $this->mServer = $server;
97 $this->mUser = $user;
98 $this->mPassword = $password;
99 $this->mDBname = $dbName;
100
101 $connectionInfo = [];
102
103 if ( $dbName ) {
104 $connectionInfo['Database'] = $dbName;
105 }
106
107 // Decide which auth scenerio to use
108 // if we are using Windows auth, then don't add credentials to $connectionInfo
109 if ( !$this->mUseWindowsAuth ) {
110 $connectionInfo['UID'] = $user;
111 $connectionInfo['PWD'] = $password;
112 }
113
114 MediaWiki\suppressWarnings();
115 $this->mConn = sqlsrv_connect( $server, $connectionInfo );
116 MediaWiki\restoreWarnings();
117
118 if ( $this->mConn === false ) {
119 throw new DBConnectionError( $this, $this->lastError() );
120 }
121
122 $this->mOpened = true;
123
124 return $this->mConn;
125 }
126
132 protected function closeConnection() {
133 return sqlsrv_close( $this->mConn );
134 }
135
140 protected function resultObject( $result ) {
141 if ( !$result ) {
142 return false;
143 } elseif ( $result instanceof MssqlResultWrapper ) {
144 return $result;
145 } elseif ( $result === true ) {
146 // Successful write query
147 return $result;
148 } else {
149 return new MssqlResultWrapper( $this, $result );
150 }
151 }
152
158 protected function doQuery( $sql ) {
159 // several extensions seem to think that all databases support limits
160 // via LIMIT N after the WHERE clause, but MSSQL uses SELECT TOP N,
161 // so to catch any of those extensions we'll do a quick check for a
162 // LIMIT clause and pass $sql through $this->LimitToTopN() which parses
163 // the LIMIT clause and passes the result to $this->limitResult();
164 if ( preg_match( '/\bLIMIT\s*/i', $sql ) ) {
165 // massage LIMIT -> TopN
166 $sql = $this->LimitToTopN( $sql );
167 }
168
169 // MSSQL doesn't have EXTRACT(epoch FROM XXX)
170 if ( preg_match( '#\bEXTRACT\s*?\‍(\s*?EPOCH\s+FROM\b#i', $sql, $matches ) ) {
171 // This is same as UNIX_TIMESTAMP, we need to calc # of seconds from 1970
172 $sql = str_replace( $matches[0], "DATEDIFF(s,CONVERT(datetime,'1/1/1970'),", $sql );
173 }
174
175 // perform query
176
177 // SQLSRV_CURSOR_STATIC is slower than SQLSRV_CURSOR_CLIENT_BUFFERED (one of the two is
178 // needed if we want to be able to seek around the result set), however CLIENT_BUFFERED
179 // has a bug in the sqlsrv driver where wchar_t types (such as nvarchar) that are empty
180 // strings make php throw a fatal error "Severe error translating Unicode"
181 if ( $this->mScrollableCursor ) {
182 $scrollArr = [ 'Scrollable' => SQLSRV_CURSOR_STATIC ];
183 } else {
184 $scrollArr = [];
185 }
186
187 if ( $this->mPrepareStatements ) {
188 // we do prepare + execute so we can get its field metadata for later usage if desired
189 $stmt = sqlsrv_prepare( $this->mConn, $sql, [], $scrollArr );
190 $success = sqlsrv_execute( $stmt );
191 } else {
192 $stmt = sqlsrv_query( $this->mConn, $sql, [], $scrollArr );
193 $success = (bool)$stmt;
194 }
195
196 // Make a copy to ensure what we add below does not get reflected in future queries
197 $ignoreErrors = $this->mIgnoreErrors;
198
199 if ( $this->mIgnoreDupKeyErrors ) {
200 // ignore duplicate key errors
201 // this emulates INSERT IGNORE in MySQL
202 $ignoreErrors[] = '2601'; // duplicate key error caused by unique index
203 $ignoreErrors[] = '2627'; // duplicate key error caused by primary key
204 $ignoreErrors[] = '3621'; // generic "the statement has been terminated" error
205 }
206
207 if ( $success === false ) {
208 $errors = sqlsrv_errors();
209 $success = true;
210
211 foreach ( $errors as $err ) {
212 if ( !in_array( $err['code'], $ignoreErrors ) ) {
213 $success = false;
214 break;
215 }
216 }
217
218 if ( $success === false ) {
219 return false;
220 }
221 }
222 // remember number of rows affected
223 $this->mAffectedRows = sqlsrv_rows_affected( $stmt );
224
225 return $stmt;
226 }
227
228 public function freeResult( $res ) {
229 if ( $res instanceof ResultWrapper ) {
230 $res = $res->result;
231 }
232
233 sqlsrv_free_stmt( $res );
234 }
235
240 public function fetchObject( $res ) {
241 // $res is expected to be an instance of MssqlResultWrapper here
242 return $res->fetchObject();
243 }
244
249 public function fetchRow( $res ) {
250 return $res->fetchRow();
251 }
252
257 public function numRows( $res ) {
258 if ( $res instanceof ResultWrapper ) {
259 $res = $res->result;
260 }
261
262 $ret = sqlsrv_num_rows( $res );
263
264 if ( $ret === false ) {
265 // we cannot get an amount of rows from this cursor type
266 // has_rows returns bool true/false if the result has rows
267 $ret = (int)sqlsrv_has_rows( $res );
268 }
269
270 return $ret;
271 }
272
277 public function numFields( $res ) {
278 if ( $res instanceof ResultWrapper ) {
279 $res = $res->result;
280 }
281
282 return sqlsrv_num_fields( $res );
283 }
284
290 public function fieldName( $res, $n ) {
291 if ( $res instanceof ResultWrapper ) {
292 $res = $res->result;
293 }
294
295 return sqlsrv_field_metadata( $res )[$n]['Name'];
296 }
297
302 public function insertId() {
303 return $this->mInsertId;
304 }
305
311 public function dataSeek( $res, $row ) {
312 return $res->seek( $row );
313 }
314
318 public function lastError() {
319 $strRet = '';
320 $retErrors = sqlsrv_errors( SQLSRV_ERR_ALL );
321 if ( $retErrors != null ) {
322 foreach ( $retErrors as $arrError ) {
323 $strRet .= $this->formatError( $arrError ) . "\n";
324 }
325 } else {
326 $strRet = "No errors found";
327 }
328
329 return $strRet;
330 }
331
336 private function formatError( $err ) {
337 return '[SQLSTATE ' .
338 $err['SQLSTATE'] . '][Error Code ' . $err['code'] . ']' . $err['message'];
339 }
340
344 public function lastErrno() {
345 $err = sqlsrv_errors( SQLSRV_ERR_ALL );
346 if ( $err !== null && isset( $err[0] ) ) {
347 return $err[0]['code'];
348 } else {
349 return 0;
350 }
351 }
352
356 public function affectedRows() {
358 }
359
378 public function select( $table, $vars, $conds = '', $fname = __METHOD__,
379 $options = [], $join_conds = []
380 ) {
381 $sql = $this->selectSQLText( $table, $vars, $conds, $fname, $options, $join_conds );
382 if ( isset( $options['EXPLAIN'] ) ) {
383 try {
384 $this->mScrollableCursor = false;
385 $this->mPrepareStatements = false;
386 $this->query( "SET SHOWPLAN_ALL ON" );
387 $ret = $this->query( $sql, $fname );
388 $this->query( "SET SHOWPLAN_ALL OFF" );
389 } catch ( DBQueryError $dqe ) {
390 if ( isset( $options['FOR COUNT'] ) ) {
391 // likely don't have privs for SHOWPLAN, so run a select count instead
392 $this->query( "SET SHOWPLAN_ALL OFF" );
393 unset( $options['EXPLAIN'] );
394 $ret = $this->select(
395 $table,
396 'COUNT(*) AS EstimateRows',
397 $conds,
398 $fname,
399 $options,
400 $join_conds
401 );
402 } else {
403 // someone actually wanted the query plan instead of an est row count
404 // let them know of the error
405 $this->mScrollableCursor = true;
406 $this->mPrepareStatements = true;
407 throw $dqe;
408 }
409 }
410 $this->mScrollableCursor = true;
411 $this->mPrepareStatements = true;
412 return $ret;
413 }
414 return $this->query( $sql, $fname );
415 }
416
430 public function selectSQLText( $table, $vars, $conds = '', $fname = __METHOD__,
431 $options = [], $join_conds = []
432 ) {
433 if ( isset( $options['EXPLAIN'] ) ) {
434 unset( $options['EXPLAIN'] );
435 }
436
437 $sql = parent::selectSQLText( $table, $vars, $conds, $fname, $options, $join_conds );
438
439 // try to rewrite aggregations of bit columns (currently MAX and MIN)
440 if ( strpos( $sql, 'MAX(' ) !== false || strpos( $sql, 'MIN(' ) !== false ) {
441 $bitColumns = [];
442 if ( is_array( $table ) ) {
443 foreach ( $table as $t ) {
444 $bitColumns += $this->getBitColumns( $this->tableName( $t ) );
445 }
446 } else {
447 $bitColumns = $this->getBitColumns( $this->tableName( $table ) );
448 }
449
450 foreach ( $bitColumns as $col => $info ) {
451 $replace = [
452 "MAX({$col})" => "MAX(CAST({$col} AS tinyint))",
453 "MIN({$col})" => "MIN(CAST({$col} AS tinyint))",
454 ];
455 $sql = str_replace( array_keys( $replace ), array_values( $replace ), $sql );
456 }
457 }
458
459 return $sql;
460 }
461
462 public function deleteJoin( $delTable, $joinTable, $delVar, $joinVar, $conds,
463 $fname = __METHOD__
464 ) {
465 $this->mScrollableCursor = false;
466 try {
467 parent::deleteJoin( $delTable, $joinTable, $delVar, $joinVar, $conds, $fname );
468 } catch ( Exception $e ) {
469 $this->mScrollableCursor = true;
470 throw $e;
471 }
472 $this->mScrollableCursor = true;
473 }
474
475 public function delete( $table, $conds, $fname = __METHOD__ ) {
476 $this->mScrollableCursor = false;
477 try {
478 parent::delete( $table, $conds, $fname );
479 } catch ( Exception $e ) {
480 $this->mScrollableCursor = true;
481 throw $e;
482 }
483 $this->mScrollableCursor = true;
484 }
485
499 public function estimateRowCount( $table, $vars = '*', $conds = '',
500 $fname = __METHOD__, $options = []
501 ) {
502 // http://msdn2.microsoft.com/en-us/library/aa259203.aspx
503 $options['EXPLAIN'] = true;
504 $options['FOR COUNT'] = true;
505 $res = $this->select( $table, $vars, $conds, $fname, $options );
506
507 $rows = -1;
508 if ( $res ) {
509 $row = $this->fetchRow( $res );
510
511 if ( isset( $row['EstimateRows'] ) ) {
512 $rows = (int)$row['EstimateRows'];
513 }
514 }
515
516 return $rows;
517 }
518
527 public function indexInfo( $table, $index, $fname = __METHOD__ ) {
528 # This does not return the same info as MYSQL would, but that's OK
529 # because MediaWiki never uses the returned value except to check for
530 # the existence of indexes.
531 $sql = "sp_helpindex '" . $this->tableName( $table ) . "'";
532 $res = $this->query( $sql, $fname );
533
534 if ( !$res ) {
535 return null;
536 }
537
538 $result = [];
539 foreach ( $res as $row ) {
540 if ( $row->index_name == $index ) {
541 $row->Non_unique = !stristr( $row->index_description, "unique" );
542 $cols = explode( ", ", $row->index_keys );
543 foreach ( $cols as $col ) {
544 $row->Column_name = trim( $col );
545 $result[] = clone $row;
546 }
547 } elseif ( $index == 'PRIMARY' && stristr( $row->index_description, 'PRIMARY' ) ) {
548 $row->Non_unique = 0;
549 $cols = explode( ", ", $row->index_keys );
550 foreach ( $cols as $col ) {
551 $row->Column_name = trim( $col );
552 $result[] = clone $row;
553 }
554 }
555 }
556
557 return empty( $result ) ? false : $result;
558 }
559
575 public function insert( $table, $arrToInsert, $fname = __METHOD__, $options = [] ) {
576 # No rows to insert, easy just return now
577 if ( !count( $arrToInsert ) ) {
578 return true;
579 }
580
581 if ( !is_array( $options ) ) {
582 $options = [ $options ];
583 }
584
585 $table = $this->tableName( $table );
586
587 if ( !( isset( $arrToInsert[0] ) && is_array( $arrToInsert[0] ) ) ) { // Not multi row
588 $arrToInsert = [ 0 => $arrToInsert ]; // make everything multi row compatible
589 }
590
591 // We know the table we're inserting into, get its identity column
592 $identity = null;
593 // strip matching square brackets and the db/schema from table name
594 $tableRawArr = explode( '.', preg_replace( '#\[([^\]]*)\]#', '$1', $table ) );
595 $tableRaw = array_pop( $tableRawArr );
596 $res = $this->doQuery(
597 "SELECT NAME AS idColumn FROM SYS.IDENTITY_COLUMNS " .
598 "WHERE OBJECT_NAME(OBJECT_ID)='{$tableRaw}'"
599 );
600 if ( $res && sqlsrv_has_rows( $res ) ) {
601 // There is an identity for this table.
602 $identityArr = sqlsrv_fetch_array( $res, SQLSRV_FETCH_ASSOC );
603 $identity = array_pop( $identityArr );
604 }
605 sqlsrv_free_stmt( $res );
606
607 // Determine binary/varbinary fields so we can encode data as a hex string like 0xABCDEF
608 $binaryColumns = $this->getBinaryColumns( $table );
609
610 // INSERT IGNORE is not supported by SQL Server
611 // remove IGNORE from options list and set ignore flag to true
612 if ( in_array( 'IGNORE', $options ) ) {
613 $options = array_diff( $options, [ 'IGNORE' ] );
614 $this->mIgnoreDupKeyErrors = true;
615 }
616
617 $ret = null;
618 foreach ( $arrToInsert as $a ) {
619 // start out with empty identity column, this is so we can return
620 // it as a result of the INSERT logic
621 $sqlPre = '';
622 $sqlPost = '';
623 $identityClause = '';
624
625 // if we have an identity column
626 if ( $identity ) {
627 // iterate through
628 foreach ( $a as $k => $v ) {
629 if ( $k == $identity ) {
630 if ( !is_null( $v ) ) {
631 // there is a value being passed to us,
632 // we need to turn on and off inserted identity
633 $sqlPre = "SET IDENTITY_INSERT $table ON;";
634 $sqlPost = ";SET IDENTITY_INSERT $table OFF;";
635 } else {
636 // we can't insert NULL into an identity column,
637 // so remove the column from the insert.
638 unset( $a[$k] );
639 }
640 }
641 }
642
643 // we want to output an identity column as result
644 $identityClause = "OUTPUT INSERTED.$identity ";
645 }
646
647 $keys = array_keys( $a );
648
649 // Build the actual query
650 $sql = $sqlPre . 'INSERT ' . implode( ' ', $options ) .
651 " INTO $table (" . implode( ',', $keys ) . ") $identityClause VALUES (";
652
653 $first = true;
654 foreach ( $a as $key => $value ) {
655 if ( isset( $binaryColumns[$key] ) ) {
656 $value = new MssqlBlob( $value );
657 }
658 if ( $first ) {
659 $first = false;
660 } else {
661 $sql .= ',';
662 }
663 if ( is_null( $value ) ) {
664 $sql .= 'null';
665 } elseif ( is_array( $value ) || is_object( $value ) ) {
666 if ( is_object( $value ) && $value instanceof Blob ) {
667 $sql .= $this->addQuotes( $value );
668 } else {
669 $sql .= $this->addQuotes( serialize( $value ) );
670 }
671 } else {
672 $sql .= $this->addQuotes( $value );
673 }
674 }
675 $sql .= ')' . $sqlPost;
676
677 // Run the query
678 $this->mScrollableCursor = false;
679 try {
680 $ret = $this->query( $sql );
681 } catch ( Exception $e ) {
682 $this->mScrollableCursor = true;
683 $this->mIgnoreDupKeyErrors = false;
684 throw $e;
685 }
686 $this->mScrollableCursor = true;
687
688 if ( $ret instanceof ResultWrapper && !is_null( $identity ) ) {
689 // Then we want to get the identity column value we were assigned and save it off
690 $row = $ret->fetchObject();
691 if ( is_object( $row ) ) {
692 $this->mInsertId = $row->$identity;
693 // It seems that mAffectedRows is -1 sometimes when OUTPUT INSERTED.identity is
694 // used if we got an identity back, we know for sure a row was affected, so
695 // adjust that here
696 if ( $this->mAffectedRows == -1 ) {
697 $this->mAffectedRows = 1;
698 }
699 }
700 }
701 }
702
703 $this->mIgnoreDupKeyErrors = false;
704
705 return $ret;
706 }
707
724 public function nativeInsertSelect( $destTable, $srcTable, $varMap, $conds, $fname = __METHOD__,
725 $insertOptions = [], $selectOptions = [], $selectJoinConds = []
726 ) {
727 $this->mScrollableCursor = false;
728 try {
729 $ret = parent::nativeInsertSelect(
730 $destTable,
731 $srcTable,
732 $varMap,
733 $conds,
734 $fname,
735 $insertOptions,
736 $selectOptions,
737 $selectJoinConds
738 );
739 } catch ( Exception $e ) {
740 $this->mScrollableCursor = true;
741 throw $e;
742 }
743 $this->mScrollableCursor = true;
744
745 return $ret;
746 }
747
773 function update( $table, $values, $conds, $fname = __METHOD__, $options = [] ) {
774 $table = $this->tableName( $table );
775 $binaryColumns = $this->getBinaryColumns( $table );
776
777 $opts = $this->makeUpdateOptions( $options );
778 $sql = "UPDATE $opts $table SET " . $this->makeList( $values, LIST_SET, $binaryColumns );
779
780 if ( $conds !== [] && $conds !== '*' ) {
781 $sql .= " WHERE " . $this->makeList( $conds, LIST_AND, $binaryColumns );
782 }
783
784 $this->mScrollableCursor = false;
785 try {
786 $this->query( $sql );
787 } catch ( Exception $e ) {
788 $this->mScrollableCursor = true;
789 throw $e;
790 }
791 $this->mScrollableCursor = true;
792 return true;
793 }
794
811 public function makeList( $a, $mode = LIST_COMMA, $binaryColumns = [] ) {
812 if ( !is_array( $a ) ) {
813 throw new DBUnexpectedError( $this, __METHOD__ . ' called with incorrect parameters' );
814 }
815
816 if ( $mode != LIST_NAMES ) {
817 // In MS SQL, values need to be specially encoded when they are
818 // inserted into binary fields. Perform this necessary encoding
819 // for the specified set of columns.
820 foreach ( array_keys( $a ) as $field ) {
821 if ( !isset( $binaryColumns[$field] ) ) {
822 continue;
823 }
824
825 if ( is_array( $a[$field] ) ) {
826 foreach ( $a[$field] as &$v ) {
827 $v = new MssqlBlob( $v );
828 }
829 unset( $v );
830 } else {
831 $a[$field] = new MssqlBlob( $a[$field] );
832 }
833 }
834 }
835
836 return parent::makeList( $a, $mode );
837 }
838
844 public function textFieldSize( $table, $field ) {
845 $table = $this->tableName( $table );
846 $sql = "SELECT CHARACTER_MAXIMUM_LENGTH,DATA_TYPE FROM INFORMATION_SCHEMA.Columns
847 WHERE TABLE_NAME = '$table' AND COLUMN_NAME = '$field'";
848 $res = $this->query( $sql );
849 $row = $this->fetchRow( $res );
850 $size = -1;
851 if ( strtolower( $row['DATA_TYPE'] ) != 'text' ) {
852 $size = $row['CHARACTER_MAXIMUM_LENGTH'];
853 }
854
855 return $size;
856 }
857
868 public function limitResult( $sql, $limit, $offset = false ) {
869 if ( $offset === false || $offset == 0 ) {
870 if ( strpos( $sql, "SELECT" ) === false ) {
871 return "TOP {$limit} " . $sql;
872 } else {
873 return preg_replace( '/\bSELECT(\s+DISTINCT)?\b/Dsi',
874 'SELECT$1 TOP ' . $limit, $sql, 1 );
875 }
876 } else {
877 // This one is fun, we need to pull out the select list as well as any ORDER BY clause
878 $select = $orderby = [];
879 $s1 = preg_match( '#SELECT\s+(.+?)\s+FROM#Dis', $sql, $select );
880 $s2 = preg_match( '#(ORDER BY\s+.+?)(\s*FOR XML .*)?$#Dis', $sql, $orderby );
881 $postOrder = '';
882 $first = $offset + 1;
883 $last = $offset + $limit;
884 $sub1 = 'sub_' . $this->mSubqueryId;
885 $sub2 = 'sub_' . ( $this->mSubqueryId + 1 );
886 $this->mSubqueryId += 2;
887 if ( !$s1 ) {
888 // wat
889 throw new DBUnexpectedError( $this, "Attempting to LIMIT a non-SELECT query\n" );
890 }
891 if ( !$s2 ) {
892 // no ORDER BY
893 $overOrder = 'ORDER BY (SELECT 1)';
894 } else {
895 if ( !isset( $orderby[2] ) || !$orderby[2] ) {
896 // don't need to strip it out if we're using a FOR XML clause
897 $sql = str_replace( $orderby[1], '', $sql );
898 }
899 $overOrder = $orderby[1];
900 $postOrder = ' ' . $overOrder;
901 }
902 $sql = "SELECT {$select[1]}
903 FROM (
904 SELECT ROW_NUMBER() OVER({$overOrder}) AS rowNumber, *
905 FROM ({$sql}) {$sub1}
906 ) {$sub2}
907 WHERE rowNumber BETWEEN {$first} AND {$last}{$postOrder}";
908
909 return $sql;
910 }
911 }
912
923 public function LimitToTopN( $sql ) {
924 // Matches: LIMIT {[offset,] row_count | row_count OFFSET offset}
925 $pattern = '/\bLIMIT\s+((([0-9]+)\s*,\s*)?([0-9]+)(\s+OFFSET\s+([0-9]+))?)/i';
926 if ( preg_match( $pattern, $sql, $matches ) ) {
927 $row_count = $matches[4];
928 $offset = $matches[3] ?: $matches[6] ?: false;
929
930 // strip the matching LIMIT clause out
931 $sql = str_replace( $matches[0], '', $sql );
932
933 return $this->limitResult( $sql, $row_count, $offset );
934 }
935
936 return $sql;
937 }
938
942 public function getSoftwareLink() {
943 return "[{{int:version-db-mssql-url}} MS SQL Server]";
944 }
945
949 public function getServerVersion() {
950 $server_info = sqlsrv_server_info( $this->mConn );
951 $version = 'Error';
952 if ( isset( $server_info['SQLServerVersion'] ) ) {
953 $version = $server_info['SQLServerVersion'];
954 }
955
956 return $version;
957 }
958
964 public function tableExists( $table, $fname = __METHOD__ ) {
965 list( $db, $schema, $table ) = $this->tableName( $table, 'split' );
966
967 if ( $db !== false ) {
968 // remote database
969 $this->queryLogger->error( "Attempting to call tableExists on a remote table" );
970 return false;
971 }
972
973 if ( $schema === false ) {
974 $schema = $this->mSchema;
975 }
976
977 $res = $this->query( "SELECT 1 FROM INFORMATION_SCHEMA.TABLES
978 WHERE TABLE_TYPE = 'BASE TABLE'
979 AND TABLE_SCHEMA = '$schema' AND TABLE_NAME = '$table'" );
980
981 if ( $res->numRows() ) {
982 return true;
983 } else {
984 return false;
985 }
986 }
987
995 public function fieldExists( $table, $field, $fname = __METHOD__ ) {
996 list( $db, $schema, $table ) = $this->tableName( $table, 'split' );
997
998 if ( $db !== false ) {
999 // remote database
1000 $this->queryLogger->error( "Attempting to call fieldExists on a remote table" );
1001 return false;
1002 }
1003
1004 $res = $this->query( "SELECT 1 FROM INFORMATION_SCHEMA.COLUMNS
1005 WHERE TABLE_SCHEMA = '$schema' AND TABLE_NAME = '$table' AND COLUMN_NAME = '$field'" );
1006
1007 if ( $res->numRows() ) {
1008 return true;
1009 } else {
1010 return false;
1011 }
1012 }
1013
1014 public function fieldInfo( $table, $field ) {
1015 list( $db, $schema, $table ) = $this->tableName( $table, 'split' );
1016
1017 if ( $db !== false ) {
1018 // remote database
1019 $this->queryLogger->error( "Attempting to call fieldInfo on a remote table" );
1020 return false;
1021 }
1022
1023 $res = $this->query( "SELECT * FROM INFORMATION_SCHEMA.COLUMNS
1024 WHERE TABLE_SCHEMA = '$schema' AND TABLE_NAME = '$table' AND COLUMN_NAME = '$field'" );
1025
1026 $meta = $res->fetchRow();
1027 if ( $meta ) {
1028 return new MssqlField( $meta );
1029 }
1030
1031 return false;
1032 }
1033
1038 protected function doBegin( $fname = __METHOD__ ) {
1039 sqlsrv_begin_transaction( $this->mConn );
1040 $this->mTrxLevel = 1;
1041 }
1042
1047 protected function doCommit( $fname = __METHOD__ ) {
1048 sqlsrv_commit( $this->mConn );
1049 $this->mTrxLevel = 0;
1050 }
1051
1057 protected function doRollback( $fname = __METHOD__ ) {
1058 sqlsrv_rollback( $this->mConn );
1059 $this->mTrxLevel = 0;
1060 }
1061
1066 public function strencode( $s ) {
1067 // Should not be called by us
1068 return str_replace( "'", "''", $s );
1069 }
1070
1075 public function addQuotes( $s ) {
1076 if ( $s instanceof MssqlBlob ) {
1077 return $s->fetch();
1078 } elseif ( $s instanceof Blob ) {
1079 // this shouldn't really ever be called, but it's here if needed
1080 // (and will quite possibly make the SQL error out)
1081 $blob = new MssqlBlob( $s->fetch() );
1082 return $blob->fetch();
1083 } else {
1084 if ( is_bool( $s ) ) {
1085 $s = $s ? 1 : 0;
1086 }
1087 return parent::addQuotes( $s );
1088 }
1089 }
1090
1095 public function addIdentifierQuotes( $s ) {
1096 // http://msdn.microsoft.com/en-us/library/aa223962.aspx
1097 return '[' . $s . ']';
1098 }
1099
1104 public function isQuotedIdentifier( $name ) {
1105 return strlen( $name ) && $name[0] == '[' && substr( $name, -1, 1 ) == ']';
1106 }
1107
1115 protected function escapeLikeInternal( $s, $escapeChar = '`' ) {
1116 return str_replace( [ $escapeChar, '%', '_', '[', ']', '^' ],
1117 [ "{$escapeChar}{$escapeChar}", "{$escapeChar}%", "{$escapeChar}_",
1118 "{$escapeChar}[", "{$escapeChar}]", "{$escapeChar}^" ],
1119 $s );
1120 }
1121
1126 public function selectDB( $db ) {
1127 try {
1128 $this->mDBname = $db;
1129 $this->query( "USE $db" );
1130 return true;
1131 } catch ( Exception $e ) {
1132 return false;
1133 }
1134 }
1135
1141 public function makeSelectOptions( $options ) {
1142 $tailOpts = '';
1143 $startOpts = '';
1144
1145 $noKeyOptions = [];
1146 foreach ( $options as $key => $option ) {
1147 if ( is_numeric( $key ) ) {
1148 $noKeyOptions[$option] = true;
1149 }
1150 }
1151
1152 $tailOpts .= $this->makeGroupByWithHaving( $options );
1153
1154 $tailOpts .= $this->makeOrderBy( $options );
1155
1156 if ( isset( $noKeyOptions['DISTINCT'] ) || isset( $noKeyOptions['DISTINCTROW'] ) ) {
1157 $startOpts .= 'DISTINCT';
1158 }
1159
1160 if ( isset( $noKeyOptions['FOR XML'] ) ) {
1161 // used in group concat field emulation
1162 $tailOpts .= " FOR XML PATH('')";
1163 }
1164
1165 // we want this to be compatible with the output of parent::makeSelectOptions()
1166 return [ $startOpts, '', $tailOpts, '', '' ];
1167 }
1168
1169 public function getType() {
1170 return 'mssql';
1171 }
1172
1177 public function buildConcat( $stringList ) {
1178 return implode( ' + ', $stringList );
1179 }
1180
1198 public function buildGroupConcatField( $delim, $table, $field, $conds = '',
1199 $join_conds = []
1200 ) {
1201 $gcsq = 'gcsq_' . $this->mSubqueryId;
1202 $this->mSubqueryId++;
1203
1204 $delimLen = strlen( $delim );
1205 $fld = "{$field} + {$this->addQuotes( $delim )}";
1206 $sql = "(SELECT LEFT({$field}, LEN({$field}) - {$delimLen}) FROM ("
1207 . $this->selectSQLText( $table, $fld, $conds, null, [ 'FOR XML' ], $join_conds )
1208 . ") {$gcsq} ({$field}))";
1209
1210 return $sql;
1211 }
1212
1219 private function getBinaryColumns( $table ) {
1220 $tableRawArr = explode( '.', preg_replace( '#\[([^\]]*)\]#', '$1', $table ) );
1221 $tableRaw = array_pop( $tableRawArr );
1222
1223 if ( $this->mBinaryColumnCache === null ) {
1224 $this->populateColumnCaches();
1225 }
1226
1227 return isset( $this->mBinaryColumnCache[$tableRaw] )
1228 ? $this->mBinaryColumnCache[$tableRaw]
1229 : [];
1230 }
1231
1236 private function getBitColumns( $table ) {
1237 $tableRawArr = explode( '.', preg_replace( '#\[([^\]]*)\]#', '$1', $table ) );
1238 $tableRaw = array_pop( $tableRawArr );
1239
1240 if ( $this->mBitColumnCache === null ) {
1241 $this->populateColumnCaches();
1242 }
1243
1244 return isset( $this->mBitColumnCache[$tableRaw] )
1245 ? $this->mBitColumnCache[$tableRaw]
1246 : [];
1247 }
1248
1249 private function populateColumnCaches() {
1250 $res = $this->select( 'INFORMATION_SCHEMA.COLUMNS', '*',
1251 [
1252 'TABLE_CATALOG' => $this->mDBname,
1253 'TABLE_SCHEMA' => $this->mSchema,
1254 'DATA_TYPE' => [ 'varbinary', 'binary', 'image', 'bit' ]
1255 ] );
1256
1257 $this->mBinaryColumnCache = [];
1258 $this->mBitColumnCache = [];
1259 foreach ( $res as $row ) {
1260 if ( $row->DATA_TYPE == 'bit' ) {
1261 $this->mBitColumnCache[$row->TABLE_NAME][$row->COLUMN_NAME] = $row;
1262 } else {
1263 $this->mBinaryColumnCache[$row->TABLE_NAME][$row->COLUMN_NAME] = $row;
1264 }
1265 }
1266 }
1267
1273 function tableName( $name, $format = 'quoted' ) {
1274 # Replace reserved words with better ones
1275 switch ( $name ) {
1276 case 'user':
1277 return $this->realTableName( 'mwuser', $format );
1278 default:
1279 return $this->realTableName( $name, $format );
1280 }
1281 }
1282
1289 function realTableName( $name, $format = 'quoted' ) {
1290 $table = parent::tableName( $name, $format );
1291 if ( $format == 'split' ) {
1292 // Used internally, we want the schema split off from the table name and returned
1293 // as a list with 3 elements (database, schema, table)
1294 $table = explode( '.', $table );
1295 while ( count( $table ) < 3 ) {
1296 array_unshift( $table, false );
1297 }
1298 }
1299 return $table;
1300 }
1301
1309 public function dropTable( $tableName, $fName = __METHOD__ ) {
1310 if ( !$this->tableExists( $tableName, $fName ) ) {
1311 return false;
1312 }
1313
1314 // parent function incorrectly appends CASCADE, which we don't want
1315 $sql = "DROP TABLE " . $this->tableName( $tableName );
1316
1317 return $this->query( $sql, $fName );
1318 }
1319
1326 public function prepareStatements( $value = null ) {
1328 if ( $value !== null ) {
1329 $this->mPrepareStatements = $value;
1330 }
1331
1332 return $old;
1333 }
1334
1341 public function scrollableCursor( $value = null ) {
1343 if ( $value !== null ) {
1344 $this->mScrollableCursor = $value;
1345 }
1346
1347 return $old;
1348 }
1349}
1350
1351class_alias( DatabaseMssql::class, 'DatabaseMssql' );
serialize()
if(!defined( 'MEDIAWIKI')) $fname
This file is not a valid entry point, perform no further processing unless MEDIAWIKI is defined.
Definition Setup.php:36
insert( $table, $arrToInsert, $fname=__METHOD__, $options=[])
INSERT wrapper, inserts an array into a table.
escapeLikeInternal( $s, $escapeChar='`')
MS SQL supports more pattern operators than other databases (ex: [,],^)
closeConnection()
Closes a database connection, if it is open Returns success, true if already closed.
doCommit( $fname=__METHOD__)
End a transaction.
fieldInfo( $table, $field)
mysql_fetch_field() wrapper Returns false if the field doesn't exist
prepareStatements( $value=null)
Called in the installer and updater.
insertId()
This must be called after nextSequenceVal.
implicitOrderby()
Returns true if this database does an implicit order by when the column has an index For example: SEL...
open( $server, $user, $password, $dbName)
Usually aborts on failure.
LimitToTopN( $sql)
If there is a limit clause, parse it, strip it, and pass the remaining SQL through limitResult() with...
selectSQLText( $table, $vars, $conds='', $fname=__METHOD__, $options=[], $join_conds=[])
SELECT wrapper.
estimateRowCount( $table, $vars=' *', $conds='', $fname=__METHOD__, $options=[])
Estimate rows in dataset Returns estimated count, based on SHOWPLAN_ALL output This is not necessaril...
realTableName( $name, $format='quoted')
call this instead of tableName() in the updater when renaming tables
buildGroupConcatField( $delim, $table, $field, $conds='', $join_conds=[])
Build a GROUP_CONCAT or equivalent statement for a query.
nativeInsertSelect( $destTable, $srcTable, $varMap, $conds, $fname=__METHOD__, $insertOptions=[], $selectOptions=[], $selectJoinConds=[])
INSERT SELECT wrapper $varMap must be an associative array of the form [ 'dest1' => 'source1',...
limitResult( $sql, $limit, $offset=false)
Construct a LIMIT query with optional offset This is used for query pages.
dropTable( $tableName, $fName=__METHOD__)
Delete a table.
freeResult( $res)
Free a result object returned by query() or select().
fieldExists( $table, $field, $fname=__METHOD__)
Query whether a given column exists in the mediawiki schema.
makeList( $a, $mode=LIST_COMMA, $binaryColumns=[])
Makes an encoded list of strings from an array.
indexInfo( $table, $index, $fname=__METHOD__)
Returns information about an index If errors are explicitly ignored, returns NULL on failure.
tableName( $name, $format='quoted')
scrollableCursor( $value=null)
Called in the installer and updater.
select( $table, $vars, $conds='', $fname=__METHOD__, $options=[], $join_conds=[])
SELECT wrapper.
doRollback( $fname=__METHOD__)
Rollback a transaction.
__construct(array $params)
Constructor and database handle and attempt to connect to the DB server.
doBegin( $fname=__METHOD__)
Begin a transaction, committing any previously open transaction.
unionSupportsOrderAndLimit()
Returns true if current database backend supports ORDER BY or LIMIT for separate subqueries within th...
tableExists( $table, $fname=__METHOD__)
deleteJoin( $delTable, $joinTable, $delVar, $joinVar, $conds, $fname=__METHOD__)
DELETE where the condition is a join.
implicitGroupby()
Returns true if this database does an implicit sort when doing GROUP BY.
update( $table, $values, $conds, $fname=__METHOD__, $options=[])
UPDATE wrapper.
getType()
Get the type of the DBMS, as it appears in $wgDBtype.
getBinaryColumns( $table)
Returns an associative array for fields that are of type varbinary, binary, or image $table can be ei...
Relational database abstraction object.
Definition Database.php:45
makeUpdateOptions( $options)
Make UPDATE options for the Database::update function.
makeGroupByWithHaving( $options)
Returns an optional GROUP BY with an optional HAVING.
resource null $mConn
Database connection.
Definition Database.php:92
makeOrderBy( $options)
Returns an optional ORDER BY.
close()
Closes a database connection.
Definition Database.php:753
Result wrapper for grabbing data queried from an IDatabase object.
We use the convention $dbr for read and $dbw for write to help you keep track of whether the database object is a the world will explode Or to be a subsequent write query which succeeded on the master may fail when replicated to the slave due to a unique key collision Replication on the slave will stop and it may take hours to repair the database and get it back online Setting read_only in my cnf on the slave will avoid this but given the dire we prefer to have as many checks as possible We provide a but the wrapper functions like select() and insert() are usually more convenient. They take care of things like table prefixes and escaping for you. If you really need to make your own SQL
$res
Definition database.txt:21
For a write query
Definition database.txt:26
We use the convention $dbr for read and $dbw for write to help you keep track of whether the database object is a the world will explode Or to be a subsequent write query which succeeded on the master may fail when replicated to the slave due to a unique key collision Replication on the slave will stop and it may take hours to repair the database and get it back online Setting read_only in my cnf on the slave will avoid this but given the dire we prefer to have as many checks as possible We provide a but the wrapper functions like please read the documentation for tableName() and addQuotes(). You will need both of them. ------------------------------------------------------------------------ Basic query optimisation ------------------------------------------------------------------------ MediaWiki developers who need to write DB queries should have some understanding of databases and the performance issues associated with them. Patches containing unacceptably slow features will not be accepted. Unindexed queries are generally not welcome in MediaWiki
deferred txt A few of the database updates required by various functions here can be deferred until after the result page is displayed to the user For updating the view updating the linked to tables after a etc PHP does not yet have any way to tell the server to actually return and disconnect while still running these but it might have such a feature in the future We handle these by creating a deferred update object and putting those objects on a global list
Definition deferred.txt:11
static configuration should be added through ResourceLoaderGetConfigVars instead & $vars
Definition hooks.txt:2198
namespace being checked & $result
Definition hooks.txt:2293
do that in ParserLimitReportFormat instead use this to modify the parameters of the image all existing parser cache entries will be invalid To avoid you ll need to handle that somehow(e.g. with the RejectParserCacheValue hook) because MediaWiki won 't do it for you. & $defaults also a ContextSource after deleting those rows but within the same transaction $rows
Definition hooks.txt:2746
null means default in associative array with keys and values unescaped Should be merged with default with a value of false meaning to suppress the attribute in associative array with keys and values unescaped & $options
Definition hooks.txt:1971
null means default in associative array with keys and values unescaped Should be merged with default with a value of false meaning to suppress the attribute in associative array with keys and values unescaped noclasses & $ret
Definition hooks.txt:1975
processing should stop and the error should be shown to the user * false
Definition hooks.txt:187
please add to it if you re going to add events to the MediaWiki code where normally authentication against an external auth plugin would be creating a local account $user
Definition hooks.txt:247
returning false will NOT prevent logging $e
Definition hooks.txt:2146
const LIST_NAMES
Definition Defines.php:46
const LIST_COMMA
Definition Defines.php:43
const LIST_SET
Definition Defines.php:45
const LIST_AND
Definition Defines.php:44
A helper class for throttling authentication attempts.
$last
$params