zhamao-framework/src/ZM/Store/Database/DBStatementWrapper.php

<?php

/**
 * @noinspection PhpMissingReturnTypeInspection
 * @noinspection PhpUnused
 */

declare(strict_types=1);

namespace ZM\Store\Database;

use Doctrine\DBAL\Driver\ResultStatement;
use Doctrine\DBAL\ForwardCompatibility\Result;

class DBStatementWrapper
{
    public ?Result $stmt;

    public function __construct(?Result $stmt)
    {
        $this->stmt = $stmt;
    }

    /**
     * 获取结果的迭代器
     * wrapper method
     * @return ResultStatement
     */
    public function getIterator()
    {
        return $this->stmt->getIterator();
    }

    /**
     * 获取列数
     * wrapper method
     * @return int
     */
    public function columnCount()
    {
        return $this->stmt->columnCount();
    }

    /**
     * wrapper method
     * @return array|false|mixed
     *@throws DBException
     */
    public function fetchNumeric()
    {
        try {
            return $this->stmt->fetchNumeric();
        } catch (\Throwable $e) {
            throw new DBException($e->getMessage(), $e->getCode(), $e);
        }
    }

    /**
     * wrapper method
     * @return array|false|mixed
     *@throws DBException
     */
    public function fetchAssociative()
    {
        try {
            return $this->stmt->fetchAssociative();
        } catch (\Throwable $e) {
            throw new DBException($e->getMessage(), $e->getCode(), $e);
        }
    }

    /**
     * wrapper method
     * @return false|mixed
     *@throws DBException
     */
    public function fetchOne()
    {
        try {
            return $this->stmt->fetchOne();
        } catch (\Throwable $e) {
            throw new DBException($e->getMessage(), $e->getCode(), $e);
        }
    }

    /**
     * wrapper method
     * @throws DBException
     */
    public function fetchAllNumeric(): array
    {
        try {
            return $this->stmt->fetchAllNumeric();
        } catch (\Throwable $e) {
            throw new DBException($e->getMessage(), $e->getCode(), $e);
        }
    }

    /**
     * wrapper method
     * @throws DBException
     */
    public function fetchAllAssociative(): array
    {
        try {
            return $this->stmt->fetchAllAssociative();
        } catch (\Throwable $e) {
            throw new DBException($e->getMessage(), $e->getCode(), $e);
        }
    }

    /**
     * wrapper method
     * @throws DBException
     */
    public function fetchAllKeyValue(): array
    {
        try {
            return $this->stmt->fetchAllKeyValue();
        } catch (\Throwable $e) {
            throw new DBException($e->getMessage(), $e->getCode(), $e);
        }
    }

    /**
     * wrapper method
     * @throws DBException
     */
    public function fetchAllAssociativeIndexed(): array
    {
        try {
            return $this->stmt->fetchAllAssociativeIndexed();
        } catch (\Throwable $e) {
            throw new DBException($e->getMessage(), $e->getCode(), $e);
        }
    }

    /**
     * wrapper method
     * @throws DBException
     */
    public function fetchFirstColumn(): array
    {
        try {
            return $this->stmt->fetchFirstColumn();
        } catch (\Throwable $e) {
            throw new DBException($e->getMessage(), $e->getCode(), $e);
        }
    }

    /**
     * wrapper method
     * @throws DBException
     */
    public function iterateNumeric(): \Traversable
    {
        try {
            return $this->stmt->iterateNumeric();
        } catch (\Throwable $e) {
            throw new DBException($e->getMessage(), $e->getCode(), $e);
        }
    }

    /**
     * wrapper method
     * @throws DBException
     */
    public function iterateAssociative(): \Traversable
    {
        try {
            return $this->stmt->iterateAssociative();
        } catch (\Throwable $e) {
            throw new DBException($e->getMessage(), $e->getCode(), $e);
        }
    }

    /**
     * wrapper method
     * @throws DBException
     */
    public function iterateKeyValue(): \Traversable
    {
        try {
            return $this->stmt->iterateKeyValue();
        } catch (\Throwable $e) {
            throw new DBException($e->getMessage(), $e->getCode(), $e);
        }
    }

    /**
     * wrapper method
     * @throws DBException
     */
    public function iterateAssociativeIndexed(): \Traversable
    {
        try {
            return $this->stmt->iterateAssociativeIndexed();
        } catch (\Throwable $e) {
            throw new DBException($e->getMessage(), $e->getCode(), $e);
        }
    }

    /**
     * wrapper method
     * @throws DBException
     */
    public function iterateColumn(): \Traversable
    {
        try {
            return $this->stmt->iterateColumn();
        } catch (\Throwable $e) {
            throw new DBException($e->getMessage(), $e->getCode(), $e);
        }
    }

    /**
     * wrapper method
     * @return int
     * @throws DBException
     */
    public function rowCount()
    {
        try {
            return $this->stmt->rowCount();
        } catch (\Throwable $e) {
            throw new DBException($e->getMessage(), $e->getCode(), $e);
        }
    }

    /**
     * wrapper method
     */
    public function free(): void
    {
        $this->stmt->free();
    }
}