1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390 |
- <?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 (is_array($ids)) {
- foreach ($ids as $key => $id) {
- if (empty($id) || ((string) $id !== (string) (int) $id)) {
- unset($ids[$key]);
- }
- }
- }
-
-
-
- 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 { }
|