Sirius\Orm\Action\Update

Complexity

Total Complexity

7

Size/Duplication

Total Lines	60
Duplicated Lines	0 %

Importance

Changes	2
Bugs	0	Features	0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	addColumns()	0	7	2
A	execute()	0	28	3
A	onSuccess()	0	6	2

<?php
declare(strict_types=1);

namespace Sirius\Orm\Action;

use Sirius\Orm\Entity\StateEnum;
use Sirius\Orm\Helpers\Arr;

class Update extends Save
{
    protected $entityState;

    protected $extraColumns = [];

    /**
     * Add extra columns to be added to the insert.
     * To be used by behaviours relations
     *
     * @param array $columns
     *
     * @return self
     */
    public function addColumns(array $columns)
    {
        foreach ($columns as $name => $value) {
            $this->extraColumns[$name] = $value;
        }

        return $this;
    }

    protected function execute()
    {
        $conditions = $this->getConditions();

        if (empty($conditions)) {
            return;
        }

        $this->entityState = $this->entity->getState();

        $connection = $this->mapper->getWriteConnection();

        $columns = $this->entityHydrator->extract($this->entity);
        $changes = Arr::renameKeys($this->entity->getChanges(), array_flip($this->mapper->getConfig()->getColumnAttributeMap()));
        $columns = Arr::only($columns, array_keys($changes));
        $columns = array_merge(
            $columns,
            $this->extraColumns,
            $this->mapper->getConfig()->getGuards()
        );
        $columns = Arr::except($columns, $this->mapper->getConfig()->getPrimaryKey());

        if (count($columns) > 0) {
            $updateSql = new \Sirius\Sql\Update($connection);
            $updateSql->table($this->mapper->getConfig()->getTable())
                      ->columns($columns)
                      ->whereAll($conditions, false);
            $updateSql->perform();
        }
    }

    public function onSuccess()
    {
        foreach ($this->extraColumns as $col => $value) {
            $this->entityHydrator->set($this->entity, $col, $value);
        }
        $this->entity->setState(StateEnum::SYNCHRONIZED);
    }
}