NodeBatch.php

Go to the documentation of this file.

<?php

 
namespace Depage\XmlDb\Helpers;
 
use Depage\XmlDb\Exceptions\XmlDbException;

class NodeBatch
{
    use IdentifierChecker;

    public const DEFAULT_SIZE = 200;
 
    protected $pdo;

    protected $entries = [];

    public function __construct(
        $pdo,
        protected string $table,
        protected int $docId,
        protected int $size = self::DEFAULT_SIZE
    ) {
        $this->pdo = $pdo;
 
        $this->checkIdentifier($table, 'table');
    }
 
    public function isEmpty(): bool
    {
        return count($this->entries) == 0;
    }

    public function isFull(): bool
    {
        return count($this->entries) >= $this->size;
    }

    public function add($key, $id, $parentId, $pos, $name, $value, $type): void
    {
        $this->entries[$key] = [
            'id' => $id,
            'parent_id' => $parentId,
            'pos' => $pos,
            'name' => $name,
            'value' => $value,
            'type' => $type,
        ];
    }

    public function write(): array
    {
        if ($this->isEmpty()) {
            return [];
        }
 
        $entries = $this->entries;
        $this->entries = [];
 
        $this->insert($entries);
 
        $ids = $this->readAutoIncrementIds($entries);
 
        foreach ($entries as $key => $entry) {
            if (!is_null($entry['id'])) {
                $ids[$key] = $entry['id'];
            }
        }
        ksort($ids);
 
        return $ids;
    }

    protected function insert(array $entries): void
    {
        $values = [];
        $params = [];
 
        foreach ($entries as $entry) {
            $values[] = '(?, ?, ?, ?, ?, ?, ?)';
            array_push(
                $params,
                $entry['id'],
                $entry['parent_id'],
                $this->docId,
                $entry['pos'],
                $entry['name'],
                $entry['value'],
                $entry['type']
            );
        }
 
        $query = $this->pdo->prepare(
            "INSERT {$this->table}
                (id, id_parent, id_doc, pos, name, value, type)
            VALUES
                " . implode(", ", $values) . "
            ON DUPLICATE KEY UPDATE id = LAST_INSERT_ID(id)"
        );
        $query->execute($params);
    }

    protected function readAutoIncrementIds(array $entries): array
    {
        $wanted = [];
        $parents = [];
        $positions = [];
 
        foreach ($entries as $key => $entry) {
            if (!is_null($entry['id'])) {
                continue;
            }
 
            $wanted[$entry['parent_id'] . ':' . $entry['pos']] = $key;
            $parents[] = (int) $entry['parent_id'];
            $positions[] = (int) $entry['pos'];
        }
 
        if (count($wanted) == 0) {
            return [];
        }
 
        $parents = array_values(array_unique($parents));
        $positions = array_values(array_unique($positions));
 
        $query = $this->pdo->prepare(
            "SELECT id, id_parent, pos
            FROM {$this->table}
            WHERE id_doc = ?
                AND id_parent IN (" . implode(", ", array_fill(0, count($parents), '?')) . ")
                AND pos IN (" . implode(", ", array_fill(0, count($positions), '?')) . ")"
        );
        $query->execute(array_merge(
            [$this->docId],
            $parents,
            $positions
        ));
        $rows = $query->fetchAll(\PDO::FETCH_OBJ);
        $query->closeCursor();
 
        $ids = [];
        foreach ($rows as $row) {
            $sibling = $row->id_parent . ':' . $row->pos;
 
            if (isset($wanted[$sibling])) {
                $ids[$wanted[$sibling]] = (int) $row->id;
                unset($wanted[$sibling]);
            }
        }
 
        if (count($wanted) > 0) {
            throw new XmlDbException("Inserted node could not be found again after batch insert.");
        }
 
        return $ids;
    }
}
 
/* vim:set ft=php sw=4 sts=4 fdm=marker et : */