123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711 |
- <?php
-
- class QPDB implements QueryPathExtension {
- protected $qp;
- protected $dsn;
- protected $db;
- protected $opts;
- protected $row = NULL;
- protected $stmt = NULL;
-
- protected static $con = NULL;
-
-
- static function baseDB($dsn, $options = array()) {
-
- $opts = $options + array(
- 'username' => NULL,
- 'password' => NULL,
- 'db params' => array(PDO::ATTR_ERRMODE => PDO::ERRMODE_EXCEPTION),
- );
-
-
-
- if ($dsn instanceof PDO) {
- self::$con = $dsn;
- return;
- }
- self::$con = new PDO($dsn, $opts['username'], $opts['password'], $opts['db params']);
- }
-
-
- static function getBaseDB() {return self::$con;}
-
-
- protected $cycleRows = FALSE;
-
-
- public function __construct(QueryPath $qp) {
- $this->qp = $qp;
-
- $this->db = self::$con;
- }
-
-
- public function dbInit($dsn, $options = array()) {
- $this->opts = $options + array(
- 'username' => NULL,
- 'password' => NULL,
- 'db params' => array(PDO::ATTR_ERRMODE => PDO::ERRMODE_EXCEPTION),
- );
- $this->dsn = $dsn;
- $this->db = new PDO($dsn, $this->opts['username'], $this->opts['password'], $this->opts['db params']);
-
-
- return $this->qp;
- }
-
-
- public function query($sql, $args = array()) {
- $this->stmt = $this->db->prepare($sql);
- $this->stmt->execute($args);
- return $this->qp;
- }
-
-
- public function queryInto($sql, $args = array(), $template = NULL) {
- $stmt = $this->db->prepare($sql);
- $stmt->setFetchMode(PDO::FETCH_ASSOC);
- $stmt->execute($args);
-
-
- if (empty($template)) {
- foreach ($stmt as $row) foreach ($row as $datum) $this->qp->append($datum);
- }
-
- else {
- foreach ($stmt as $row) $this->qp->tpl($template, $row);
- }
-
- $stmt->closeCursor();
- return $this->qp;
- }
-
-
- public function doneWithQuery() {
- if (isset($this->stmt) && $this->stmt instanceof PDOStatement) {
-
-
- $this->stmt->closeCursor();
- }
-
- unset($this->stmt);
- $this->row = NULL;
- $this->cycleRows = FALSE;
- return $this->qp;
- }
-
-
- public function exec($sql) {
- $this->db->exec($sql);
- return $this->qp;
- }
-
-
- public function nextRow() {
- $this->row = $this->stmt->fetch(PDO::FETCH_ASSOC);
- return $this->qp;
- }
-
-
- public function withEachRow() {
- $this->cycleRows = TRUE;
- return $this->qp;
- }
-
-
- protected function addData($columnName, $qpFunc = 'append', $wrap = NULL) {
- $columns = is_array($columnName) ? $columnName : array($columnName);
- $hasWrap = !empty($wrap);
- if ($this->cycleRows) {
- while (($row = $this->stmt->fetch(PDO::FETCH_ASSOC)) !== FALSE) {
- foreach ($columns as $col) {
- if (isset($row[$col])) {
- $data = $row[$col];
- if ($hasWrap)
- $data = qp()->append($wrap)->deepest()->append($data)->top();
- $this->qp->$qpFunc($data);
- }
- }
- }
- $this->cycleRows = FALSE;
- $this->doneWithQuery();
- }
- else {
- if ($this->row !== FALSE) {
- foreach ($columns as $col) {
- if (isset($this->row[$col])) {
- $data = $this->row[$col];
- if ($hasWrap)
- $data = qp()->append($wrap)->deepest()->append($data)->top();
- $this->qp->$qpFunc($data);
- }
- }
- }
- }
- return $this->qp;
- }
-
-
- public function getStatement() {
- return $this->stmt;
- }
-
-
- public function getLastInsertID() {
- $con = self::$con;
- return $con->lastInsertId();
- }
-
-
- public function appendColumn($columnName, $wrap = NULL) {
- return $this->addData($columnName, 'append', $wrap);
- }
-
-
- public function prependColumn($columnName, $wrap = NULL) {
- return $this->addData($columnName, 'prepend', $wrap);
- }
-
-
- public function columnBefore($columnName, $wrap = NULL) {
- return $this->addData($columnName, 'before', $wrap);
- }
-
-
- public function columnAfter($columnName, $wrap = NULL) {
- return $this->addData($columnName, 'after', $wrap);
- }
-
- }
- if (!defined('QPDB_OVERRIDE'))
- QueryPathExtensionRegistry::extend('QPDB');
|