Wikibase\EntityStore\InMemory\InMemoryEntityLookup

Complexity

Total Complexity

8

Size/Duplication

Total Lines	45
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	1

Test Coverage

Coverage

100%

Importance

Changes	3
Bugs	0	Features	1

4 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	3	1
A	getEntityDocumentForId()	0	4	2
A	buildEntitiesArray()	0	6	2
A	getEntityDocumentsForIds()	0	12	3

<?php

namespace Wikibase\EntityStore\InMemory;

use Wikibase\DataModel\Entity\EntityDocument;
use Wikibase\DataModel\Entity\EntityId;
use Wikibase\EntityStore\EntityDocumentLookup;

/**
 * Internal class
 *
 * @licence GPLv2+
 * @author Thomas Pellissier Tanon
 */
class InMemoryEntityLookup implements EntityDocumentLookup {

	/**
	 * @var EntityDocument[]
	 */
	private $entities = [];

	/**
	 * @param EntityDocument[] $entities
	 */
	public function __construct( array $entities ) {
		$this->buildEntitiesArray( $entities );
	}

	private function buildEntitiesArray( array $entities ) {
		/** @var EntityDocument $entity */
		foreach( $entities as $entity ) {
			$this->entities[$entity->getId()->getSerialization()] = $entity;
		}
	}

	/**
	 * @see EntityDocumentLookup::getEntityDocumentForId
	 */
	public function getEntityDocumentForId( EntityId $entityId ) {
		$key = $entityId->getSerialization();
		return array_key_exists( $key, $this->entities ) ? $this->entities[$key] : null;
	}

	/**
	 * @see EntityDocumentLookup::getEntityDocumentsForIds
	 */
	public function getEntityDocumentsForIds( array $entityIds ) {
		$entities = [];

		foreach( $entityIds as $entityId ) {
			$entity = $this->getEntityDocumentForId( $entityId );
			if( $entity !== null ) {
				$entities[] = $entity;
			}
		}

		return $entities;
	}
}