В PHP, как мне установить класс выборки PDO по умолчанию?

Question

Используя PDO :: setAttribute , как мне указать имя класса при установке PDO::ATTR_DEFAULT_FETCH_MODE в PDO::FETCH_CLASS.

Это код, который я использую .. Я хотел бы установить его так, чтобы все мои строки возвращались как экземпляр DB_Row:

class DB_Row extends ArrayObject {}

$db = new PDO('mysql:dbname=example;host=localhost', 'user', 'pass');
$db->setAttribute(PDO::ATTR_DEFAULT_FETCH_MODE, PDO::FETCH_CLASS);

$stmt = $db->query("SELECT * FROM `table` WHERE `id` = 1;");

$row = $stmt->fetch(); // I want a DB_Row by default!

Приведенный выше код приводит к PDOException, поскольку имя класса DB_Row не было назначено.

Fatal error: Uncaught exception 'PDOException' with message 'SQLSTATE[HY000]: General    error: No fetch class specified

Как бы я поступил об этом?

Заранее спасибо ..

РЕШЕНИЕ: Я использовал fireeyedboy . Это сработало лучше всего в моей ситуации, так как я уже расширял PDOStatement для целей ведения журнала ...

class DB extends PDO {
    public function __construct($host = null, $user = null, $pass = null, $db = null) {
        try {
            parent::__construct('mysql:dbname=' . $name .';host=' . $host, $user, $pass);
            $this->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
            $this->setAttribute(PDO::ATTR_DEFAULT_FETCH_MODE, PDO::FETCH_CLASS);
            $this->setAttribute(PDO::ATTR_STATEMENT_CLASS, array('DB_Query', array('DB_Row')));
        } catch (PDOException $e) {
            die('Database Error');
        }
    }
}

class DB_Query extends PDOStatement {
    private $class;
    protected function __construct ($class = 'DB_Row') {
        $this->class = $class;
        $this->setFetchMode(PDO::FETCH_CLASS, $this->class);
    }
}

class DB_Row extends ArrayObject {
    public function __set($name, $val) {
        $this[$name] = $val;
    }
    public function __get($name) {
        return $this[$name];
    }
}

Answer 1

Я думаю, что он должен возвращать экземпляр stdclass, так что это будет ошибка - но для проверки придется поискать его в коде. будет делать, если до сих пор нет принятого ответа.

Что будет работать, так это использовать PDO :: FETCH_CLASS | PDO :: FETCH_CLASSTYPE, а затем укажите имя класса в качестве первого столбца. Пока это взломать:

$db->setAttribute(PDO::ATTR_DEFAULT_FETCH_MODE, PDO::FETCH_CLASS | PDO::FETCH_CLASSTYPE);

$stmt = $db->query("SELECT 'classname', * FROM `table` WHERE `id` = 1;");

РЕДАКТИРОВАТЬ: Как и обещал здесь код. Соответствующий код здесь http://lxr.php.net/xref/PHP_5_4/ext/pdo/pdo_stmt.c#940 Так что установка для использования fetch () возможна только с FETCH_CLASSTYPE. Альтернативой является использование PDOStatement :: fetchObject ()

Answer 2

Еще один вид взлома - расширение PDOStatement, переопределение его методов извлечения и разрешение вашему экземпляру PDO использовать его в качестве класса операторов по умолчанию.

В качестве примера я просто продемонстрирую переопределение fetch() ¹ и оставлю fetchAll(), и что у вас до вас, если вы хотите пойти по этому пути:

class Db_Row
{
}

class PDOStatementWithClass
    extends PDOStatement
{
    private $fetch_class;

    // PHP complained when I tried to make this public
    protected function __construct( $fetch_class = 'StdClass' )
    {
        // internally set the fetch class for later use
        $this->fetch_class = $fetch_class;
    }

    // let $fetch_style default to PDO::FETCH_CLASS in stead of PDO::FETCH_BOTH
    public function fetch( $fetch_style = PDO::FETCH_CLASS, $cursor_orientation = PDO::FETCH_ORI_NEXT, $cursor_offset = 0 )
    {
        // make sure we're really dealing with the correct fetch style
        if( $fetch_style == PDO::FETCH_CLASS )
        {
            // then automatically set the fetch mode of this statement
            parent::setFetchMode( $fetch_style, $this->fetch_class );
        }

        // go ahead and fetch, we should be good now
        return parent::fetch( $fetch_style, $cursor_orientation, $cursor_offset );
    }
}

$db = new PDO( /* etc... */ );
// set default fetch mode to FETCH_CLASS
$db->setAttribute( PDO::ATTR_DEFAULT_FETCH_MODE, PDO::FETCH_CLASS );
// override what statement class to use, and provide constructor arguments (found out by trial and error)
$db->setAttribute( PDO::ATTR_STATEMENT_CLASS, array( 'PDOStatementWithClass', array( 'Db_Row' ) ) );

Это дает дополнительное преимущество, так как вам придется определять PDO::FETCH_CLASS только один раз в приложении, а не в каждом запросе.

---

^{1) Я удивлен, что PHP не жаловался на переопределение сигнатуры метода.}

Answer 3

public function fetchAll($fetch_style = \PDO::FETCH_CLASS, $fetch_argument = null, $ctor_args = null) {
    if ($fetch_style == \PDO::FETCH_CLASS){
        parent::setFetchMode( $fetch_style, 'App\Core\DBFetch' );
    }
    $fetchedData = call_user_func_array(array('parent', __FUNCTION__), func_get_args());
    return $fetchedData;
}