Improved public API in VisitLocation entity, reducing anemic model

This commit is contained in:
Alejandro Celaya
2018-10-28 15:13:45 +01:00
parent 084b1169d7
commit e046eddda9
7 changed files with 41 additions and 93 deletions

View File

@@ -6,7 +6,6 @@ namespace Shlinkio\Shlink\Core\Entity;
use Doctrine\ORM\Mapping as ORM;
use JsonSerializable;
use Shlinkio\Shlink\Common\Entity\AbstractEntity;
use Zend\Stdlib\ArraySerializableInterface;
use function array_key_exists;
/**
@@ -17,7 +16,7 @@ use function array_key_exists;
* @ORM\Entity()
* @ORM\Table(name="visit_locations")
*/
class VisitLocation extends AbstractEntity implements ArraySerializableInterface, JsonSerializable
class VisitLocation extends AbstractEntity implements JsonSerializable
{
/**
* @var string
@@ -55,15 +54,9 @@ class VisitLocation extends AbstractEntity implements ArraySerializableInterface
*/
private $timezone;
public function getCountryCode(): string
public function __construct(array $locationInfo)
{
return $this->countryCode ?? '';
}
public function setCountryCode(string $countryCode)
{
$this->countryCode = $countryCode;
return $this;
$this->exchangeArray($locationInfo);
}
public function getCountryName(): string
@@ -71,99 +64,50 @@ class VisitLocation extends AbstractEntity implements ArraySerializableInterface
return $this->countryName ?? '';
}
public function setCountryName(string $countryName): self
{
$this->countryName = $countryName;
return $this;
}
public function getRegionName(): string
{
return $this->regionName ?? '';
}
public function setRegionName(string $regionName): self
{
$this->regionName = $regionName;
return $this;
}
public function getCityName(): string
{
return $this->cityName ?? '';
}
public function setCityName(string $cityName): self
{
$this->cityName = $cityName;
return $this;
}
public function getLatitude(): string
{
return $this->latitude ?? '';
}
public function setLatitude(string $latitude): self
{
$this->latitude = $latitude;
return $this;
}
public function getLongitude(): string
{
return $this->longitude ?? '';
}
public function setLongitude(string $longitude): self
public function getCityName(): string
{
$this->longitude = $longitude;
return $this;
}
public function getTimezone(): string
{
return $this->timezone ?? '';
}
public function setTimezone(string $timezone): self
{
$this->timezone = $timezone;
return $this;
return $this->cityName ?? '';
}
/**
* Exchange internal values from provided array
*/
public function exchangeArray(array $array): void
private function exchangeArray(array $array): void
{
if (array_key_exists('country_code', $array)) {
$this->setCountryCode((string) $array['country_code']);
$this->countryCode = (string) $array['country_code'];
}
if (array_key_exists('country_name', $array)) {
$this->setCountryName((string) $array['country_name']);
$this->countryName = (string) $array['country_name'];
}
if (array_key_exists('region_name', $array)) {
$this->setRegionName((string) $array['region_name']);
$this->regionName = (string) $array['region_name'];
}
if (array_key_exists('city', $array)) {
$this->setCityName((string) $array['city']);
$this->cityName = (string) $array['city'];
}
if (array_key_exists('latitude', $array)) {
$this->setLatitude((string) $array['latitude']);
$this->latitude = (string) $array['latitude'];
}
if (array_key_exists('longitude', $array)) {
$this->setLongitude((string) $array['longitude']);
$this->longitude = (string) $array['longitude'];
}
if (array_key_exists('time_zone', $array)) {
$this->setTimezone((string) $array['time_zone']);
$this->timezone = (string) $array['time_zone'];
}
}
/**
* Return an array representation of the object
*/
public function getArrayCopy(): array
public function jsonSerialize(): array
{
return [
'countryCode' => $this->countryCode,
@@ -175,9 +119,4 @@ class VisitLocation extends AbstractEntity implements ArraySerializableInterface
'timezone' => $this->timezone,
];
}
public function jsonSerialize(): array
{
return $this->getArrayCopy();
}
}

View File

@@ -39,7 +39,7 @@ class VisitRepositoryTest extends DatabaseTestCase
$visit = new Visit();
if ($i % 2 === 0) {
$location = new VisitLocation();
$location = new VisitLocation([]);
$this->getEntityManager()->persist($location);
$visit->setVisitLocation($location);
}

View File

@@ -18,8 +18,7 @@ class VisitLocationTest extends TestCase
'longitude' => -2000.4,
];
$location = new VisitLocation();
$location->exchangeArray($payload);
$location = new VisitLocation($payload);
$this->assertSame('1000.7', $location->getLatitude());
$this->assertSame('-2000.4', $location->getLongitude());