1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483 |
- <?php
- /**
- * Lithium: the most rad php framework
- *
- * @copyright Copyright 2013, Union of RAD (http://union-of-rad.org)
- * @license http://opensource.org/licenses/bsd-license.php The BSD License
- */
- namespace lithium\data\source;
- use PDO;
- use PDOException;
- use lithium\util\Set;
- use lithium\util\String;
- use lithium\util\Inflector;
- use InvalidArgumentException;
- use lithium\core\ConfigException;
- use lithium\core\NetworkException;
- use lithium\data\model\QueryException;
- /**
- * The `Database` class provides the base-level abstraction for SQL-oriented relational databases.
- * It handles all aspects of abstraction, including formatting for basic query types and SQL
- * fragments (i.e. for joins), converting `Query` objects to SQL, and various other functionality
- * which is shared across multiple relational databases.
- *
- * @see lithium\data\model\Query
- */
- abstract class Database extends \lithium\data\Source {
- /**
- * @var PDO
- */
- public $connection;
- /**
- * The supported column types and their default values
- *
- * @var array
- */
- protected $_columns = array(
- 'string' => array('length' => 255)
- );
- /**
- * Strings used to render the given statement
- *
- * @see lithium\data\source\Database::renderCommand()
- * @var array
- */
- protected $_strings = array(
- 'create' => "INSERT INTO {:source} ({:fields}) VALUES ({:values});{:comment}",
- 'update' => "UPDATE {:source} SET {:fields} {:conditions};{:comment}",
- 'delete' => "DELETE {:flags} FROM {:source} {:conditions};{:comment}",
- 'schema' => "CREATE TABLE {:source} (\n{:columns}{:indexes});{:comment}",
- 'join' => "{:type} JOIN {:source} {:alias} {:constraints}"
- );
- /**
- * Classes used by `Database`.
- *
- * @var array
- */
- protected $_classes = array(
- 'entity' => 'lithium\data\entity\Record',
- 'set' => 'lithium\data\collection\RecordSet',
- 'relationship' => 'lithium\data\model\Relationship',
- 'result' => 'lithium\data\source\database\adapter\pdo\Result',
- 'schema' => 'lithium\data\Schema',
- 'query' => 'lithium\data\model\Query'
- );
- /**
- * List of SQL operators, paired with handling options.
- *
- * @var array
- */
- protected $_operators = array(
- '=' => array('multiple' => 'IN'),
- '<' => array(),
- '>' => array(),
- '<=' => array(),
- '>=' => array(),
- '!=' => array('multiple' => 'NOT IN'),
- '<>' => array('multiple' => 'NOT IN'),
- 'between' => array('format' => 'BETWEEN ? AND ?'),
- 'BETWEEN' => array('format' => 'BETWEEN ? AND ?'),
- 'like' => array(),
- 'LIKE' => array(),
- 'not like' => array(),
- 'NOT LIKE' => array()
- );
- protected $_constraintTypes = array(
- 'AND' => true,
- 'and' => true,
- 'OR' => true,
- 'or' => true
- );
- /**
- * A pair of opening/closing quote characters used for quoting identifiers in SQL queries.
- *
- * @var array
- */
- protected $_quotes = array();
- /**
- * Array of named callable objects representing different strategies for performing specific
- * types of queries.
- *
- * @var array
- */
- protected $_strategies = array();
- /**
- * Getter/Setter for the connection's encoding
- * Abstract. Must be defined by child class.
- *
- * @param mixed $encoding
- * @return mixed.
- */
- abstract public function encoding($encoding = null);
- /**
- * Return the last errors produced by a the execution of a query.
- * Abstract. Must be defined by child class.
- *
- */
- abstract public function error();
- /**
- * Execute a given query
- * Abstract. Must be defined by child class.
- *
- * @see lithium\data\source\Database::renderCommand()
- * @param string $sql The sql string to execute
- * @return resource
- */
- abstract protected function _execute($sql);
- /**
- * Get the last insert id from the database.
- * Abstract. Must be defined by child class.
- *
- * @param $query lithium\data\model\Query $context The given query.
- * @return void
- */
- abstract protected function _insertId($query);
- /**
- * Creates the database object and set default values for it.
- *
- * Options defined:
- * - 'database' _string_ Name of the database to use. Defaults to `null`.
- * - 'host' _string_ Name/address of server to connect to. Defaults to 'localhost'.
- * - 'login' _string_ Username to use when connecting to server. Defaults to 'root'.
- * - 'password' _string_ Password to use when connecting to server. Defaults to `''`.
- * - 'persistent' _boolean_ If true a persistent connection will be attempted, provided the
- * adapter supports it. Defaults to `true`.
- *
- * @param $config array Array of configuration options.
- * @return Database object.
- */
- public function __construct(array $config = array()) {
- $defaults = array(
- 'persistent' => true,
- 'host' => 'localhost',
- 'login' => 'root',
- 'password' => '',
- 'database' => null,
- 'encoding' => null,
- 'dsn' => null,
- 'options' => array()
- );
- $this->_strings += array(
- 'read' => 'SELECT {:fields} FROM {:source} {:alias} {:joins} {:conditions} {:group} ' .
- '{:having} {:order} {:limit};{:comment}'
- );
- $this->_strategies += array(
- 'joined' => function($self, $model, $context) {
- $with = $context->with();
- $strategy = function($me, $model, $tree, $path, $from, &$deps) use ($self, $context, $with) {
- foreach ($tree as $name => $childs) {
- if (!$rel = $model::relations($name)) {
- throw new QueryException("Model relationship `{$name}` not found.");
- }
- $constraints = array();
- $alias = $name;
- $relPath = $path ? $path . '.' . $name : $name;
- if (isset($with[$relPath])) {
- list($unallowed, $allowed) = Set::slice($with[$relPath], array(
- 'alias',
- 'constraints'
- ));
- if ($unallowed) {
- $message = "Only `'alias'` and `'constraints'` are allowed in ";
- $message .= "`'with'` using the `'joined'` strategy.";
- throw new QueryException($message);
- }
- extract($with[$relPath]);
- }
- $to = $context->alias($alias, $relPath);
- $deps[$to] = $deps[$from];
- $deps[$to][] = $from;
- if ($context->relationships($relPath) === null) {
- $context->relationships($relPath, array(
- 'type' => $rel->type(),
- 'model' => $rel->to(),
- 'fieldName' => $rel->fieldName(),
- 'alias' => $to
- ));
- $self->join($context, $rel, $from, $to, $constraints);
- }
- if (!empty($childs)) {
- $me($me, $rel->to(), $childs, $relPath, $to, $deps);
- }
- }
- };
- $tree = Set::expand(Set::normalize(array_keys($with)));
- $alias = $context->alias();
- $deps = array($alias => array());
- $strategy($strategy, $model, $tree, '', $alias, $deps);
- $models = $context->models();
- foreach ($context->fields() as $field) {
- if (!is_string($field)) {
- continue;
- }
- list($alias, $field) = $self->invokeMethod('_splitFieldname', array($field));
- $alias = $alias ?: $field;
- if ($alias && isset($models[$alias])) {
- foreach ($deps[$alias] as $depAlias) {
- $depModel = $models[$depAlias];
- $context->fields(array($depAlias => (array) $depModel::meta('key')));
- }
- }
- }
- },
- 'nested' => function($self, $model, $context) {
- throw new QueryException("This strategy is not yet implemented.");
- }
- );
- parent::__construct($config + $defaults);
- }
- public function connect() {
- $this->_isConnected = false;
- $config = $this->_config;
- if (!$config['database']) {
- throw new ConfigException('No Database configured');
- }
- if (!$config['dsn']) {
- throw new ConfigException('No DSN setup for DB Connection');
- }
- $dsn = $config['dsn'];
- $options = $config['options'] + array(
- PDO::ATTR_PERSISTENT => $config['persistent'],
- PDO::ATTR_ERRMODE => PDO::ERRMODE_EXCEPTION
- );
- try {
- $this->connection = new PDO($dsn, $config['login'], $config['password'], $options);
- } catch (PDOException $e) {
- preg_match('/SQLSTATE\[(.+?)\]/', $e->getMessage(), $code);
- $code = $code[1] ?: 0;
- switch (true) {
- case $code === 'HY000' || substr($code, 0, 2) === '08':
- $msg = "Unable to connect to host `{$config['host']}`.";
- throw new NetworkException($msg, null, $e);
- case in_array($code, array('28000', '42000')):
- $msg = "Host connected, but could not access database `{$config['database']}`.";
- throw new ConfigException($msg, null, $e);
- }
- throw new ConfigException("An unknown configuration error has occured.", null, $e);
- }
- $this->_isConnected = true;
- if ($this->_config['encoding']) {
- $this->encoding($this->_config['encoding']);
- }
- return $this->_isConnected;
- }
- /**
- * Disconnects the adapter from the database.
- *
- * @return boolean Returns `true` on success, else `false`.
- */
- public function disconnect() {
- if ($this->_isConnected) {
- unset($this->connection);
- $this->_isConnected = false;
- }
- return true;
- }
- /**
- * Field name handler to ensure proper escaping.
- *
- * @param string $name Field or identifier name.
- * @return string Returns `$name` quoted according to the rules and quote characters of the
- * database adapter subclass.
- */
- public function name($name) {
- $open = reset($this->_quotes);
- $close = next($this->_quotes);
- list($first, $second) = $this->_splitFieldname($name);
- if ($first) {
- return "{$open}{$first}{$close}.{$open}{$second}{$close}";
- }
- return preg_match('/^[a-z0-9_-]+$/i', $name) ? "{$open}{$name}{$close}" : $name;
- }
- /**
- * Return the alias and the field name from an identifier name.
- *
- * @param string $field Field name or identifier name.
- * @return array Returns an array with the alias (or `null` if not applicable) as first value
- * and the field name as second value.
- */
- protected function _splitFieldname($field) {
- if (is_string($field)) {
- if (preg_match('/^[a-z0-9_-]+\.([a-z0-9_-]+|\*)$/i', $field)) {
- return explode('.', $field, 2);
- }
- }
- return array(null, $field);
- }
- /**
- * Return the field name from a conditions key.
- *
- * @param string $field Field or identifier name.
- * @return string Returns the field name without the table alias, if applicable.
- * @todo Eventually, this should be refactored and moved to the Query or Schema
- * class. Also, by handling field resolution in this way we are not handling
- * cases where query conditions use the same field name in multiple tables.
- * e.g. Foos.bar and Bars.bar will both return bar.
- */
- protected function _fieldName($field) {
- if (is_string($field)) {
- if (preg_match('/^[a-z0-9_-]+\.[a-z0-9_-]+$/i', $field)) {
- list($first, $second) = explode('.', $field, 2);
- return $second;
- }
- }
- return $field;
- }
- /**
- * Converts a given value into the proper type based on a given schema definition.
- *
- * @see lithium\data\source\Database::schema()
- * @param mixed $value The value to be converted. Arrays will be recursively converted.
- * @param array $schema Formatted array from `lithium\data\source\Database::schema()`
- * @return mixed value with converted type
- */
- public function value($value, array $schema = array()) {
- if (is_array($value)) {
- foreach ($value as $key => $val) {
- $value[$key] = $this->value($val, isset($schema[$key]) ? $schema[$key] : $schema);
- }
- return $value;
- }
- if (is_object($value) && isset($value->scalar)) {
- return $value->scalar;
- }
- if ($value === null) {
- return 'NULL';
- }
- switch ($type = isset($schema['type']) ? $schema['type'] : $this->_introspectType($value)) {
- case 'boolean':
- case 'float':
- case 'integer':
- return $this->_cast($type, $value);
- default:
- return $this->connection->quote($this->_cast($type, $value));
- }
- }
- /**
- * Inserts a new record into the database based on a the `Query`. The record is updated
- * with the id of the insert.
- *
- * @see lithium\util\String::insert()
- * @param object $query An SQL query string, or `lithium\data\model\Query` object instance.
- * @param array $options If $query is a string, $options contains an array of bind values to be
- * escaped, quoted, and inserted into `$query` using `String::insert()`.
- * @return boolean Returns `true` if the query succeeded, otherwise `false`.
- * @filter
- */
- public function create($query, array $options = array()) {
- return $this->_filter(__METHOD__, compact('query', 'options'), function($self, $params) {
- $query = $params['query'];
- $model = $entity = $object = $id = null;
- if (is_object($query)) {
- $object = $query;
- $model = $query->model();
- $params = $query->export($self);
- $entity =& $query->entity();
- $query = $self->renderCommand('create', $params, $query);
- } else {
- $query = String::insert($query, $self->value($params['options']));
- }
- if (!$self->invokeMethod('_execute', array($query))) {
- return false;
- }
- if ($entity) {
- if (($model) && !$model::key($entity)) {
- $id = $self->invokeMethod('_insertId', array($object));
- }
- $entity->sync($id);
- }
- return true;
- });
- }
- /**
- * Reads records from a database using a `lithium\data\model\Query` object or raw SQL string.
- *
- * @param string|object $query `lithium\data\model\Query` object or SQL string.
- * @param array $options If `$query` is a raw string, contains the values that will be escaped
- * and quoted. Other options:
- * - `'return'` _string_: switch return between `'array'`, `'item'`, or
- * `'resource'` _string_: Defaults to `'item'`.
- * @return mixed Determined by `$options['return']`.
- * @filter
- */
- public function read($query, array $options = array()) {
- $defaults = array(
- 'return' => is_string($query) ? 'array' : 'item',
- 'schema' => null,
- 'quotes' => $this->_quotes
- );
- $options += $defaults;
- return $this->_filter(__METHOD__, compact('query', 'options'), function($self, $params) {
- $query = $params['query'];
- $args = $params['options'];
- $return = $args['return'];
- unset($args['return']);
- $model = is_object($query) ? $query->model() : null;
- if (is_string($query)) {
- $sql = String::insert($query, $self->value($args));
- } else {
- if (!$data = $self->invokeMethod('_queryExport', array($query))) {
- return false;
- }
- $sql = $self->renderCommand($data['type'], $data);
- }
- $result = $self->invokeMethod('_execute', array($sql));
- switch ($return) {
- case 'resource':
- return $result;
- case 'array':
- $columns = $args['schema'] ?: $self->schema($query, $result);
- if (!is_array(reset($columns))) {
- $columns = array('' => $columns);
- }
- $i = 0;
- $records = array();
- foreach ($result as $data) {
- $offset = 0;
- $records[$i] = array();
- foreach ($columns as $path => $cols) {
- $len = count($cols);
- $values = array_combine($cols, array_slice($data, $offset, $len));
- if ($path) {
- $records[$i][$path] = $values;
- } else {
- $records[$i] += $values;
- }
- $offset += $len;
- }
- $i++;
- }
- return Set::expand($records);
- case 'item':
- return $self->item($query->model(), array(), compact('query', 'result') + array(
- 'class' => 'set'
- ));
- }
- });
- }
- /**
- * Helper which export the query export
- *
- * @param object $query The query object
- * @return array The export array
- */
- protected function &_queryExport($query) {
- $data = $query->export($this);
- if ($query->limit() && ($model = $query->model())) {
- foreach ($query->relationships() as $relation) {
- if ($relation['type'] === 'hasMany') {
- $name = $model::meta('name');
- $key = $model::key();
- $fields = $data['fields'];
- $fieldname = $this->name("{$name}.{$key}");
- $data['fields'] = "DISTINCT({$fieldname}) AS _ID_";
- $sql = $this->renderCommand('read', $data);
- $result = $this->_execute($sql);
- $ids = array();
- while ($row = $result->next()) {
- $ids[] = $row[0];
- }
- if (!$ids) {
- $return = null;
- return $return;
- }
- $data['fields'] = $fields;
- $data['limit'] = '';
- $data['conditions'] = $this->conditions(array("{$name}.{$key}" => $ids), $query);
- return $data;
- }
- }
- }
- return $data;
- }
- /**
- * Updates a record in the database based on the given `Query`.
- *
- * @param object $query A `lithium\data\model\Query` object
- * @param array $options none
- * @return boolean
- * @filter
- */
- public function update($query, array $options = array()) {
- return $this->_filter(__METHOD__, compact('query', 'options'), function($self, $params) {
- $query = $params['query'];
- $params = $query->export($self);
- $sql = $self->renderCommand('update', $params, $query);
- $result = (boolean) $self->invokeMethod('_execute', array($sql));
- if ($result && is_object($query) && $query->entity()) {
- $query->entity()->sync();
- }
- return $result;
- });
- }
- /**
- * Deletes a record in the database based on the given `Query`.
- *
- * @param object $query An SQL string, or `lithium\data\model\Query` object instance.
- * @param array $options If `$query` is a string, `$options` is the array of quoted/escaped
- * parameter values to be inserted into the query.
- * @return boolean Returns `true` on successful query execution (not necessarily if records are
- * deleted), otherwise `false`.
- * @filter
- */
- public function delete($query, array $options = array()) {
- return $this->_filter(__METHOD__, compact('query', 'options'), function($self, $params) {
- $query = $params['query'];
- $isObject = is_object($query);
- if ($isObject) {
- $sql = $self->renderCommand('delete', $query->export($self), $query);
- } else {
- $sql = String::insert($query, $self->value($params['options']));
- }
- $result = (boolean) $self->invokeMethod('_execute', array($sql));
- if ($result && $isObject && $query->entity()) {
- $query->entity()->sync(null, array(), array('dematerialize' => true));
- }
- return $result;
- });
- }
- /**
- * Executes calculation-related queries, such as those required for `count` and other
- * aggregates.
- *
- * @param string $type Only accepts `count`.
- * @param mixed $query The query to be executed.
- * @param array $options Optional arguments for the `read()` query that will be executed
- * to obtain the calculation result.
- * @return integer Result of the calculation.
- */
- public function calculation($type, $query, array $options = array()) {
- $query->calculate($type);
- switch ($type) {
- case 'count':
- if (strpos($fields = $this->fields($query->fields(), $query), ',') !== false) {
- $fields = "*";
- }
- $query->fields("COUNT({$fields}) as count", true);
- $query->map(array($query->alias() => array('count')));
- list($record) = $this->read($query, $options)->data();
- return isset($record['count']) ? intval($record['count']) : null;
- }
- }
- /**
- * Defines or modifies the default settings of a relationship between two models.
- *
- * @param string $class the primary model of the relationship
- * @param string $type the type of the relationship (hasMany, hasOne, belongsTo)
- * @param string $name the name of the relationship
- * @param array $config relationship options
- * @return array Returns an array containing the configuration for a model relationship.
- */
- public function relationship($class, $type, $name, array $config = array()) {
- $field = Inflector::underscore(Inflector::singularize($name));
- $key = "{$field}_id";
- $primary = $class::meta('key');
- if (is_array($primary)) {
- $key = array_combine($primary, $primary);
- } elseif ($type === 'hasMany' || $type === 'hasOne') {
- if ($type === 'hasMany') {
- $field = Inflector::pluralize($field);
- }
- $secondary = Inflector::underscore(Inflector::singularize($class::meta('name')));
- $key = array($primary => "{$secondary}_id");
- }
- $from = $class;
- $fieldName = $field;
- $config += compact('type', 'name', 'key', 'from', 'fieldName');
- return $this->_instance('relationship', $config);
- }
- /**
- * Determines the set of methods to be used when exporting query values.
- *
- * @return array
- */
- public function methods() {
- $result = parent::methods();
- $key = array_search('schema', $result);
- unset($result[$key]);
- return $result;
- }
- /**
- * Returns a given `type` statement for the given data, rendered from `Database::$_strings`.
- *
- * @param string $type One of `'create'`, `'read'`, `'update'`, `'delete'` or `'join'`.
- * @param string $data The data to replace in the string.
- * @param string $context
- * @return string
- */
- public function renderCommand($type, $data = null, $context = null) {
- if (is_object($type)) {
- $context = $type;
- $data = $context->export($this);
- $type = $context->type();
- }
- if (!isset($this->_strings[$type])) {
- throw new InvalidArgumentException("Invalid query type `{$type}`.");
- }
- $template = $this->_strings[$type];
- $data = array_filter($data);
- return trim(String::insert($template, $data, array('clean' => true)));
- }
- /**
- * Builds an array of keyed on the fully-namespaced `Model` with array of fields as values
- * for the given `Query`
- *
- * @param data\model\Query $query A Query instance.
- * @param object $resource
- * @param object $context
- */
- public function schema($query, $resource = null, $context = null) {
- if (is_object($query)) {
- $query->applyStrategy($this);
- return $this->_schema($query, $this->_fields($query->fields(), $query));
- }
- $result = array();
- $count = $resource->resource()->columnCount();
- for ($i = 0; $i < $count; $i++) {
- $meta = $resource->resource()->getColumnMeta($i);
- $result[] = $meta['name'];
- }
- return $result;
- }
- /**
- * Helper method for `data\model\Database::shema()`
- *
- * @param data\model\Query $query A Query instance.
- * @param array $fields Array of formatted fields.
- */
- protected function _schema($query, $fields = null) {
- $model = $query->model();
- $paths = $query->paths($this);
- $models = $query->models($this);
- $alias = $query->alias();
- $result = array();
- if (!$model) {
- foreach ($fields as $field => $value) {
- if (is_array($value)) {
- $result[$field] = array_keys($value);
- } else {
- $result[''][] = $field;
- }
- }
- return $result;
- }
- if (!$fields) {
- foreach ($paths as $alias => $relation) {
- $model = $models[$alias];
- $result[$relation] = $model::schema()->names();
- }
- return $result;
- }
- $unalias = function ($value) {
- if (is_object($value) && isset($value->scalar)) {
- $value = $value->scalar;
- }
- $aliasing = preg_split("/\s+as\s+/i", $value);
- return isset($aliasing[1]) ? $aliasing[1] : $value;
- };
- if (isset($fields[0])) {
- $raw = array_map($unalias, $fields[0]);
- unset($fields[0]);
- }
- $fields = isset($fields[$alias]) ? array($alias => $fields[$alias]) + $fields : $fields;
- foreach ($fields as $field => $value) {
- if (is_array($value)) {
- if (isset($value['*'])) {
- $relModel = $models[$field];
- $result[$paths[$field]] = $relModel::schema()->names();
- } else {
- $result[$paths[$field]] = array_map($unalias, array_keys($value));
- }
- }
- }
- if (isset($raw)) {
- $result[''] = isset($result['']) ? array_merge($raw, $result['']) : $raw;
- }
- return $result;
- }
- /**
- * Returns a string of formatted conditions to be inserted into the query statement. If the
- * query conditions are defined as an array, key pairs are converted to SQL strings.
- *
- * Conversion rules are as follows:
- *
- * - If `$key` is numeric and `$value` is a string, `$value` is treated as a literal SQL
- * fragment and returned.
- *
- * @param string|array $conditions The conditions for this query.
- * @param object $context The current `lithium\data\model\Query` instance.
- * @param array $options
- * - `prepend` _boolean_: Whether the return string should be prepended with the
- * `WHERE` keyword.
- * @return string Returns the `WHERE` clause of an SQL query.
- */
- public function conditions($conditions, $context, array $options = array()) {
- $defaults = array('prepend' => 'WHERE');
- $options += $defaults;
- return $this->_conditions($conditions, $context, $options);
- }
- /**
- * Returns a string of formatted havings to be inserted into the query statement. If the
- * query havings are defined as an array, key pairs are converted to SQL strings.
- *
- * Conversion rules are as follows:
- *
- * - If `$key` is numeric and `$value` is a string, `$value` is treated as a literal SQL
- * fragment and returned.
- *
- * @param string|array $conditions The havings for this query.
- * @param object $context The current `lithium\data\model\Query` instance.
- * @param array $options
- * - `prepend` _boolean_: Whether the return string should be prepended with the
- * `HAVING` keyword.
- * @return string Returns the `HAVING` clause of an SQL query.
- */
- public function having($conditions, $context, array $options = array()) {
- $defaults = array('prepend' => 'HAVING');
- $options += $defaults;
- return $this->_conditions($conditions, $context, $options);
- }
- /**
- * Returns a string of formatted conditions to be inserted into the query statement. If the
- * query conditions are defined as an array, key pairs are converted to SQL strings.
- *
- * Conversion rules are as follows:
- *
- * - If `$key` is numeric and `$value` is a string, `$value` is treated as a literal SQL
- * fragment and returned.
- *
- * @param string|array $conditions The conditions for this query.
- * @param object $context The current `lithium\data\model\Query` instance.
- * @param array $options
- * - `prepend` mixed: The string to prepend or false for no prepending
- * @return string Returns an SQL conditions clause.
- */
- protected function _conditions($conditions, $context, array $options = array()) {
- $defaults = array('prepend' => false);
- $ops = $this->_operators;
- $options += $defaults;
- switch (true) {
- case empty($conditions):
- return '';
- case is_string($conditions):
- return $options['prepend'] ? $options['prepend'] . " {$conditions}" : $conditions;
- case !is_array($conditions):
- return '';
- }
- $result = array();
- foreach ($conditions as $key => $value) {
- $return = $this->_processConditions($key, $value, $context);
- if ($return) {
- $result[] = $return;
- }
- }
- $result = join(" AND ", $result);
- return ($options['prepend'] && $result) ? $options['prepend'] . " {$result}" : $result;
- }
- protected function _processConditions($key, $value, $context, $schema = null, $glue = 'AND') {
- $constraintTypes =& $this->_constraintTypes;
- $model = $context->model();
- $models = $context->models();
- list($first, $second) = $this->_splitFieldname($key);
- if ($first && isset($models[$first]) && $class = $models[$first]) {
- $schema = $class::schema();
- } elseif ($model) {
- $schema = $model::schema();
- }
- $fieldMeta = $schema ? (array) $schema->fields($second) : array();
- switch (true) {
- case (is_numeric($key) && is_string($value)):
- return $value;
- case is_object($value) && isset($value->scalar):
- if (is_numeric($key)) {
- return $this->value($value);
- }
- case is_scalar($value) || is_null($value):
- if ($context && ($context->type() === 'read') && ($alias = $context->alias())) {
- $key = $this->_aliasing($key, $alias);
- }
- if (isset($value)) {
- return $this->name($key) . ' = ' . $this->value($value, $fieldMeta);
- }
- return $this->name($key) . " IS NULL";
- case is_numeric($key) && is_array($value):
- $result = array();
- foreach ($value as $cKey => $cValue) {
- $result[] = $this->_processConditions($cKey, $cValue, $context, $schema, $glue);
- }
- return '(' . implode(' ' . $glue . ' ', $result) . ')';
- case (is_string($key) && is_object($value)):
- $value = trim(rtrim($this->renderCommand($value), ';'));
- return "{$this->name($key)} IN ({$value})";
- case is_array($value) && isset($constraintTypes[strtoupper($key)]):
- $result = array();
- $glue = strtoupper($key);
- foreach ($value as $cKey => $cValue) {
- $result[] = $this->_processConditions($cKey, $cValue, $context, $schema, $glue);
- }
- return '(' . implode(' ' . $glue . ' ', $result) . ')';
- case (is_string($key) && is_array($value) && isset($this->_operators[key($value)])):
- foreach ($value as $op => $val) {
- $result[] = $this->_operator($key, array($op => $val), $fieldMeta);
- }
- return '(' . implode(' ' . $glue . ' ', $result) . ')';
- case is_array($value):
- if (!is_numeric($op = key($value))) {
- throw new QueryException("Unsupported operator `{$op}`.");
- }
- $value = join(', ', $this->value($value, $fieldMeta));
- return "{$this->name($key)} IN ({$value})";
- }
- }
- /**
- * Returns a string of formatted fields to be inserted into the query statement.
- *
- * @param array $fields Array of fields.
- * @param object $context Generally a `data\model\Query` instance.
- * @return string A SQL formatted string
- */
- public function fields($fields, $context) {
- $type = $context->type();
- $schema = $context->schema()->fields();
- $alias = $context->alias();
- if (!is_array($fields)) {
- return $this->_fieldsReturn($type, $context, $fields, $schema);
- }
- $context->applyStrategy($this);
- $fields = $this->_fields($fields ? : $context->fields(), $context);
- $context->map($this->_schema($context, $fields));
- $toMerge = array();
- if (isset($fields[0])) {
- foreach ($fields[0] as $val) {
- $toMerge[] = (is_object($val) && isset($val->scalar)) ? $val->scalar : $val;
- }
- unset($fields[0]);
- }
- $fields = isset($fields[$alias]) ? array($alias => $fields[$alias]) + $fields : $fields;
- foreach ($fields as $field => $value) {
- if (is_array($value)) {
- if (isset($value['*'])) {
- $toMerge[] = $this->name($field) . '.*';
- continue;
- }
- foreach ($value as $fieldname => $mode) {
- $toMerge[] = $this->_fieldsQuote($field, $fieldname);
- }
- }
- }
- return $this->_fieldsReturn($type, $context, $toMerge, $schema);
- }
- /**
- * Helper for `Database::fields()` && `Database::schema()`.
- * Reformat fields to be alias based.
- *
- * @param array $fields Array of fields.
- * @param object $context Generally a `data\model\Query` instance.
- * @return array Reformatted fields
- */
- protected function _fields($fields, $context) {
- $alias = $context->alias();
- $models = $context->models($this);
- $list = array();
- foreach ($fields as $key => $field) {
- if (!is_string($field)) {
- if (isset($models[$key])) {
- $field = array_fill_keys($field, true);
- $list[$key] = isset($list[$key]) ? array_merge($list[$key], $field) : $field;
- } else {
- $list[0][] = is_array($field) ? reset($field) : $field;
- }
- continue;
- }
- if (preg_match('/^([a-z0-9_-]+|\*)$/i', $field)) {
- isset($models[$field]) ? $list[$field]['*'] = true : $list[$alias][$field] = true;
- } elseif (preg_match('/^([a-z0-9_-]+)\.(.*)$/i', $field, $matches)) {
- $list[$matches[1]][$matches[2]] = true;
- } else {
- $list[0][] = $field;
- }
- }
- return $list;
- }
- protected function _fieldsQuote($alias, $field) {
- $open = $this->_quotes[0];
- $close = $this->_quotes[1];
- $aliasing = preg_split("/\s+as\s+/i", $field);
- if (isset($aliasing[1])) {
- list($aliasname, $fieldname) = $this->_splitFieldname($aliasing[0]);
- $alias = $aliasname ? : $alias;
- return "{$open}{$alias}{$close}.{$open}{$fieldname}{$close} as {$aliasing[1]}";
- } elseif ($alias) {
- return "{$open}{$alias}{$close}.{$open}{$field}{$close}";
- } else {
- return "{$open}{$field}{$close}";
- }
- }
- protected function _fieldsReturn($type, $context, $fields, $schema) {
- if ($type === 'create' || $type === 'update') {
- $data = $context->data();
- if (isset($data['data']) && is_array($data['data']) && count($data) === 1) {
- $data = $data['data'];
- }
- if ($fields && is_array($fields) && is_int(key($fields))) {
- $data = array_intersect_key($data, array_combine($fields, $fields));
- }
- $method = "_{$type}Fields";
- return $this->{$method}($data, $schema, $context);
- }
- return empty($fields) ? '*' : join(', ', $fields);
- }
- /**
- * Returns a LIMIT statement from the given limit and the offset of the context object.
- *
- * @param integer $limit An
- * @param object $context The `lithium\data\model\Query` object
- * @return string
- */
- public function limit($limit, $context) {
- if (!$limit) {
- return;
- }
- if ($offset = $context->offset() ?: '') {
- $offset = " OFFSET {$offset}";
- }
- return "LIMIT {$limit}{$offset}";
- }
- /**
- * Returns a join statement for given array of query objects
- *
- * @param object|array $joins A single or array of `lithium\data\model\Query` objects
- * @param object $context The parent `lithium\data\model\Query` object
- * @return string
- */
- public function joins(array $joins, $context) {
- $result = null;
- foreach ($joins as $key => $join) {
- if ($result) {
- $result .= ' ';
- }
- $join = is_array($join) ? $this->_instance('query', $join) : $join;
- $options['keys'] = array('source', 'alias', 'constraints');
- $result .= $this->renderCommand('join', $join->export($this, $options));
- }
- return $result;
- }
- /**
- * Returns a string of formatted constraints to be inserted into the query statement. If the
- * query constraints are defined as an array, key pairs are converted to SQL strings.
- *
- * Conversion rules are as follows:
- *
- * - If `$key` is numeric and `$value` is a string, `$value` is treated as a literal SQL
- * fragment and returned.
- *
- * @param string|array $constraints The constraints for a `ON` clause.
- * @param object $context The current `lithium\data\model\Query` instance.
- * @param array $options
- * - `prepend` _boolean_: Whether the return string should be prepended with the
- * `ON` keyword.
- * @return string Returns the `ON` clause of an SQL query.
- */
- public function constraints($constraints, $context, array $options = array()) {
- $defaults = array('prepend' => 'ON');
- $options += $defaults;
- if (is_array($constraints)) {
- $constraints = $this->_constraints($constraints);
- }
- return $this->_conditions($constraints, $context, $options);
- }
- /**
- * Auto escape string value to a field name value
- *
- * @param array $constraints The constraints array
- * @return array The escaped constraints array
- */
- protected function _constraints(array $constraints) {
- foreach ($constraints as &$value) {
- if (is_string($value)) {
- $value = (object) $this->name($value);
- } elseif (is_array($value)) {
- $value = $this->_constraints($value);
- }
- }
- return $constraints;
- }
- /**
- * Return formatted clause for order.
- *
- * @param mixed $order The `order` clause to be formatted
- * @param object $context
- * @return mixed Formatted `order` clause.
- */
- public function order($order, $context) {
- $direction = 'ASC';
- $model = $context->model();
- if (is_string($order)) {
- if (!$model::schema($order)) {
- $match = '/\s+(A|DE)SC/i';
- return "ORDER BY {$order}" . (preg_match($match, $order) ? '' : " {$direction}");
- }
- $order = array($order => $direction);
- }
- if (!is_array($order)) {
- return;
- }
- $result = array();
- foreach ($order as $column => $dir) {
- if (is_int($column)) {
- $column = $dir;
- $dir = $direction;
- }
- $dir = in_array($dir, array('ASC', 'asc', 'DESC', 'desc')) ? $dir : $direction;
- if (!$model) {
- $result[] = "{$column} {$dir}";
- continue;
- }
- if ($field = $model::schema($column)) {
- $name = $this->name($model::meta('name')) . '.' . $this->name($column);
- $result[] = "{$name} {$dir}";
- continue;
- }
- $result[] = "{$column} {$dir}";
- }
- $order = join(', ', $result);
- return "ORDER BY {$order}";
- }
- public function group($group, $context = null) {
- if (!$group) {
- return null;
- }
- return 'GROUP BY ' . join(', ', (array) $group);
- }
- /**
- * Adds formatting to SQL comments before they're embedded in queries.
- *
- * @param string $comment
- * @return string
- */
- public function comment($comment) {
- return $comment ? "/* {$comment} */" : null;
- }
- public function alias($alias, $context) {
- if (!$alias && ($model = $context->model())) {
- $alias = $model::meta('name');
- }
- return $alias ? "AS " . $this->name($alias) : null;
- }
- public function cast($entity, array $data, array $options = array()) {
- return $data;
- }
- /**
- * Cast a value according to a column type.
- *
- * @param string $type Name of the column type
- * @param string $value Value to cast
- *
- * @return mixed Casted value
- *
- */
- protected function _cast($type, $value) {
- if (is_object($value) || $value === null) {
- return $value;
- }
- if ($type === 'boolean') {
- return $this->_toNativeBoolean($value);
- }
- if (!isset($this->_columns[$type]) || !isset($this->_columns[$type]['formatter'])) {
- return $value;
- }
- $column = $this->_columns[$type];
- switch ($column['formatter']) {
- case 'date':
- return $column['formatter']($column['format'], strtotime($value));
- default:
- return $column['formatter']($value);
- }
- }
- protected function _createFields($data, $schema, $context) {
- $fields = $values = array();
- foreach ($data as $field => $value) {
- $fields[] = $this->name($field);
- $values[] = $this->value($value, isset($schema[$field]) ? $schema[$field] : array());
- }
- $fields = join(', ', $fields);
- $values = join(', ', $values);
- return compact('fields', 'values');
- }
- protected function _updateFields($data, $schema, $context) {
- $fields = array();
- foreach ($data as $field => $value) {
- $schema += array($field => array('default' => null));
- $fields[] = $this->name($field) . ' = ' . $this->value($value, $schema[$field]);
- }
- return join(', ', $fields);
- }
- /**
- * Handles conversion of SQL operator keys to SQL statements.
- *
- * @param string $key Key in a conditions array. Usually a field name.
- * @param mixed $value An SQL operator or comparison value.
- * @param array $schema An array defining the schema of the field used in the criteria.
- * @param array $options
- * @return string Returns an SQL string representing part of a `WHERE` clause of a query.
- */
- protected function _operator($key, $value, array $schema = array(), array $options = array()) {
- $defaults = array('boolean' => 'AND');
- $options += $defaults;
- list($op, $value) = each($value);
- $config = $this->_operators[$op];
- $key = $this->name($key);
- $values = array();
- if (!is_object($value)) {
- foreach ((array) $value as $val) {
- $values[] = $this->value($val, $schema);
- }
- } elseif (isset($value->scalar)) {
- return "{$key} {$op} {$value->scalar}";
- }
- switch (true) {
- case (isset($config['format'])):
- return $key . ' ' . String::insert($config['format'], $values);
- case (is_object($value) && isset($config['multiple'])):
- $op = $config['multiple'];
- $value = trim(rtrim($this->renderCommand($value), ';'));
- return "{$key} {$op} ({$value})";
- case (count($values) > 1 && isset($config['multiple'])):
- $op = $config['multiple'];
- $values = join(', ', $values);
- return "{$key} {$op} ({$values})";
- case (count($values) > 1):
- return join(" {$options['boolean']} ", array_map(
- function($v) use ($key, $op) { return "{$key} {$op} {$v}"; }, $values
- ));
- }
- return "{$key} {$op} {$values[0]}";
- }
- /**
- * Returns a fully-qualified table name (i.e. with prefix), quoted.
- *
- * @param string $entity A table name or fully-namespaced model class name.
- * @param array $options Available options:
- * - `'quoted'` _boolean_: Indicates whether the name should be quoted.
- * @return string Returns a quoted table name.
- */
- protected function _entityName($entity, array $options = array()) {
- $defaults = array('quoted' => false);
- $options += $defaults;
- if (class_exists($entity, false) && method_exists($entity, 'meta')) {
- $entity = $entity::meta('source');
- }
- return $options['quoted'] ? $this->name($entity) : $entity;
- }
- /**
- * Attempts to automatically determine the column type of a value. Used by the `value()` method
- * of various database adapters to determine how to prepare a value if the schema is not
- * specified.
- *
- * @param mixed $value The value to be prepared for an SQL query.
- * @return string Returns the name of the column type which `$value` most likely belongs to.
- */
- protected function _introspectType($value) {
- switch (true) {
- case (is_bool($value)):
- return 'boolean';
- case (is_float($value) || preg_match('/^\d+\.\d+$/', $value)):
- return 'float';
- case (is_int($value) || preg_match('/^\d+$/', $value)):
- return 'integer';
- case (is_string($value) && strlen($value) <= $this->_columns['string']['length']):
- return 'string';
- default:
- return 'text';
- }
- }
- /**
- * Casts a value which is being written or compared to a boolean-type database column.
- *
- * @param mixed $value A value of unknown type to be cast to boolean. Numeric values not equal
- * to zero evaluate to `true`, otherwise `false`. String values equal to `'true'`,
- * `'t'` or `'T'` evaluate to `true`, all others to `false`. In all other cases,
- * uses PHP's default casting.
- * @return boolean Returns a boolean representation of `$value`, based on the comparison rules
- * specified above. Database adapters may override this method if boolean type coercion
- * is required and falls outside the rules defined.
- */
- protected function _toBoolean($value) {
- if (is_bool($value)) {
- return $value;
- }
- if (is_int($value) || is_float($value)) {
- return ($value !== 0);
- }
- if (is_string($value)) {
- return ($value === 't' || $value === 'T' || $value === 'true');
- }
- return (boolean) $value;
- }
- protected function _toNativeBoolean($value) {
- return $value ? 1 : 0;
- }
- /**
- * Throw a `QueryException` error
- *
- * @param string $sql The offending SQL string
- * @filter
- */
- protected function _error($sql){
- $params = compact('sql');
- return $this->_filter(__METHOD__, $params, function($self, $params) {
- $sql = $params['sql'];
- list($code, $error) = $self->error();
- throw new QueryException("{$sql}: {$error}", $code);
- });
- }
- /**
- * Applying a strategy to a `lithium\data\model\Query` object
- *
- * @param array $options The option array
- * @param object $context A query object to configure
- */
- public function applyStrategy($options, $context) {
- $options += array('strategy' => 'joined');
- if (!$model = $context->model()) {
- throw new ConfigException('The `\'with\'` option need a valid `\'model\'` option.');
- }
- $strategy = $options['strategy'];
- if (isset($this->_strategies[$strategy])) {
- $strategy = $this->_strategies[$strategy];
- $strategy($this, $model, $context);
- } else {
- throw new QueryException("Undefined query strategy `{$strategy}`.");
- }
- }
- /**
- * Set a query's join according a Relationship.
- *
- * @param object $context A Query instance
- * @param object $rel A Relationship instance
- * @param string $fromAlias Set a specific alias for the `'from'` `Model`.
- * @param string $toAlias Set a specific alias for `'to'` `Model`.
- * @param mixed $constraints If `$constraints` is an array, it will be merged to defaults
- * constraints. If `$constraints` is an object, defaults won't be merged.
- */
- public function join($context, $rel, $fromAlias = null, $toAlias = null, $constraints = array()) {
- $model = $rel->to();
- if ($fromAlias === null) {
- $from = $rel->from();
- $fromAlias = $context->alias();
- }
- if ($toAlias === null) {
- $toAlias = $context->alias(null, $rel->name());
- }
- if (!is_object($constraints)) {
- $constraints = $this->on($rel, $fromAlias, $toAlias, $constraints);
- } else {
- $constraints = (array) $constraints;
- }
- $context->joins($toAlias, compact('constraints', 'model') + array(
- 'type' => 'LEFT',
- 'alias' => $toAlias
- ));
- }
- /**
- * Helper which add an alias basename to a field name if necessary
- *
- * @param string $name The field name.
- * @param string $alias The alias name
- * @param array $map An array of `'modelname' => 'aliasname'` mapping
- * @return string
- */
- protected function _aliasing($name, $alias, $map = array()) {
- list($first, $second) = $this->_splitFieldname($name);
- if (!$first && preg_match('/^[a-z0-9_-]+$/i', $second)) {
- return $alias . "." . $second;
- } elseif (isset($map[$first])) {
- return $map[$first] . "." . $second;
- }
- return $name;
- }
- /**
- * Build the `ON` constraints from a `Relationship` instance
- *
- * @param object $rel A Relationship instance
- * @param string $fromAlias Set a specific alias for the `'from'` `Model`.
- * @param string $toAlias Set a specific alias for `'to'` `Model`.
- * @param array $constraints Array of additionnal $constraints.
- * @return array A constraints array.
- */
- public function on($rel, $aliasFrom = null, $aliasTo = null, $constraints = array()) {
- $model = $rel->from();
- $aliasFrom = $aliasFrom ?: $model::meta('name');
- $aliasTo = $aliasTo ?: $rel->name();
- $keyConstraints = array();
- foreach ($rel->key() as $from => $to) {
- $keyConstraints["{$aliasFrom}.{$from}"] = "{$aliasTo}.{$to}";
- }
- $mapAlias = array($model::meta('name') => $aliasFrom, $rel->name() => $aliasTo);
- $relConstraints = $this->_on((array) $rel->constraints(), $aliasFrom, $aliasTo, $mapAlias);
- $constraints = $this->_on($constraints, $aliasFrom, $aliasTo, array());
- return $constraints + $relConstraints + $keyConstraints;
- }
- protected function _on(array $constraints, $aliasFrom, $aliasTo, $mapAlias = array()) {
- $result = array();
- foreach ($constraints as $key => $value) {
- if (
- !is_numeric($key) &&
- !isset($this->_constraintTypes[$key]) &&
- !isset($this->_operators[$key])
- ) {
- $key = $this->_aliasing($key, $aliasFrom, $mapAlias);
- }
- if (is_string($value)) {
- $result[$key] = $this->_aliasing($value, $aliasTo, $mapAlias);
- } elseif (is_array($value)) {
- $result[$key] = $this->_on($value, $aliasFrom, $aliasTo, $mapAlias);
- } else {
- $result[$key] = $value;
- }
- }
- return $result;
- }
- }
- ?>
|