opus\elastic\components\Command

Complexity

Total Complexity

16

Size/Duplication

Total Lines	116
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	6

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	bulk()	0	9	3
B	formatBulkData()	0	40	5
A	getCachedMapping()	0	19	5
A	getMapping()	0	8	3

<?php
/**
 *
 * @author Mihkel Viilveer <mihkel@opus.ee>
 * @date 25.08.2014
 */

namespace opus\elastic\components;
use yii\base\ErrorException;
use yii\caching\Cache;
use yii\helpers\Json;

/**
 * Class Command
 *
 * @author Mihkel Viilveer <mihkel@opus.ee>
 * @package common\components\elasticsearch
 */
class Command extends \yii\elasticsearch\Command
{
    /**
     * @param $index
     * @param $type
     * @param $data
     * @param array $options
     * @return mixed
     * @throws ErrorException
     */
    public function bulk($index, $type, $data, $options = [])
    {
        if (empty($data)) {
            $body = '{}';
        } else {
            $body = is_array($data) ? $this->formatBulkData($index, $type, $data) : $data;
        }
        return $this->db->post([$index, $type, '_bulk'], $options, $body);
    }

    /**
     * @param $index
     * @param $type
     * @param $data
     * @return string
     * @throws ErrorException
     */
    private function formatBulkData($index, $type, $data)
    {
        $formattedData = [];

        foreach ($data as $operation => $objects) {
            foreach ($objects as $object) {
                switch ($operation) {
                    case 'INDEX' :

There must be no space before the colon in a CASE statement

                        $formattedData[] = Json::encode(
                            [
                                'index' => [
                                    '_index' => $index,
                                    '_type' => $type,
                                    '_id' => $object['id']
                                ]
                            ]
                        );
                        $formattedData[] = Json::encode($object);
                        break;

                    case 'DELETE' :

There must be no space before the colon in a CASE statement

                        $formattedData[] = Json::encode(
                            [
                                'delete' => [
                                    '_index' => $index,
                                    '_type' => $type,
                                    '_id' => $object['id']
                                ]
                            ]
                        );
                        break;
                    default:
                        throw new ErrorException(
                            'Unknown operation ' . $operation
                        );
                }
            }
        };
        return implode("\n", $formattedData) . "\n";
    }

    /**
     * getMapping with caching support
     *
     * @param string $index
     * @param string $type
     * @param string $data
     * @throws \yii\base\InvalidConfigException
     * @return array
     * @see http://www.elasticsearch.org/guide/en/elasticsearch/reference/current/indices-get-mapping.html
     */
    public function getCachedMapping($index = '_all', $type = '_all', $data = 'properties')
    {
        $command = \Yii::$app->elasticsearch;
        /* @var $cache Cache */
        $cache = is_string($command->cache) ? \Yii::$app->get($command->cache, false) : $command->cache;
        if ($cache instanceof Cache) {
            $key = sprintf('%s_%s_%s', $index, $type, $data);
            if (($mapping = $cache->get($key)) === false) {
                $mapping = $this->getMapping($index, $type, $data);
                if ($mapping !== null) {
                    $cache->set($key, $mapping);
                }
            }
        }
        else {
            $mapping = $this->getMapping($index, $type, $data);
        }
        return $mapping;
    }

    /**
     * Overridden to return exact type mapping if possible
     *
     * @param string $index
     * @param string $type
     * @param string $data Type of data to be retrieved
     * @return array
     * @see http://www.elasticsearch.org/guide/en/elasticsearch/reference/current/indices-get-mapping.html
     */
    public function getMapping($index = '_all', $type = '_all', $data = 'properties')
    {
        $mapping = parent::getMapping($index, $type);
        if ($index !== '_all' && $type !== '_all') {
            $mapping = $mapping[$index]['mappings'][$type][$data];
        }
        return $mapping;
    }
}