DocumentFactory::addToDoc() - Code Metrics - Inspection of "Hans/its the same (#191)" - Firesphere/silverstripe-solr-search - Measure and Improve Code Quality continuously with Scrutinizer

Passed

Push — master ( 6e4405...4199c0 )

by Marco

created 2020-01-12 04:40 UTC

DocumentFactory::addToDoc() A

↳ Parent: DocumentFactory

Complexity

Conditions	3
Paths	2

Size

Total Lines	10
Code Lines	4

Duplication

Lines	0
Ratio	0 %

Code Coverage

Tests	5
CRAP Score	3

Importance

Changes	3
Bugs	2	Features	0

Metric	Value
eloc	4
c	3
b	2
f	0
dl	0
loc	10
ccs	5
cts	5
cp	1
rs	10
cc	3
nc	2
nop	4
crap	3

<?php
/**
 * class DocumentFactory|Firesphere\SolrSearch\Factories\DocumentFactory Build a Solarium document to push
 *
 * @package Firesphere\SolrSearch\Factories
 * @author Simon `Firesphere` Erkelens; Marco `Sheepy` Hermo
 * @copyright Copyright (c) 2018 - now() Firesphere & Sheepy
 */

namespace Firesphere\SolrSearch\Factories;

use Exception;
use Firesphere\SolrSearch\Extensions\DataObjectExtension;
use Firesphere\SolrSearch\Helpers\DataResolver;
use Firesphere\SolrSearch\Helpers\FieldResolver;
use Firesphere\SolrSearch\Helpers\Statics;
use Firesphere\SolrSearch\Indexes\BaseIndex;
use Firesphere\SolrSearch\Services\SolrCoreService;
use Firesphere\SolrSearch\Traits\DocumentFactoryTrait;
use Firesphere\SolrSearch\Traits\LoggerTrait;
use SilverStripe\Core\ClassInfo;
use SilverStripe\Core\Config\Configurable;
use SilverStripe\Core\Extensible;
use SilverStripe\Core\Injector\Injector;
use SilverStripe\ORM\DataObject;
use SilverStripe\ORM\FieldType\DBDate;
use SilverStripe\ORM\FieldType\DBField;
use Solarium\QueryType\Update\Query\Document;
use Solarium\QueryType\Update\Query\Query;

/**
 * Class DocumentFactory
 * Factory to create documents to be pushed to Solr
 *
 * @package Firesphere\SolrSearch\Factories
 */
class DocumentFactory
{
    use Configurable;
    use Extensible;
    use DocumentFactoryTrait;
    use LoggerTrait;

    /**
     * Numeral types in Solr
     *
     * @var array
     */
    protected static $numerals = [
        'tint',
        'tfloat',
        'tdouble',
    ];
    /**
     * Debug this build
     *
     * @var bool
     */
    protected $debug = false;

    /**
     * DocumentFactory constructor, sets up the field resolver
     */
    public function __construct()
    {
        $this->fieldResolver = Injector::inst()->get(FieldResolver::class);
    }

    /**
     * Note, it can only take one type of class at a time!
     * So make sure you properly loop and set $class
     *
     * @param array $fields Fields to index
     * @param BaseIndex $index Index to push the documents to
     * @param Query $update Update Query object
     * @return array Documents to be pushed
     * @throws Exception
     */
    public function buildItems($fields, $index, $update): array
    {
        $this->getFieldResolver()->setIndex($index);
        $boostFields = $index->getBoostedFields();
        $docs = [];
        if ($this->debug) {
            $this->indexGroupMessage($index);
        }

        foreach ($this->getItems() as $item) {
            // Don't index items that should not show in search explicitly.
            // Just a "not" is insufficient, as it could be null or false (both meaning, not set)
            if ($item->ShowInSearch === 0) {
                continue;
            }
            /** @var Document $doc */
            $doc = $update->createDocument();
            $this->addDefaultFields($doc, $item);

            $this->buildFields($fields, $doc, $item, $boostFields);
            $item->destroy();

            $docs[] = $doc;
        }

        return $docs;
    }

    /**
     * Show the message about what is being indexed
     *
     * @param BaseIndex $index
     */
    protected function indexGroupMessage(BaseIndex $index): void
    {
        $debugString = sprintf(
            'Indexing %s on %s (%s items)%s',
            $this->getClass(),
            $index->getIndexName(),
            $this->getItems()->count(),
            PHP_EOL
        );
        $this->getLogger()->info($debugString);
    }

    /**
     * Add fields that should always be included
     *
     * @param Document $doc Solr Document
     * @param DataObject|DataObjectExtension $item Item to get the data from
     */
    protected function addDefaultFields(Document $doc, DataObject $item)
    {
        $doc->setKey(SolrCoreService::ID_FIELD, $item->ClassName . '-' . $item->ID);
        $doc->addField(SolrCoreService::CLASS_ID_FIELD, $item->ID);
        $doc->addField('ClassName', $item->ClassName);
        $doc->addField('ClassHierarchy', ClassInfo::ancestry($item));
        $doc->addField('ViewStatus', $item->getViewStatus());
        $this->extend('updateDefaultFields', $doc, $item);
    }

    /**
     * Create the required record for a field
     *
     * @param array $fields Fields to build a record for
     * @param Document $doc Document for Solr
     * @param DataObject $item Object to get the data for
     * @param array $boostFields Custom set of index-time-boosted fields
     * @throws Exception
     */
    protected function buildFields($fields, Document $doc, DataObject $item, array $boostFields): void
    {
        foreach ($fields as $field) {
            $fieldData = $this->getFieldResolver()->resolveField($field);
            foreach ($fieldData as $dataField => $options) {
                $options['boost'] = $boostFields[$field] ?? null;
                $this->addField($doc, $item, $options);
            }
        }
    }

    /**
     * Add a single field to the Solr index
     *
     * @param Document $doc Solr Document
     * @param DataObject $object Object whose field is to be added
     * @param array $options Additional options
     */
    protected function addField($doc, $object, $options): void
    {
        if (!$this->classIs($object, $options['origin'])) {
            return;
        }

        $this->extend('onBeforeAddField', $options);

        $valuesForField = $this->getValuesForField($object, $options);

        $typeMap = Statics::getTypeMap();
        $type = $typeMap[$options['type']] ?? $typeMap['*'];

        foreach ($valuesForField as $value) {
            if ($value === null) {
                continue;
            }
            $this->extend('onBeforeAddDoc', $options, $value);
            $this->addToDoc($doc, $options, $type, $value);
        }
    }

    /**
     * Determine if the given object is one of the given type
     *
     * @param string|DataObject $class Class to compare
     * @param array|string $base Class or list of base classes
     * @return bool
     * @todo remove in favour of the inheritance check from PHP
     */
    protected function classIs($class, $base): bool
    {
        $base = is_array($base) ? $base : [$base];

        foreach ($base as $nextBase) {
            if ($this->classEquals($class, $nextBase)) {
                return true;
            }
        }

        return false;
    }

    /**
     * Check if a base class is an instance of the expected base group
     *
     * @param string|DataObject $class Class to compare
     * @param string $base Base class
     * @return bool
     */
    protected function classEquals($class, $base): bool
    {
        return $class === $base || ($class instanceof $base);
    }

    /**
     * Use the DataResolver to find the value(s) for a field.
     * Returns an array of values, and if it's multiple, it becomes a long array
     *
     * @param DataObject $object Object to resolve
     * @param array $options Customised options
     * @return array
     */
    protected function getValuesForField($object, $options): array
    {
        try {
            $valuesForField = [DataResolver::identify($object, $options['fullfield'])];
        } catch (Exception $error) {
            // @codeCoverageIgnoreStart
            $valuesForField = [];
            // @codeCoverageIgnoreEnd
        }

        return $valuesForField;
    }

    /**
     * Push field to a document
     *
     * @param Document $doc Solr document
     * @param array $options Custom options
     * @param string $type Type of Solr field
     * @param DBField|string $value Value(s) of the field
     */
    protected function addToDoc($doc, $options, $type, $value): void
    {
        /* Solr requires dates in the form 1995-12-31T23:59:59Z, so we need to normalize to GMT */
        if ($type === 'tdate' || $value instanceof DBDate) {
            $value = gmdate('Y-m-d\TH:i:s\Z', strtotime($value));
        }

        $name = getShortFieldName($options['name']);

        $doc->addField($name, $value, $options['boost'], Document::MODIFIER_SET);
    }

    /**
     * Are we debugging?
     *
     * @return bool
     */
    public function isDebug(): bool
    {
        return $this->debug;
    }

    /**
     * Set to true if debugging should be enabled
     *
     * @param bool $debug
     * @return DocumentFactory
     */
    public function setDebug(bool $debug): DocumentFactory
    {
        $this->debug = $debug;

        return $this;
    }
}


1		<?php
2		/**
3		* class DocumentFactory\|Firesphere\SolrSearch\Factories\DocumentFactory Build a Solarium document to push
4		*
5		* @package Firesphere\SolrSearch\Factories
6		* @author Simon `Firesphere` Erkelens; Marco `Sheepy` Hermo
7		* @copyright Copyright (c) 2018 - now() Firesphere & Sheepy
8		*/
9
10		namespace Firesphere\SolrSearch\Factories;
11
12		use Exception;
13		use Firesphere\SolrSearch\Extensions\DataObjectExtension;
14		use Firesphere\SolrSearch\Helpers\DataResolver;
15		use Firesphere\SolrSearch\Helpers\FieldResolver;
16		use Firesphere\SolrSearch\Helpers\Statics;
17		use Firesphere\SolrSearch\Indexes\BaseIndex;
18		use Firesphere\SolrSearch\Services\SolrCoreService;
19		use Firesphere\SolrSearch\Traits\DocumentFactoryTrait;
20		use Firesphere\SolrSearch\Traits\LoggerTrait;
21		use SilverStripe\Core\ClassInfo;
22		use SilverStripe\Core\Config\Configurable;
23		use SilverStripe\Core\Extensible;
24		use SilverStripe\Core\Injector\Injector;
25		use SilverStripe\ORM\DataObject;
26		use SilverStripe\ORM\FieldType\DBDate;
27		use SilverStripe\ORM\FieldType\DBField;
28		use Solarium\QueryType\Update\Query\Document;
29		use Solarium\QueryType\Update\Query\Query;
30
31		/**
32		* Class DocumentFactory
33		* Factory to create documents to be pushed to Solr
34		*
35		* @package Firesphere\SolrSearch\Factories
36		*/
37		class DocumentFactory
38		{
39		use Configurable;
40		use Extensible;
41		use DocumentFactoryTrait;
42		use LoggerTrait;
43
44		/**
45		* Numeral types in Solr
46		*
47		* @var array
48		*/
49		protected static $numerals = [
50		'tint',
51		'tfloat',
52		'tdouble',
53		];
54		/**
55		* Debug this build
56		*
57		* @var bool
58		*/
59		protected $debug = false;
60
61		/**
62		* DocumentFactory constructor, sets up the field resolver
63		*/
64	10	public function __construct()
65		{
66	10	$this->fieldResolver = Injector::inst()->get(FieldResolver::class);
67	10	}
68
69		/**
70		* Note, it can only take one type of class at a time!
71		* So make sure you properly loop and set $class
72		*
73		* @param array $fields Fields to index
74		* @param BaseIndex $index Index to push the documents to
75		* @param Query $update Update Query object
76		* @return array Documents to be pushed
77		* @throws Exception
78		*/
79	7	public function buildItems($fields, $index, $update): array
80		{
81	7	$this->getFieldResolver()->setIndex($index);
82	7	$boostFields = $index->getBoostedFields();
83	7	$docs = [];
84	7	if ($this->debug) {
85	1	$this->indexGroupMessage($index);
86		}
87
88	7	foreach ($this->getItems() as $item) {
89		// Don't index items that should not show in search explicitly.
90		// Just a "not" is insufficient, as it could be null or false (both meaning, not set)
91	7	if ($item->ShowInSearch === 0) {
92	1	continue;
93		}
94		/** @var Document $doc */
95	7	$doc = $update->createDocument();
96	7	$this->addDefaultFields($doc, $item);
97
98	7	$this->buildFields($fields, $doc, $item, $boostFields);
99	7	$item->destroy();
100
101	7	$docs[] = $doc;
102		}
103
104	7	return $docs;
105		}
106
107		/**
108		* Show the message about what is being indexed
109		*
110		* @param BaseIndex $index
111		*/
112	1	protected function indexGroupMessage(BaseIndex $index): void
113		{
114	1	$debugString = sprintf(
115	1	'Indexing %s on %s (%s items)%s',
116	1	$this->getClass(),
117	1	$index->getIndexName(),
118	1	$this->getItems()->count(),
119	1	PHP_EOL
120		);
121	1	$this->getLogger()->info($debugString);
122	1	}
123
124		/**
125		* Add fields that should always be included
126		*
127		* @param Document $doc Solr Document
128		* @param DataObject\|DataObjectExtension $item Item to get the data from
129		*/
130	7	protected function addDefaultFields(Document $doc, DataObject $item)
131		{
132	7	$doc->setKey(SolrCoreService::ID_FIELD, $item->ClassName . '-' . $item->ID);
133	7	$doc->addField(SolrCoreService::CLASS_ID_FIELD, $item->ID);
134	7	$doc->addField('ClassName', $item->ClassName);
135	7	$doc->addField('ClassHierarchy', ClassInfo::ancestry($item));
136	7	$doc->addField('ViewStatus', $item->getViewStatus());
137	7	$this->extend('updateDefaultFields', $doc, $item);
138	7	}
139
140		/**
141		* Create the required record for a field
142		*
143		* @param array $fields Fields to build a record for
144		* @param Document $doc Document for Solr
145		* @param DataObject $item Object to get the data for
146		* @param array $boostFields Custom set of index-time-boosted fields
147		* @throws Exception
148		*/
149	7	protected function buildFields($fields, Document $doc, DataObject $item, array $boostFields): void
150		{
151	7	foreach ($fields as $field) {
152	7	$fieldData = $this->getFieldResolver()->resolveField($field);
153	7	foreach ($fieldData as $dataField => $options) {
154	7	$options['boost'] = $boostFields[$field] ?? null;
155	7	$this->addField($doc, $item, $options);
156		}
157		}
158	7	}
159
160		/**
161		* Add a single field to the Solr index
162		*
163		* @param Document $doc Solr Document
164		* @param DataObject $object Object whose field is to be added
165		* @param array $options Additional options
166		*/
167	7	protected function addField($doc, $object, $options): void
168		{
169	7	if (!$this->classIs($object, $options['origin'])) {
170		return;
171		}
172
173	7	$this->extend('onBeforeAddField', $options);
174
175	7	$valuesForField = $this->getValuesForField($object, $options);
176
177	7	$typeMap = Statics::getTypeMap();
178	7	$type = $typeMap[$options['type']] ?? $typeMap['*'];
179
180	7	foreach ($valuesForField as $value) {
181	7	if ($value === null) {
182	5	continue;
183		}
184	7	$this->extend('onBeforeAddDoc', $options, $value);
185	7	$this->addToDoc($doc, $options, $type, $value);
186		}
187	7	}
188
189		/**
190		* Determine if the given object is one of the given type
191		*
192		* @param string\|DataObject $class Class to compare
193		* @param array\|string $base Class or list of base classes
194		* @return bool
195		* @todo remove in favour of the inheritance check from PHP
196		*/
197	7	protected function classIs($class, $base): bool
198		{
199	7	$base = is_array($base) ? $base : [$base];
200
201	7	foreach ($base as $nextBase) {
202	7	if ($this->classEquals($class, $nextBase)) {
203	7	return true;
204		}
205		}
206
207		return false;
208		}
209
210		/**
211		* Check if a base class is an instance of the expected base group
212		*
213		* @param string\|DataObject $class Class to compare
214		* @param string $base Base class
215		* @return bool
216		*/
217	7	protected function classEquals($class, $base): bool
218		{
219	7	return $class === $base \|\| ($class instanceof $base);
220		}
221
222		/**
223		* Use the DataResolver to find the value(s) for a field.
224		* Returns an array of values, and if it's multiple, it becomes a long array
225		*
226		* @param DataObject $object Object to resolve
227		* @param array $options Customised options
228		* @return array
229		*/
230	7	protected function getValuesForField($object, $options): array
231		{
232		try {
233	7	$valuesForField = [DataResolver::identify($object, $options['fullfield'])];
234	6	} catch (Exception $error) {
235		// @codeCoverageIgnoreStart
236		$valuesForField = [];
237		// @codeCoverageIgnoreEnd
238		}
239
240	7	return $valuesForField;
241		}
242
243		/**
244		* Push field to a document
245		*
246		* @param Document $doc Solr document
247		* @param array $options Custom options
248		* @param string $type Type of Solr field
249		* @param DBField\|string $value Value(s) of the field
250		*/
251	7	protected function addToDoc($doc, $options, $type, $value): void
252		{
253		/* Solr requires dates in the form 1995-12-31T23:59:59Z, so we need to normalize to GMT */
254	7	if ($type === 'tdate' \|\| $value instanceof DBDate) {
255	6	$value = gmdate('Y-m-d\TH:i:s\Z', strtotime($value));
256		}
257
258	7	$name = getShortFieldName($options['name']);
259
260	7	$doc->addField($name, $value, $options['boost'], Document::MODIFIER_SET);
261	7	}
262
263		/**
264		* Are we debugging?
265		*
266		* @return bool
267		*/
268	1	public function isDebug(): bool
269		{
270	1	return $this->debug;
271		}
272
273		/**
274		* Set to true if debugging should be enabled
275		*
276		* @param bool $debug
277		* @return DocumentFactory
278		*/
279	8	public function setDebug(bool $debug): DocumentFactory
280		{
281	8	$this->debug = $debug;
282
283	8	return $this;
284		}
285		}
286

Firesphere / silverstripe-solr-search

Push — master ( 6e4405...4199c0 )

DocumentFactory::addToDoc() A

Complexity

Size

Duplication

Code Coverage

Importance

Duplication Side-by-Side

Filter issues like