You are here

class Utility in Search API Solr 8

Same name and namespace in other branches
  1. 8.3 src/Utility/Utility.php \Drupal\search_api_solr\Utility\Utility
  2. 8.2 src/Utility/Utility.php \Drupal\search_api_solr\Utility\Utility
  3. 4.x src/Utility/Utility.php \Drupal\search_api_solr\Utility\Utility

Utility functions specific to solr.

Hierarchy

  • class \Drupal\search_api_solr\Utility\Utility

Expanded class hierarchy of Utility

3 files declare their use of Utility
SearchApiSolrBackend.php in src/Plugin/search_api/backend/SearchApiSolrBackend.php
SolrConfigForm.php in src/Form/SolrConfigForm.php
UtilitiesTest.php in tests/src/Kernel/UtilitiesTest.php

File

src/Utility/Utility.php, line 11

Namespace

Drupal\search_api_solr\Utility
View source
class Utility {

  /**
   * Retrieves Solr-specific data for available data types.
   *
   * Returns the data type information for both the default Search API data
   * types and custom data types defined by hook_search_api_data_type_info().
   * Names for default data types are not included, since they are not relevant
   * to the Solr service class.
   *
   * We're adding some extra Solr field information for the default search api
   * data types (as well as on behalf of a couple contrib field types). The
   * extra information we're adding is documented in
   * search_api_solr_hook_search_api_data_type_info(). You can use the same
   * additional keys in hook_search_api_data_type_info() to support custom
   * dynamic fields in your indexes with Solr.
   *
   * @param string|null $type
   *   (optional) A specific type for which the information should be returned.
   *   Defaults to returning all information.
   *
   * @return array|null
   *   If $type was given, information about that type or NULL if it is unknown.
   *   Otherwise, an array of all types. The format in both cases is the same as
   *   for search_api_get_data_type_info().
   *
   * @see search_api_get_data_type_info()
   * @see search_api_solr_hook_search_api_data_type_info()
   */
  public static function getDataTypeInfo($type = NULL) {
    $types =& drupal_static(__FUNCTION__);
    if (!isset($types)) {

      // Grab the stock search_api data types.

      /** @var \Drupal\search_api\DataType\DataTypePluginManager $data_type_service */
      $data_type_service = \Drupal::service('plugin.manager.search_api.data_type');
      $types = $data_type_service
        ->getDefinitions();

      // Add our extras for the default search api fields.
      $types = NestedArray::mergeDeep($types, array(
        'text' => array(
          'prefix' => 't',
        ),
        'string' => array(
          'prefix' => 's',
        ),
        'integer' => array(
          // Use trie field for better sorting.
          'prefix' => 'it',
        ),
        'decimal' => array(
          // Use trie field for better sorting.
          'prefix' => 'ft',
        ),
        'date' => array(
          'prefix' => 'd',
        ),
        'duration' => array(
          // Use trie field for better sorting.
          'prefix' => 'it',
        ),
        'boolean' => array(
          'prefix' => 'b',
        ),
        'uri' => array(
          'prefix' => 's',
        ),
      ));

      // Extra data type info.
      $extra_types_info = array(
        // Provided by Search API Location module.
        'location' => array(
          'prefix' => 'loc',
        ),
        // @todo Who provides that type?
        'geohash' => array(
          'prefix' => 'geo',
        ),
        // Provided by Search API Location module.
        'rpt' => [
          'prefix' => 'rpt',
        ],
      );

      // For the extra types, only add our extra info if it's already been
      // defined.
      foreach ($extra_types_info as $key => $info) {
        if (array_key_exists($key, $types)) {

          // Merge our extras into the data type info.
          $types[$key] += $info;
        }
      }
    }

    // Return the info.
    if (isset($type)) {
      return isset($types[$type]) ? $types[$type] : NULL;
    }
    return $types;
  }

  /**
   * Returns a unique hash for the current site.
   *
   * This is used to identify Solr documents from different sites within a
   * single Solr server.
   *
   * @return string
   *   A unique site hash, containing only alphanumeric characters.
   */
  public static function getSiteHash() {

    // Copied from apachesolr_site_hash().
    if (!($hash = \Drupal::config('search_api_solr.settings')
      ->get('site_hash'))) {
      global $base_url;
      $hash = substr(base_convert(sha1(uniqid($base_url, TRUE)), 16, 36), 0, 6);
      \Drupal::configFactory()
        ->getEditable('search_api_solr.settings')
        ->set('site_hash', $hash)
        ->save();
    }
    return $hash;
  }

  /**
   * Retrieves a list of all config files of a server's Solr backend.
   *
   * @param \Drupal\search_api\ServerInterface $server
   *   The Solr server whose files should be retrieved.
   * @param string $dir_name
   *   (optional) The directory that should be searched for files. Defaults to
   *   the root config directory.
   *
   * @return array
   *   An associative array of all config files in the given directory. The keys
   *   are the file names, values are arrays with information about the file.
   *   The files are returned in alphabetical order and breadth-first.
   *
   * @throws \Drupal\search_api\SearchApiException
   *   If a problem occurred while retrieving the files.
   */
  public static function getServerFiles(ServerInterface $server, $dir_name = NULL) {

    /** @var \Drupal\search_api_solr\SolrBackendInterface $backend */
    $backend = $server
      ->getBackend();
    $response = $backend
      ->getSolrConnector()
      ->getFile($dir_name);

    // Search for directories and recursively merge directory files.
    $files_data = json_decode($response
      ->getBody(), TRUE);
    $files_list = $files_data['files'];
    $dir_length = strlen($dir_name) + 1;
    $result = array(
      '' => array(),
    );
    foreach ($files_list as $file_name => $file_info) {

      // Annoyingly, Solr 4.7 changed the way the admin/file handler returns
      // the file names when listing directory contents: the returned name is
      // now only the base name, not the complete path from the config root
      // directory. We therefore have to check for this case.
      if ($dir_name && substr($file_name, 0, $dir_length) !== "{$dir_name}/") {
        $file_name = "{$dir_name}/" . $file_name;
      }
      if (empty($file_info['directory'])) {
        $result[''][$file_name] = $file_info;
      }
      else {
        $result[$file_name] = static::getServerFiles($server, $file_name);
      }
    }
    ksort($result);
    ksort($result['']);
    return array_reduce($result, 'array_merge', array());
  }

  /**
   * Changes highlighting tags from our custom, HTML-safe ones to HTML.
   *
   * @param string|array $snippet
   *   The snippet(s) to format.
   *
   * @return string|array
   *   The snippet(s), properly formatted as HTML.
   */
  public static function formatHighlighting($snippet) {
    return preg_replace('#\\[(/?)HIGHLIGHT\\]#', '<$1strong>', $snippet);
  }

  /**
   * Encodes field names to avoid characters that are not supported by solr.
   *
   * Solr doesn't restrict the characters used to build field names. But using
   * non java identifiers within a field name can cause different kind of
   * trouble when running queries. Java identifiers are only consist of
   * letters, digits, '$' and '_'. See
   * https://issues.apache.org/jira/browse/SOLR-3996 and
   * http://docs.oracle.com/cd/E19798-01/821-1841/bnbuk/index.html
   * For full compatibility the '$' has to be avoided, too. And there're more
   * restrictions regarding the field name itself. See
   * https://cwiki.apache.org/confluence/display/solr/Defining+Fields
   * "Field names should consist of alphanumeric or underscore characters only
   * and not start with a digit ... Names with both leading and trailing
   * underscores (e.g. _version_) are reserved." Field names starting with
   * digits or underscores are already avoided by our schema. The same is true
   * for the names of field types. See
   * https://cwiki.apache.org/confluence/display/solr/Field+Type+Definitions+and+Properties
   * "It is strongly recommended that names consist of alphanumeric or
   * underscore characters only and not start with a digit. This is not
   * currently strictly enforced."
   *
   * This function therefore encodes all forbidden characters in their
   * hexadecimal equivalent encapsulated by a leading sequence of '_X' and a
   * termination character '_'. Example:
   * "tm_entity:node/body" becomes "tm_entity_X3a_node_X2f_body".
   *
   * As a consequence the sequence '_X' itself needs to be encoded if it occurs
   * within a field name. Example: "last_XMas" becomes "last_X5f58_Mas".
   *
   * @param string $field_name
   *   The field name.
   *
   * @return string
   *   The encoded field name.
   */
  public static function encodeSolrName($field_name) {
    return preg_replace_callback('/([^\\da-zA-Z_]|_X)/u', function ($matches) {
      return '_X' . bin2hex($matches[1]) . '_';
    }, $field_name);
  }

  /**
   * Decodes solr field names.
   *
   * This function therefore decodes all forbidden characters from their
   * hexadecimal equivalent encapsulated by a leading sequence of '_X' and a
   * termination character '_'. Example:
   * "tm_entity_X3a_node_X2f_body" becomes "tm_entity:node/body".
   *
   * @see encodeSolrDynamicFieldName() for details.
   *
   * @param string $field_name
   *   Encoded field name.
   *
   * @return string
   *   The decoded field name
   */
  public static function decodeSolrName($field_name) {
    return preg_replace_callback('/_X([\\dabcdef]+?)_/', function ($matches) {
      return hex2bin($matches[1]);
    }, $field_name);
  }

}

Members

Namesort descending Modifiers Type Description Overrides
Utility::decodeSolrName public static function Decodes solr field names.
Utility::encodeSolrName public static function Encodes field names to avoid characters that are not supported by solr.
Utility::formatHighlighting public static function Changes highlighting tags from our custom, HTML-safe ones to HTML.
Utility::getDataTypeInfo public static function Retrieves Solr-specific data for available data types.
Utility::getServerFiles public static function Retrieves a list of all config files of a server's Solr backend.
Utility::getSiteHash public static function Returns a unique hash for the current site.