Wikibase\DataModel\Services\Lookup\EntityRetrievingDataTypeLookup

Complexity

Total Complexity

5

Size/Duplication

Total Lines	47
Duplicated Lines	0 %

Test Coverage

Coverage

90%

Importance

Changes	3
Bugs	0	Features	0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	2	1
A	getDataTypeIdForProperty()	0	10	2
A	getProperty()	0	8	2

<?php

namespace Wikibase\DataModel\Services\Lookup;

use Wikibase\DataModel\Entity\Property;
use Wikibase\DataModel\Entity\PropertyId;

/**
 * PropertyDataTypeLookup that uses an EntityLookup to find
 * a property's data type ID.
 *
 * @since 1.1
 *
 * @license GPL-2.0-or-later
 * @author Jeroen De Dauw < jeroendedauw@gmail.com >
 */
class EntityRetrievingDataTypeLookup implements PropertyDataTypeLookup {

	/**
	 * @var EntityLookup
	 */
	private $entityLookup;
	private array $propertyIdsInProcess = [];

	public function __construct( EntityLookup $entityLookup ) {
		$this->entityLookup = $entityLookup;
	}

	/**
	 * @since 2.0
	 *
	 * @param PropertyId $propertyId
	 *
	 * @return string
	 * @throws PropertyDataTypeLookupException
	 */
	public function getDataTypeIdForProperty( PropertyId $propertyId ) {
		if ( array_key_exists( $propertyId->getSerialization(), $this->propertyIdsInProcess ) ) {
			// avoid self-referencing loop for newly created properties (T374230)
			throw new PropertyDataTypeLookupException( $propertyId, 'loop detected' );
		}
		$this->propertyIdsInProcess[ $propertyId->getSerialization() ] = true;
		try {
			return $this->getProperty( $propertyId )->getDataTypeId();
		} finally {
			unset( $this->propertyIdsInProcess[ $propertyId->getSerialization() ] );
		}
	}

	/**
	 * @param PropertyId $propertyId
	 *
	 * @return Property
	 * @throws PropertyDataTypeLookupException
	 */
	private function getProperty( PropertyId $propertyId ) {
		$property = $this->entityLookup->getEntity( $propertyId );

		if ( !( $property instanceof Property ) ) {
			throw new PropertyDataTypeLookupException( $propertyId );
		}

		return $property;
	}

}