123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374 |
- <?php
- interface DrupalEntityControllerInterface {
-
- public function resetCache(array $ids = NULL);
-
- public function load($ids = array(), $conditions = array());
- }
- class DrupalDefaultEntityController implements DrupalEntityControllerInterface {
-
- protected $entityCache;
-
- protected $entityType;
-
- protected $entityInfo;
-
- protected $hookLoadArguments;
-
- protected $idKey;
-
- protected $revisionKey;
-
- protected $revisionTable;
-
- protected $cache;
-
- public function __construct($entityType) {
- $this->entityType = $entityType;
- $this->entityInfo = entity_get_info($entityType);
- $this->entityCache = array();
- $this->hookLoadArguments = array();
- $this->idKey = $this->entityInfo['entity keys']['id'];
-
- if (!empty($this->entityInfo['entity keys']['revision'])) {
- $this->revisionKey = $this->entityInfo['entity keys']['revision'];
- $this->revisionTable = $this->entityInfo['revision table'];
- }
- else {
- $this->revisionKey = FALSE;
- }
-
- $this->cache = !empty($this->entityInfo['static cache']);
- }
-
- public function resetCache(array $ids = NULL) {
- if (isset($ids)) {
- foreach ($ids as $id) {
- unset($this->entityCache[$id]);
- }
- }
- else {
- $this->entityCache = array();
- }
- }
-
- public function load($ids = array(), $conditions = array()) {
- $entities = array();
-
-
- if ($this->revisionKey && isset($conditions[$this->revisionKey])) {
- $revision_id = $conditions[$this->revisionKey];
- unset($conditions[$this->revisionKey]);
- }
- else {
- $revision_id = FALSE;
- }
-
-
-
-
-
- $passed_ids = !empty($ids) ? array_flip($ids) : FALSE;
-
-
- if ($this->cache && !$revision_id) {
- $entities += $this->cacheGet($ids, $conditions);
-
- if ($passed_ids) {
- $ids = array_keys(array_diff_key($passed_ids, $entities));
- }
- }
-
-
-
- if ($ids === FALSE || $ids || $revision_id || ($conditions && !$passed_ids)) {
-
- $query = $this->buildQuery($ids, $conditions, $revision_id);
- $queried_entities = $query
- ->execute()
- ->fetchAllAssoc($this->idKey);
- }
-
-
-
- if (!empty($queried_entities)) {
- $this->attachLoad($queried_entities, $revision_id);
- $entities += $queried_entities;
- }
- if ($this->cache) {
-
- if (!empty($queried_entities) && !$revision_id) {
- $this->cacheSet($queried_entities);
- }
- }
-
-
- if ($passed_ids) {
-
- $passed_ids = array_intersect_key($passed_ids, $entities);
- foreach ($entities as $entity) {
- $passed_ids[$entity->{$this->idKey}] = $entity;
- }
- $entities = $passed_ids;
- }
- return $entities;
- }
-
- protected function buildQuery($ids, $conditions = array(), $revision_id = FALSE) {
- $query = db_select($this->entityInfo['base table'], 'base');
- $query->addTag($this->entityType . '_load_multiple');
- if ($revision_id) {
- $query->join($this->revisionTable, 'revision', "revision.{$this->idKey} = base.{$this->idKey} AND revision.{$this->revisionKey} = :revisionId", array(':revisionId' => $revision_id));
- }
- elseif ($this->revisionKey) {
- $query->join($this->revisionTable, 'revision', "revision.{$this->revisionKey} = base.{$this->revisionKey}");
- }
-
- $entity_fields = $this->entityInfo['schema_fields_sql']['base table'];
- if ($this->revisionKey) {
-
- $entity_revision_fields = drupal_map_assoc($this->entityInfo['schema_fields_sql']['revision table']);
-
- unset($entity_revision_fields[$this->idKey]);
-
-
- $entity_field_keys = array_flip($entity_fields);
- foreach ($entity_revision_fields as $key => $name) {
- if (isset($entity_field_keys[$name])) {
- unset($entity_fields[$entity_field_keys[$name]]);
- }
- }
- $query->fields('revision', $entity_revision_fields);
- }
- $query->fields('base', $entity_fields);
- if ($ids) {
- $query->condition("base.{$this->idKey}", $ids, 'IN');
- }
- if ($conditions) {
- foreach ($conditions as $field => $value) {
- $query->condition('base.' . $field, $value);
- }
- }
- return $query;
- }
-
- protected function attachLoad(&$queried_entities, $revision_id = FALSE) {
-
- if ($this->entityInfo['fieldable']) {
- if ($revision_id) {
- field_attach_load_revision($this->entityType, $queried_entities);
- }
- else {
- field_attach_load($this->entityType, $queried_entities);
- }
- }
-
- foreach (module_implements('entity_load') as $module) {
- $function = $module . '_entity_load';
- $function($queried_entities, $this->entityType);
- }
-
-
-
- $args = array_merge(array($queried_entities), $this->hookLoadArguments);
- foreach (module_implements($this->entityInfo['load hook']) as $module) {
- call_user_func_array($module . '_' . $this->entityInfo['load hook'], $args);
- }
- }
-
- protected function cacheGet($ids, $conditions = array()) {
- $entities = array();
-
- if (!empty($this->entityCache)) {
- if ($ids) {
- $entities += array_intersect_key($this->entityCache, array_flip($ids));
- }
-
-
- elseif ($conditions) {
- $entities = $this->entityCache;
- }
- }
-
-
- if ($conditions) {
- foreach ($entities as $entity) {
-
-
-
- foreach ($conditions as $property_name => $condition) {
- if (is_array($condition)) {
-
-
-
- if (!in_array($entity->{$property_name}, $condition)) {
- unset($entities[$entity->{$this->idKey}]);
- continue 2;
- }
- }
- elseif ($condition != $entity->{$property_name}) {
- unset($entities[$entity->{$this->idKey}]);
- continue 2;
- }
- }
- }
- }
- return $entities;
- }
-
- protected function cacheSet($entities) {
- $this->entityCache += $entities;
- }
- }
- class EntityFieldQueryException extends Exception {}
- class EntityFieldQuery {
-
- const RETURN_ALL = NULL;
-
- public $altered = FALSE;
-
- public $entityConditions = array();
-
- public $fieldConditions = array();
-
- public $fieldMetaConditions = array();
-
- public $propertyConditions = array();
-
- public $order = array();
-
- public $range = array();
-
- public $pager = array();
-
- public $deleted = FALSE;
-
- public $fields = array();
-
- public $count = FALSE;
-
- public $age = FIELD_LOAD_CURRENT;
-
- public $tags = array();
-
- public $metaData = array();
-
- public $orderedResults = array();
-
- public $executeCallback = '';
-
- public function entityCondition($name, $value, $operator = NULL) {
-
-
- if ($operator == '!=') {
- $operator = '<>';
- }
- $this->entityConditions[$name] = array(
- 'value' => $value,
- 'operator' => $operator,
- );
- return $this;
- }
-
- public function fieldCondition($field, $column = NULL, $value = NULL, $operator = NULL, $delta_group = NULL, $language_group = NULL) {
- return $this->addFieldCondition($this->fieldConditions, $field, $column, $value, $operator, $delta_group, $language_group);
- }
-
- public function fieldLanguageCondition($field, $value = NULL, $operator = NULL, $delta_group = NULL, $language_group = NULL) {
- return $this->addFieldCondition($this->fieldMetaConditions, $field, 'language', $value, $operator, $delta_group, $language_group);
- }
-
- public function fieldDeltaCondition($field, $value = NULL, $operator = NULL, $delta_group = NULL, $language_group = NULL) {
- return $this->addFieldCondition($this->fieldMetaConditions, $field, 'delta', $value, $operator, $delta_group, $language_group);
- }
-
- protected function addFieldCondition(&$conditions, $field, $column = NULL, $value = NULL, $operator = NULL, $delta_group = NULL, $language_group = NULL) {
-
-
- if ($operator == '!=') {
- $operator = '<>';
- }
- if (is_scalar($field)) {
- $field_definition = field_info_field($field);
- if (empty($field_definition)) {
- throw new EntityFieldQueryException(t('Unknown field: @field_name', array('@field_name' => $field)));
- }
- $field = $field_definition;
- }
-
- $index = count($this->fields);
- $this->fields[$index] = $field;
- if (isset($column)) {
- $conditions[$index] = array(
- 'field' => $field,
- 'column' => $column,
- 'value' => $value,
- 'operator' => $operator,
- 'delta_group' => $delta_group,
- 'language_group' => $language_group,
- );
- }
- return $this;
- }
-
- public function propertyCondition($column, $value, $operator = NULL) {
-
-
- if ($operator == '!=') {
- $operator = '<>';
- }
- $this->propertyConditions[] = array(
- 'column' => $column,
- 'value' => $value,
- 'operator' => $operator,
- );
- return $this;
- }
-
- public function entityOrderBy($name, $direction = 'ASC') {
- $this->order[] = array(
- 'type' => 'entity',
- 'specifier' => $name,
- 'direction' => $direction,
- );
- return $this;
- }
-
- public function fieldOrderBy($field, $column, $direction = 'ASC') {
- if (is_scalar($field)) {
- $field_definition = field_info_field($field);
- if (empty($field_definition)) {
- throw new EntityFieldQueryException(t('Unknown field: @field_name', array('@field_name' => $field)));
- }
- $field = $field_definition;
- }
-
- $index = count($this->fields);
- $this->fields[$index] = $field;
- $this->order[] = array(
- 'type' => 'field',
- 'specifier' => array(
- 'field' => $field,
- 'index' => $index,
- 'column' => $column,
- ),
- 'direction' => $direction,
- );
- return $this;
- }
-
- public function propertyOrderBy($column, $direction = 'ASC') {
- $this->order[] = array(
- 'type' => 'property',
- 'specifier' => $column,
- 'direction' => $direction,
- );
- return $this;
- }
-
- public function count() {
- $this->count = TRUE;
- return $this;
- }
-
- public function range($start = NULL, $length = NULL) {
- $this->range = array(
- 'start' => $start,
- 'length' => $length,
- );
- return $this;
- }
-
- public function pager($limit = 10, $element = NULL) {
- if (!isset($element)) {
- $element = PagerDefault::$maxElement++;
- }
- elseif ($element >= PagerDefault::$maxElement) {
- PagerDefault::$maxElement = $element + 1;
- }
- $this->pager = array(
- 'limit' => $limit,
- 'element' => $element,
- );
- return $this;
- }
-
- public function tableSort(&$headers) {
-
- foreach ($headers as $key =>$header) {
- if (is_array($header) && isset($header['specifier'])) {
- $headers[$key]['field'] = '';
- }
- }
- $order = tablesort_get_order($headers);
- $direction = tablesort_get_sort($headers);
- foreach ($headers as $header) {
- if (is_array($header) && ($header['data'] == $order['name'])) {
- if ($header['type'] == 'field') {
- $this->fieldOrderBy($header['specifier']['field'], $header['specifier']['column'], $direction);
- }
- else {
- $header['direction'] = $direction;
- $this->order[] = $header;
- }
- }
- }
- return $this;
- }
-
- public function deleted($deleted = TRUE) {
- $this->deleted = $deleted;
- return $this;
- }
-
- public function age($age) {
- $this->age = $age;
- return $this;
- }
-
- public function addTag($tag) {
- $this->tags[$tag] = $tag;
- return $this;
- }
-
- public function addMetaData($key, $object) {
- $this->metaData[$key] = $object;
- return $this;
- }
-
- public function execute() {
-
- drupal_alter('entity_query', $this);
- $this->altered = TRUE;
-
- $this->initializePager();
-
- $result = call_user_func($this->queryCallback(), $this);
- return $result;
- }
-
- public function queryCallback() {
-
-
- if (function_exists($this->executeCallback)) {
- return $this->executeCallback;
- }
-
-
- if (empty($this->fields)) {
- return array($this, 'propertyQuery');
- }
-
- foreach ($this->fields as $field) {
- if (!isset($storage)) {
- $storage = $field['storage']['module'];
- }
- elseif ($storage != $field['storage']['module']) {
- throw new EntityFieldQueryException(t("Can't handle more than one field storage engine"));
- }
- }
- if ($storage) {
-
- return $storage . '_field_storage_query';
- }
- else {
- throw new EntityFieldQueryException(t("Field storage engine not found."));
- }
- }
-
- protected function propertyQuery() {
- if (empty($this->entityConditions['entity_type'])) {
- throw new EntityFieldQueryException(t('For this query an entity type must be specified.'));
- }
- $entity_type = $this->entityConditions['entity_type']['value'];
- $entity_info = entity_get_info($entity_type);
- if (empty($entity_info['base table'])) {
- throw new EntityFieldQueryException(t('Entity %entity has no base table.', array('%entity' => $entity_type)));
- }
- $base_table = $entity_info['base table'];
- $base_table_schema = drupal_get_schema($base_table);
- $select_query = db_select($base_table);
- $select_query->addExpression(':entity_type', 'entity_type', array(':entity_type' => $entity_type));
-
- foreach ($this->propertyConditions as $property_condition) {
- $this->addCondition($select_query, $base_table . '.' . $property_condition['column'], $property_condition);
- }
-
-
- $sql_field = $entity_info['entity keys']['id'];
- $id_map['entity_id'] = $sql_field;
- $select_query->addField($base_table, $sql_field, 'entity_id');
- if (isset($this->entityConditions['entity_id'])) {
- $this->addCondition($select_query, $base_table . '.' . $sql_field, $this->entityConditions['entity_id']);
- }
-
- if (!empty($entity_info['entity keys']['revision'])) {
- $sql_field = $entity_info['entity keys']['revision'];
- $select_query->addField($base_table, $sql_field, 'revision_id');
- if (isset($this->entityConditions['revision_id'])) {
- $this->addCondition($select_query, $base_table . '.' . $sql_field, $this->entityConditions['revision_id']);
- }
- }
- else {
- $sql_field = 'revision_id';
- $select_query->addExpression('NULL', 'revision_id');
- }
- $id_map['revision_id'] = $sql_field;
-
- if (!empty($entity_info['entity keys']['bundle'])) {
- $sql_field = $entity_info['entity keys']['bundle'];
- $having = FALSE;
- if (!empty($base_table_schema['fields'][$sql_field])) {
- $select_query->addField($base_table, $sql_field, 'bundle');
- }
- }
- else {
- $sql_field = 'bundle';
- $select_query->addExpression(':bundle', 'bundle', array(':bundle' => $entity_type));
- $having = TRUE;
- }
- $id_map['bundle'] = $sql_field;
- if (isset($this->entityConditions['bundle'])) {
- if (!empty($entity_info['entity keys']['bundle'])) {
- $this->addCondition($select_query, $base_table . '.' . $sql_field, $this->entityConditions['bundle'], $having);
- }
- else {
-
- $select_query->where('1 = 0');
- }
- }
-
- foreach ($this->order as $order) {
- if ($order['type'] == 'entity') {
- $key = $order['specifier'];
- if (!isset($id_map[$key])) {
- throw new EntityFieldQueryException(t('Do not know how to order on @key for @entity_type', array('@key' => $key, '@entity_type' => $entity_type)));
- }
- $select_query->orderBy($id_map[$key], $order['direction']);
- }
- elseif ($order['type'] == 'property') {
- $select_query->orderBy($base_table . '.' . $order['specifier'], $order['direction']);
- }
- }
- return $this->finishQuery($select_query);
- }
-
- function initializePager() {
- if ($this->pager && !empty($this->pager['limit']) && !$this->count) {
- $page = pager_find_page($this->pager['element']);
- $count_query = clone $this;
- $this->pager['total'] = $count_query->count()->execute();
- $this->pager['start'] = $page * $this->pager['limit'];
- pager_default_initialize($this->pager['total'], $this->pager['limit'], $this->pager['element']);
- $this->range($this->pager['start'], $this->pager['limit']);
- }
- }
-
- function finishQuery($select_query, $id_key = 'entity_id') {
- foreach ($this->tags as $tag) {
- $select_query->addTag($tag);
- }
- foreach ($this->metaData as $key => $object) {
- $select_query->addMetaData($key, $object);
- }
- $select_query->addMetaData('entity_field_query', $this);
- if ($this->range) {
- $select_query->range($this->range['start'], $this->range['length']);
- }
- if ($this->count) {
- return $select_query->countQuery()->execute()->fetchField();
- }
- $return = array();
- foreach ($select_query->execute() as $partial_entity) {
- $bundle = isset($partial_entity->bundle) ? $partial_entity->bundle : NULL;
- $entity = entity_create_stub_entity($partial_entity->entity_type, array($partial_entity->entity_id, $partial_entity->revision_id, $bundle));
- $return[$partial_entity->entity_type][$partial_entity->$id_key] = $entity;
- $this->ordered_results[] = $partial_entity;
- }
- return $return;
- }
-
- public function addCondition(SelectQuery $select_query, $sql_field, $condition, $having = FALSE) {
- $method = $having ? 'havingCondition' : 'condition';
- $like_prefix = '';
- switch ($condition['operator']) {
- case 'CONTAINS':
- $like_prefix = '%';
- case 'STARTS_WITH':
- $select_query->$method($sql_field, $like_prefix . db_like($condition['value']) . '%', 'LIKE');
- break;
- default:
- $select_query->$method($sql_field, $condition['value'], $condition['operator']);
- }
- }
- }
- class EntityMalformedException extends Exception { }
|