123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042 |
- <?php
- abstract class DatabaseConnection extends PDO {
-
- protected $target = NULL;
-
- protected $key = NULL;
-
- protected $logger = NULL;
-
- protected $transactionLayers = array();
-
- protected $driverClasses = array();
-
- protected $statementClass = 'DatabaseStatementBase';
-
- protected $transactionSupport = TRUE;
-
- protected $transactionalDDLSupport = FALSE;
-
- protected $temporaryNameIndex = 0;
-
- protected $connectionOptions = array();
-
- protected $schema = NULL;
-
- protected $prefixes = array();
-
- protected $prefixSearch = array();
-
- protected $prefixReplace = array();
- function __construct($dsn, $username, $password, $driver_options = array()) {
-
- $this->setPrefix(isset($this->connectionOptions['prefix']) ? $this->connectionOptions['prefix'] : '');
-
- $driver_options[PDO::ATTR_ERRMODE] = PDO::ERRMODE_EXCEPTION;
-
- parent::__construct($dsn, $username, $password, $driver_options);
-
- if (!empty($this->statementClass)) {
- $this->setAttribute(PDO::ATTR_STATEMENT_CLASS, array($this->statementClass, array($this)));
- }
- }
-
- public function destroy() {
-
-
-
- $this->setAttribute(PDO::ATTR_STATEMENT_CLASS, array('PDOStatement', array()));
- $this->schema = NULL;
- }
-
- protected function defaultOptions() {
- return array(
- 'target' => 'default',
- 'fetch' => PDO::FETCH_OBJ,
- 'return' => Database::RETURN_STATEMENT,
- 'throw_exception' => TRUE,
- );
- }
-
- public function getConnectionOptions() {
- return $this->connectionOptions;
- }
-
- protected function setPrefix($prefix) {
- if (is_array($prefix)) {
- $this->prefixes = $prefix + array('default' => '');
- }
- else {
- $this->prefixes = array('default' => $prefix);
- }
-
-
- $this->prefixSearch = array();
- $this->prefixReplace = array();
- foreach ($this->prefixes as $key => $val) {
- if ($key != 'default') {
- $this->prefixSearch[] = '{' . $key . '}';
- $this->prefixReplace[] = $val . $key;
- }
- }
-
- $this->prefixSearch[] = '{';
- $this->prefixReplace[] = $this->prefixes['default'];
- $this->prefixSearch[] = '}';
- $this->prefixReplace[] = '';
- }
-
- public function prefixTables($sql) {
- return str_replace($this->prefixSearch, $this->prefixReplace, $sql);
- }
-
- public function tablePrefix($table = 'default') {
- if (isset($this->prefixes[$table])) {
- return $this->prefixes[$table];
- }
- else {
- return $this->prefixes['default'];
- }
- }
-
- public function prepareQuery($query) {
- $query = $this->prefixTables($query);
-
- return parent::prepare($query);
- }
-
- public function setTarget($target = NULL) {
- if (!isset($this->target)) {
- $this->target = $target;
- }
- }
-
- public function getTarget() {
- return $this->target;
- }
-
- public function setKey($key) {
- if (!isset($this->key)) {
- $this->key = $key;
- }
- }
-
- public function getKey() {
- return $this->key;
- }
-
- public function setLogger(DatabaseLog $logger) {
- $this->logger = $logger;
- }
-
- public function getLogger() {
- return $this->logger;
- }
-
- public function makeSequenceName($table, $field) {
- return $this->prefixTables('{' . $table . '}_' . $field . '_seq');
- }
-
- public function makeComment($comments) {
- if (empty($comments))
- return '';
-
- $comment = implode('; ', $comments);
-
- return '/* ' . $this->filterComment($comment) . ' */ ';
- }
-
- protected function filterComment($comment = '') {
- return strtr($comment, array('*' => ' * '));
- }
-
- public function query($query, array $args = array(), $options = array()) {
-
- $options += $this->defaultOptions();
- try {
-
-
-
- if ($query instanceof DatabaseStatementInterface) {
- $stmt = $query;
- $stmt->execute(NULL, $options);
- }
- else {
- $this->expandArguments($query, $args);
- $stmt = $this->prepareQuery($query);
- $stmt->execute($args, $options);
- }
-
-
-
- switch ($options['return']) {
- case Database::RETURN_STATEMENT:
- return $stmt;
- case Database::RETURN_AFFECTED:
- return $stmt->rowCount();
- case Database::RETURN_INSERT_ID:
- return $this->lastInsertId();
- case Database::RETURN_NULL:
- return;
- default:
- throw new PDOException('Invalid return directive: ' . $options['return']);
- }
- }
- catch (PDOException $e) {
- if ($options['throw_exception']) {
-
- if ($query instanceof DatabaseStatementInterface) {
- $e->query_string = $stmt->getQueryString();
- }
- else {
- $e->query_string = $query;
- }
- $e->args = $args;
- throw $e;
- }
- return NULL;
- }
- }
-
- protected function expandArguments(&$query, &$args) {
- $modified = FALSE;
-
-
- foreach (array_filter($args, 'is_array') as $key => $data) {
- $new_keys = array();
- foreach (array_values($data) as $i => $value) {
-
-
-
-
-
- $new_keys[$key . '_' . $i] = $value;
- }
-
-
-
-
-
-
-
- $query = preg_replace('#' . $key . '\b#', implode(', ', array_keys($new_keys)), $query);
-
- unset($args[$key]);
- $args += $new_keys;
- $modified = TRUE;
- }
- return $modified;
- }
-
- public function getDriverClass($class, array $files = array(), $use_autoload = FALSE) {
- if (empty($this->driverClasses[$class])) {
- $driver = $this->driver();
- $this->driverClasses[$class] = $class . '_' . $driver;
- Database::loadDriverFile($driver, $files);
- if (!class_exists($this->driverClasses[$class], $use_autoload)) {
- $this->driverClasses[$class] = $class;
- }
- }
- return $this->driverClasses[$class];
- }
-
- public function select($table, $alias = NULL, array $options = array()) {
- $class = $this->getDriverClass('SelectQuery', array('query.inc', 'select.inc'));
- return new $class($table, $alias, $this, $options);
- }
-
- public function insert($table, array $options = array()) {
- $class = $this->getDriverClass('InsertQuery', array('query.inc'));
- return new $class($this, $table, $options);
- }
-
- public function merge($table, array $options = array()) {
- $class = $this->getDriverClass('MergeQuery', array('query.inc'));
- return new $class($this, $table, $options);
- }
-
- public function update($table, array $options = array()) {
- $class = $this->getDriverClass('UpdateQuery', array('query.inc'));
- return new $class($this, $table, $options);
- }
-
- public function delete($table, array $options = array()) {
- $class = $this->getDriverClass('DeleteQuery', array('query.inc'));
- return new $class($this, $table, $options);
- }
-
- public function truncate($table, array $options = array()) {
- $class = $this->getDriverClass('TruncateQuery', array('query.inc'));
- return new $class($this, $table, $options);
- }
-
- public function schema() {
- if (empty($this->schema)) {
- $class = $this->getDriverClass('DatabaseSchema', array('schema.inc'));
- if (class_exists($class)) {
- $this->schema = new $class($this);
- }
- }
- return $this->schema;
- }
-
- public function escapeTable($table) {
- return preg_replace('/[^A-Za-z0-9_.]+/', '', $table);
- }
-
- public function escapeField($field) {
- return preg_replace('/[^A-Za-z0-9_.]+/', '', $field);
- }
-
- public function escapeAlias($field) {
- return preg_replace('/[^A-Za-z0-9_]+/', '', $field);
- }
-
- public function escapeLike($string) {
- return addcslashes($string, '\%_');
- }
-
- public function inTransaction() {
- return ($this->transactionDepth() > 0);
- }
-
- public function transactionDepth() {
- return count($this->transactionLayers);
- }
-
- public function startTransaction($name = '') {
- $class = $this->getDriverClass('DatabaseTransaction');
- return new $class($this, $name);
- }
-
- public function rollback($savepoint_name = 'drupal_transaction') {
- if (!$this->supportsTransactions()) {
- return;
- }
- if (!$this->inTransaction()) {
- throw new DatabaseTransactionNoActiveException();
- }
-
-
- if (!isset($this->transactionLayers[$savepoint_name])) {
- throw new DatabaseTransactionNoActiveException();
- }
-
-
-
- $rolled_back_other_active_savepoints = FALSE;
- while ($savepoint = array_pop($this->transactionLayers)) {
- if ($savepoint == $savepoint_name) {
-
-
-
- if (empty($this->transactionLayers)) {
- break;
- }
- $this->query('ROLLBACK TO SAVEPOINT ' . $savepoint);
- $this->popCommittableTransactions();
- if ($rolled_back_other_active_savepoints) {
- throw new DatabaseTransactionOutOfOrderException();
- }
- return;
- }
- else {
- $rolled_back_other_active_savepoints = TRUE;
- }
- }
- parent::rollBack();
- if ($rolled_back_other_active_savepoints) {
- throw new DatabaseTransactionOutOfOrderException();
- }
- }
-
- public function pushTransaction($name) {
- if (!$this->supportsTransactions()) {
- return;
- }
- if (isset($this->transactionLayers[$name])) {
- throw new DatabaseTransactionNameNonUniqueException($name . " is already in use.");
- }
-
-
- if ($this->inTransaction()) {
- $this->query('SAVEPOINT ' . $name);
- }
- else {
- parent::beginTransaction();
- }
- $this->transactionLayers[$name] = $name;
- }
-
- public function popTransaction($name) {
- if (!$this->supportsTransactions()) {
- return;
- }
-
-
-
-
- if (!isset($this->transactionLayers[$name])) {
- return;
- }
-
- $this->transactionLayers[$name] = FALSE;
- $this->popCommittableTransactions();
- }
-
- protected function popCommittableTransactions() {
-
- foreach (array_reverse($this->transactionLayers) as $name => $active) {
-
- if ($active) {
- break;
- }
-
- unset($this->transactionLayers[$name]);
- if (empty($this->transactionLayers)) {
- if (!parent::commit()) {
- throw new DatabaseTransactionCommitFailedException();
- }
- }
- else {
- $this->query('RELEASE SAVEPOINT ' . $name);
- }
- }
- }
-
- abstract public function queryRange($query, $from, $count, array $args = array(), array $options = array());
-
- protected function generateTemporaryTableName() {
- return "db_temporary_" . $this->temporaryNameIndex++;
- }
-
- abstract function queryTemporary($query, array $args = array(), array $options = array());
-
- abstract public function driver();
-
- public function version() {
- return $this->getAttribute(PDO::ATTR_SERVER_VERSION);
- }
-
- public function supportsTransactions() {
- return $this->transactionSupport;
- }
-
- public function supportsTransactionalDDL() {
- return $this->transactionalDDLSupport;
- }
-
- abstract public function databaseType();
-
- abstract public function mapConditionOperator($operator);
-
- public function commit() {
- throw new DatabaseTransactionExplicitCommitNotAllowedException();
- }
-
- abstract public function nextId($existing_id = 0);
- }
- abstract class Database {
-
- const RETURN_NULL = 0;
-
- const RETURN_STATEMENT = 1;
-
- const RETURN_AFFECTED = 2;
-
- const RETURN_INSERT_ID = 3;
-
- static protected $connections = array();
-
- static protected $databaseInfo = NULL;
-
- static protected $ignoreTargets = array();
-
- static protected $activeKey = 'default';
-
- static protected $logs = array();
-
- final public static function startLog($logging_key, $key = 'default') {
- if (empty(self::$logs[$key])) {
- self::$logs[$key] = new DatabaseLog($key);
-
-
- if (!empty(self::$connections[$key])) {
- foreach (self::$connections[$key] as $connection) {
- $connection->setLogger(self::$logs[$key]);
- }
- }
- }
- self::$logs[$key]->start($logging_key);
- return self::$logs[$key];
- }
-
- final public static function getLog($logging_key, $key = 'default') {
- if (empty(self::$logs[$key])) {
- return NULL;
- }
- $queries = self::$logs[$key]->get($logging_key);
- self::$logs[$key]->end($logging_key);
- return $queries;
- }
-
- final public static function getConnection($target = 'default', $key = NULL) {
- if (!isset($key)) {
-
- $key = self::$activeKey;
- }
-
-
-
-
-
- if (!empty(self::$ignoreTargets[$key][$target]) || !isset(self::$databaseInfo[$key][$target])) {
- $target = 'default';
- }
- if (!isset(self::$connections[$key][$target])) {
-
- self::$connections[$key][$target] = self::openConnection($key, $target);
- }
- return self::$connections[$key][$target];
- }
-
- final public static function isActiveConnection() {
- return !empty(self::$activeKey) && !empty(self::$connections) && !empty(self::$connections[self::$activeKey]);
- }
-
- final public static function setActiveConnection($key = 'default') {
- if (empty(self::$databaseInfo)) {
- self::parseConnectionInfo();
- }
- if (!empty(self::$databaseInfo[$key])) {
- $old_key = self::$activeKey;
- self::$activeKey = $key;
- return $old_key;
- }
- }
-
- final public static function parseConnectionInfo() {
- global $databases;
- $database_info = is_array($databases) ? $databases : array();
- foreach ($database_info as $index => $info) {
- foreach ($database_info[$index] as $target => $value) {
-
-
-
- if (empty($value['driver'])) {
- $database_info[$index][$target] = $database_info[$index][$target][mt_rand(0, count($database_info[$index][$target]) - 1)];
- }
-
- if (!isset($database_info[$index][$target]['prefix'])) {
-
- $database_info[$index][$target]['prefix'] = array(
- 'default' => '',
- );
- }
- elseif (!is_array($database_info[$index][$target]['prefix'])) {
-
- $database_info[$index][$target]['prefix'] = array(
- 'default' => $database_info[$index][$target]['prefix'],
- );
- }
- }
- }
- if (!is_array(self::$databaseInfo)) {
- self::$databaseInfo = $database_info;
- }
-
-
-
- else {
- foreach ($database_info as $database_key => $database_values) {
- foreach ($database_values as $target => $target_values) {
- self::$databaseInfo[$database_key][$target] = $target_values;
- }
- }
- }
- }
-
- public static function addConnectionInfo($key, $target, $info) {
- if (empty(self::$databaseInfo[$key][$target])) {
- self::$databaseInfo[$key][$target] = $info;
- }
- }
-
- final public static function getConnectionInfo($key = 'default') {
- if (empty(self::$databaseInfo)) {
- self::parseConnectionInfo();
- }
- if (!empty(self::$databaseInfo[$key])) {
- return self::$databaseInfo[$key];
- }
- }
-
- final public static function renameConnection($old_key, $new_key) {
- if (empty(self::$databaseInfo)) {
- self::parseConnectionInfo();
- }
- if (!empty(self::$databaseInfo[$old_key]) && empty(self::$databaseInfo[$new_key])) {
-
- self::$databaseInfo[$new_key] = self::$databaseInfo[$old_key];
- unset(self::$databaseInfo[$old_key]);
-
- if (isset(self::$connections[$old_key])) {
- self::$connections[$new_key] = self::$connections[$old_key];
- unset(self::$connections[$old_key]);
- }
- return TRUE;
- }
- else {
- return FALSE;
- }
- }
-
- final public static function removeConnection($key) {
- if (isset(self::$databaseInfo[$key])) {
- self::closeConnection(NULL, $key);
- unset(self::$databaseInfo[$key]);
- return TRUE;
- }
- else {
- return FALSE;
- }
- }
-
- final protected static function openConnection($key, $target) {
- if (empty(self::$databaseInfo)) {
- self::parseConnectionInfo();
- }
-
-
- if (!isset(self::$databaseInfo[$key])) {
- throw new DatabaseConnectionNotDefinedException('The specified database connection is not defined: ' . $key);
- }
- if (!$driver = self::$databaseInfo[$key][$target]['driver']) {
- throw new DatabaseDriverNotSpecifiedException('Driver not specified for this database connection: ' . $key);
- }
-
-
- $driver_class = 'DatabaseConnection_' . $driver;
- require_once DRUPAL_ROOT . '/includes/database/' . $driver . '/database.inc';
- $new_connection = new $driver_class(self::$databaseInfo[$key][$target]);
- $new_connection->setTarget($target);
- $new_connection->setKey($key);
-
-
- if (!empty(self::$logs[$key])) {
- $new_connection->setLogger(self::$logs[$key]);
- }
- return $new_connection;
- }
-
- public static function closeConnection($target = NULL, $key = NULL) {
-
- if (!isset($key)) {
- $key = self::$activeKey;
- }
-
-
-
-
- if (isset($target)) {
- if (isset(self::$connections[$key][$target])) {
- self::$connections[$key][$target]->destroy();
- self::$connections[$key][$target] = NULL;
- }
- unset(self::$connections[$key][$target]);
- }
- else {
- if (isset(self::$connections[$key])) {
- foreach (self::$connections[$key] as $target => $connection) {
- self::$connections[$key][$target]->destroy();
- self::$connections[$key][$target] = NULL;
- }
- }
- unset(self::$connections[$key]);
- }
- }
-
- public static function ignoreTarget($key, $target) {
- self::$ignoreTargets[$key][$target] = TRUE;
- }
-
- public static function loadDriverFile($driver, array $files = array()) {
- static $base_path;
- if (empty($base_path)) {
- $base_path = dirname(realpath(__FILE__));
- }
- $driver_base_path = "$base_path/$driver";
- foreach ($files as $file) {
-
-
- foreach (array("$base_path/$file", "$driver_base_path/$file") as $filename) {
-
-
- if (file_exists($filename)) {
- require_once $filename;
- }
- }
- }
- }
- }
- class DatabaseTransactionNoActiveException extends Exception { }
- class DatabaseTransactionNameNonUniqueException extends Exception { }
- class DatabaseTransactionCommitFailedException extends Exception { }
- class DatabaseTransactionExplicitCommitNotAllowedException extends Exception { }
- class DatabaseTransactionOutOfOrderException extends Exception { }
- class InvalidMergeQueryException extends Exception {}
- class FieldsOverlapException extends Exception {}
- class NoFieldsException extends Exception {}
- class DatabaseConnectionNotDefinedException extends Exception {}
- class DatabaseDriverNotSpecifiedException extends Exception {}
- class DatabaseTransaction {
-
- protected $connection;
-
- protected $rolledBack = FALSE;
-
- protected $name;
- public function __construct(DatabaseConnection $connection, $name = NULL) {
- $this->connection = $connection;
-
-
- if (!$depth = $connection->transactionDepth()) {
- $this->name = 'drupal_transaction';
- }
-
-
- elseif (!$name) {
- $this->name = 'savepoint_' . $depth;
- }
- else {
- $this->name = $name;
- }
- $this->connection->pushTransaction($this->name);
- }
- public function __destruct() {
-
- if (!$this->rolledBack) {
- $this->connection->popTransaction($this->name);
- }
- }
-
- public function name() {
- return $this->name;
- }
-
- public function rollback() {
- $this->rolledBack = TRUE;
- $this->connection->rollback($this->name);
- }
- }
- interface DatabaseStatementInterface extends Traversable {
-
- public function execute($args = array(), $options = array());
-
- public function getQueryString();
-
- public function rowCount();
-
-
-
-
-
- public function fetchField($index = 0);
-
-
-
- public function fetchAssoc();
-
-
-
- public function fetchCol($index = 0);
-
- public function fetchAllKeyed($key_index = 0, $value_index = 1);
-
- public function fetchAllAssoc($key, $fetch = NULL);
- }
- class DatabaseStatementBase extends PDOStatement implements DatabaseStatementInterface {
-
- public $dbh;
- protected function __construct($dbh) {
- $this->dbh = $dbh;
- $this->setFetchMode(PDO::FETCH_OBJ);
- }
- public function execute($args = array(), $options = array()) {
- if (isset($options['fetch'])) {
- if (is_string($options['fetch'])) {
-
-
-
- $this->setFetchMode(PDO::FETCH_CLASS, $options['fetch']);
- }
- else {
- $this->setFetchMode($options['fetch']);
- }
- }
- $logger = $this->dbh->getLogger();
- if (!empty($logger)) {
- $query_start = microtime(TRUE);
- }
- $return = parent::execute($args);
- if (!empty($logger)) {
- $query_end = microtime(TRUE);
- $logger->log($this, $args, $query_end - $query_start);
- }
- return $return;
- }
- public function getQueryString() {
- return $this->queryString;
- }
- public function fetchCol($index = 0) {
- return $this->fetchAll(PDO::FETCH_COLUMN, $index);
- }
- public function fetchAllAssoc($key, $fetch = NULL) {
- $return = array();
- if (isset($fetch)) {
- if (is_string($fetch)) {
- $this->setFetchMode(PDO::FETCH_CLASS, $fetch);
- }
- else {
- $this->setFetchMode($fetch);
- }
- }
- foreach ($this as $record) {
- $record_key = is_object($record) ? $record->$key : $record[$key];
- $return[$record_key] = $record;
- }
- return $return;
- }
- public function fetchAllKeyed($key_index = 0, $value_index = 1) {
- $return = array();
- $this->setFetchMode(PDO::FETCH_NUM);
- foreach ($this as $record) {
- $return[$record[$key_index]] = $record[$value_index];
- }
- return $return;
- }
- public function fetchField($index = 0) {
-
- return $this->fetchColumn($index);
- }
- public function fetchAssoc() {
-
- return $this->fetch(PDO::FETCH_ASSOC);
- }
- }
- class DatabaseStatementEmpty implements Iterator, DatabaseStatementInterface {
- public function execute($args = array(), $options = array()) {
- return FALSE;
- }
- public function getQueryString() {
- return '';
- }
- public function rowCount() {
- return 0;
- }
- public function setFetchMode($mode, $a1 = NULL, $a2 = array()) {
- return;
- }
- public function fetch($mode = NULL, $cursor_orientation = NULL, $cursor_offset = NULL) {
- return NULL;
- }
- public function fetchField($index = 0) {
- return NULL;
- }
- public function fetchObject() {
- return NULL;
- }
- public function fetchAssoc() {
- return NULL;
- }
- function fetchAll($mode = NULL, $column_index = NULL, array $constructor_arguments = array()) {
- return array();
- }
- public function fetchCol($index = 0) {
- return array();
- }
- public function fetchAllKeyed($key_index = 0, $value_index = 1) {
- return array();
- }
- public function fetchAllAssoc($key, $fetch = NULL) {
- return array();
- }
-
- public function current() {
- return NULL;
- }
- public function key() {
- return NULL;
- }
- public function rewind() {
-
- }
- public function next() {
-
- }
- public function valid() {
- return FALSE;
- }
- }
- function db_query($query, array $args = array(), array $options = array()) {
- if (empty($options['target'])) {
- $options['target'] = 'default';
- }
- return Database::getConnection($options['target'])->query($query, $args, $options);
- }
- function db_query_range($query, $from, $count, array $args = array(), array $options = array()) {
- if (empty($options['target'])) {
- $options['target'] = 'default';
- }
- return Database::getConnection($options['target'])->queryRange($query, $from, $count, $args, $options);
- }
- function db_query_temporary($query, array $args = array(), array $options = array()) {
- if (empty($options['target'])) {
- $options['target'] = 'default';
- }
- return Database::getConnection($options['target'])->queryTemporary($query, $args, $options);
- }
- function db_insert($table, array $options = array()) {
- if (empty($options['target']) || $options['target'] == 'slave') {
- $options['target'] = 'default';
- }
- return Database::getConnection($options['target'])->insert($table, $options);
- }
- function db_merge($table, array $options = array()) {
- if (empty($options['target']) || $options['target'] == 'slave') {
- $options['target'] = 'default';
- }
- return Database::getConnection($options['target'])->merge($table, $options);
- }
- function db_update($table, array $options = array()) {
- if (empty($options['target']) || $options['target'] == 'slave') {
- $options['target'] = 'default';
- }
- return Database::getConnection($options['target'])->update($table, $options);
- }
- function db_delete($table, array $options = array()) {
- if (empty($options['target']) || $options['target'] == 'slave') {
- $options['target'] = 'default';
- }
- return Database::getConnection($options['target'])->delete($table, $options);
- }
- function db_truncate($table, array $options = array()) {
- if (empty($options['target']) || $options['target'] == 'slave') {
- $options['target'] = 'default';
- }
- return Database::getConnection($options['target'])->truncate($table, $options);
- }
- function db_select($table, $alias = NULL, array $options = array()) {
- if (empty($options['target'])) {
- $options['target'] = 'default';
- }
- return Database::getConnection($options['target'])->select($table, $alias, $options);
- }
- function db_transaction($name = NULL, array $options = array()) {
- if (empty($options['target'])) {
- $options['target'] = 'default';
- }
- return Database::getConnection($options['target'])->startTransaction($name);
- }
- function db_set_active($key = 'default') {
- return Database::setActiveConnection($key);
- }
- function db_escape_table($table) {
- return Database::getConnection()->escapeTable($table);
- }
- function db_escape_field($field) {
- return Database::getConnection()->escapeField($field);
- }
- function db_like($string) {
- return Database::getConnection()->escapeLike($string);
- }
- function db_driver() {
- return Database::getConnection()->driver();
- }
- function db_close(array $options = array()) {
- if (empty($options['target'])) {
- $options['target'] = NULL;
- }
- Database::closeConnection($options['target']);
- }
- function db_next_id($existing_id = 0) {
- return Database::getConnection()->nextId($existing_id);
- }
- function db_or() {
- return new DatabaseCondition('OR');
- }
- function db_and() {
- return new DatabaseCondition('AND');
- }
- function db_xor() {
- return new DatabaseCondition('XOR');
- }
- function db_condition($conjunction) {
- return new DatabaseCondition($conjunction);
- }
- function db_create_table($name, $table) {
- return Database::getConnection()->schema()->createTable($name, $table);
- }
- function db_field_names($fields) {
- return Database::getConnection()->schema()->fieldNames($fields);
- }
- function db_index_exists($table, $name) {
- return Database::getConnection()->schema()->indexExists($table, $name);
- }
- function db_table_exists($table) {
- return Database::getConnection()->schema()->tableExists($table);
- }
- function db_field_exists($table, $field) {
- return Database::getConnection()->schema()->fieldExists($table, $field);
- }
- function db_find_tables($table_expression) {
- return Database::getConnection()->schema()->findTables($table_expression);
- }
- function _db_create_keys_sql($spec) {
- return Database::getConnection()->schema()->createKeysSql($spec);
- }
- function db_rename_table($table, $new_name) {
- return Database::getConnection()->schema()->renameTable($table, $new_name);
- }
- function db_drop_table($table) {
- return Database::getConnection()->schema()->dropTable($table);
- }
- function db_add_field($table, $field, $spec, $keys_new = array()) {
- return Database::getConnection()->schema()->addField($table, $field, $spec, $keys_new);
- }
- function db_drop_field($table, $field) {
- return Database::getConnection()->schema()->dropField($table, $field);
- }
- function db_field_set_default($table, $field, $default) {
- return Database::getConnection()->schema()->fieldSetDefault($table, $field, $default);
- }
- function db_field_set_no_default($table, $field) {
- return Database::getConnection()->schema()->fieldSetNoDefault($table, $field);
- }
- function db_add_primary_key($table, $fields) {
- return Database::getConnection()->schema()->addPrimaryKey($table, $fields);
- }
- function db_drop_primary_key($table) {
- return Database::getConnection()->schema()->dropPrimaryKey($table);
- }
- function db_add_unique_key($table, $name, $fields) {
- return Database::getConnection()->schema()->addUniqueKey($table, $name, $fields);
- }
- function db_drop_unique_key($table, $name) {
- return Database::getConnection()->schema()->dropUniqueKey($table, $name);
- }
- function db_add_index($table, $name, $fields) {
- return Database::getConnection()->schema()->addIndex($table, $name, $fields);
- }
- function db_drop_index($table, $name) {
- return Database::getConnection()->schema()->dropIndex($table, $name);
- }
- function db_change_field($table, $field, $field_new, $spec, $keys_new = array()) {
- return Database::getConnection()->schema()->changeField($table, $field, $field_new, $spec, $keys_new);
- }
- function db_ignore_slave() {
- $connection_info = Database::getConnectionInfo();
-
-
- if (count($connection_info) > 1) {
-
-
-
- $duration = variable_get('maximum_replication_lag', 300);
-
- $_SESSION['ignore_slave_server'] = REQUEST_TIME + $duration;
- }
- }
|