function views_data_export_plugin_query_default_batched::execute in Views data export 7

Same name and namespace in other branches

1. 6.3 plugins/views_data_export_plugin_display_export.inc \views_data_export_plugin_query_default_batched::execute()
2. 7.4 plugins/views_data_export_plugin_display_export.inc \views_data_export_plugin_query_default_batched::execute()
3. 7.3 plugins/views_data_export_plugin_display_export.inc \views_data_export_plugin_query_default_batched::execute()

Executes the query and fills the associated view object with according values.

Values to set: $view->result, $view->total_rows, $view->execute_time, $view->current_page.

Overrides views_plugin_query_default::execute

File

plugins/views_data_export_plugin_display_export.inc, line 689

Contains the bulk export display plugin.

Class

views_data_export_plugin_query_default_batched

Code

function execute(&$view) {
  $display_handler =& $view->display_handler;
  $external = FALSE;

  // Whether this query will run against an external database.
  $query = $view->build_info['query'];
  $count_query = $view->build_info['count_query'];
  $query
    ->addMetaData('view', $view);
  $count_query
    ->addMetaData('view', $view);
  if (empty($this->options['disable_sql_rewrite'])) {
    $base_table_data = views_fetch_data($this->base_table);
    if (isset($base_table_data['table']['base']['access query tag'])) {
      $access_tag = $base_table_data['table']['base']['access query tag'];
      $query
        ->addTag($access_tag);
      $count_query
        ->addTag($access_tag);
    }
  }
  $items = array();
  if ($query) {
    $additional_arguments = module_invoke_all('views_query_substitutions', $view);

    // Build the count query. db_select::countQuery does not work with groupby.
    $count_query = $count_query
      ->countQuery();

    //  $count_query->distinct = FALSE;
    // Add additional arguments as a fake condition.
    // XXX: this doesn't work... because PDO mandates that all bound arguments
    // are used on the query. TODO: Find a better way to do this.
    if (!empty($additional_arguments)) {

      // $query->where('1 = 1', $additional_arguments);
      // $count_query->where('1 = 1', $additional_arguments);
    }

    // Detect an external database.
    if (isset($view->base_database)) {
      db_set_active($view->base_database);
      $external = TRUE;
    }
    $start = microtime(TRUE);
    if ($this->pager
      ->use_count_query() || !empty($view->get_total_rows)) {
      $this->pager
        ->execute_count_query($count_query);
    }

    // Let the pager modify the query to add limits.
    $this->pager
      ->pre_execute($query);
    if (!empty($this->limit) || !empty($this->offset)) {

      // We can't have an offset without a limit, so provide a very large limit instead.
      $limit = intval(!empty($this->limit) ? $this->limit : 999999);
      $offset = intval(!empty($this->offset) ? $this->offset : 0);
      $query
        ->range($offset, $limit);
    }
    try {

      // The $query is final and ready to go, we are going to redirect it to
      // become an insert into our table, sneaky!
      // Our query will look like:
      // CREATE TABLE {idx} SELECT @row := @row + 1 AS weight_alias, cl.* FROM
      // (-query-) AS cl, (SELECT @row := 0) AS r
      // We do some magic to get the row count.
      $display_handler->batched_execution_state->sandbox['weight_field_alias'] = $display_handler
        ->_weight_alias_create($view);
      $display_handler->batched_execution_state->sandbox['field_aliases'] = $display_handler
        ->field_aliases_create($view);
      $select_aliases = array();
      foreach ($display_handler->batched_execution_state->sandbox['field_aliases'] as $hash => $alias) {
        $select_aliases[] = "cl.{$alias} AS {$hash}";
      }

      // TODO: this could probably be replaced with a query extender and new query type.
      $args = $query
        ->getArguments();
      $insert_query = 'CREATE TABLE {' . $display_handler
        ->index_tablename() . '} SELECT @row := @row + 1 AS ' . $display_handler->batched_execution_state->sandbox['weight_field_alias'] . ', ' . implode(', ', $select_aliases) . ' FROM (' . (string) $query . ') AS cl, (SELECT @row := 0) AS r';
      db_query($insert_query, $args);
      $view->result = array();
      $this->pager
        ->post_execute($view->result);
      if ($this->pager
        ->use_pager()) {
        $view->total_rows = $this->pager
          ->get_total_items();
      }

      // Now create an index for the weight field, otherwise the queries on the
      // index will take a long time to execute.
      db_add_unique_key($display_handler
        ->index_tablename(), $display_handler->batched_execution_state->sandbox['weight_field_alias'], array(
        $display_handler->batched_execution_state->sandbox['weight_field_alias'],
      ));
    } catch (Exception $e) {
      $view->result = array();
      debug('Exception: ' . $e
        ->getMessage());
    }
    if ($external) {
      db_set_active();
    }
  }
  $view->execute_time = microtime(TRUE) - $start;
}