Source of file QueryBuilder.php
Size: 30,873 Bytes - Last Modified: 2019-12-11T16:15:51-05:00
src/QueryBuilder.php
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337 | <?php declare(strict_types=1); /** * Query * * SQL Query Builder / Database Abstraction Layer * * PHP version 7.1 * * @package Query * @author Timothy J. Warren <tim@timshomepage.net> * @copyright 2012 - 2018 Timothy J. Warren * @license http://www.opensource.org/licenses/mit-license.html MIT License * @link https://git.timshomepage.net/aviat4ion/Query */ namespace Query; use function regexInArray; use BadMethodCallException; use PDO; use PDOStatement; use Query\Drivers\DriverInterface; /** * Convenience class for creating sql queries * * @method affectedRows(): int * @method beginTransaction(): bool * @method commit(): bool * @method errorCode(): string * @method errorInfo(): array * @method exec(string $statement): int * @method getAttribute(int $attribute) * @method getColumns(string $table): array | null * @method getDbs(): array | null * @method getFks(string $table): array | null * @method getFunctions(): array | null * @method getIndexes(string $table): array | null * @method getLastQuery(): string * @method getProcedures(): array | null * @method getSchemas(): array | null * @method getSequences(): array | null * @method getSystemTables(): array | null * @method getTables(): array * @method getTriggers(): array | null * @method getTypes(): array | null * @method getUtil(): \Query\Drivers\AbstractUtil * @method getViews(): array | null * @method inTransaction(): bool * @method lastInsertId(string $name = NULL): string * @method numRows(): int | null * @method prepare(string $statement, array $driver_options = []): PDOStatement * @method prepareExecute(string $sql, array $params): PDOStatement * @method prepareQuery(string $sql, array $data): PDOStatement * @method query(string $statement): PDOStatement * @method quote(string $string, int $parameter_type = PDO::PARAM_STR): string * @method rollback(): bool * @method setAttribute(int $attribute, $value): bool * @method setTablePrefix(string $prefix): void * @method truncate(string $table): PDOStatement */ class QueryBuilder implements QueryBuilderInterface { /** * Convenience property for connection management * @var string */ public $connName = ''; /** * List of queries executed * @var array */ public $queries = [ 'total_time' => 0 ]; /** * Whether to do only an explain on the query * @var boolean */ protected $explain = FALSE; /** * The current database driver * @var DriverInterface */ protected $driver; /** * Query parser class instance * @var QueryParser */ protected $parser; /** * Query Builder state * @var State */ protected $state; // -------------------------------------------------------------------------- // ! Methods // -------------------------------------------------------------------------- /** * Constructor * * @param DriverInterface $driver * @param QueryParser $parser */ public function __construct(DriverInterface $driver, QueryParser $parser) { // Inject driver and parser $this->driver = $driver; $this->parser = $parser; // Create new State object $this->state = new State(); } /** * Destructor * @codeCoverageIgnore */ public function __destruct() { $this->driver = NULL; } /** * Calls a function further down the inheritance chain. * 'Implements' methods on the driver object * * @param string $name * @param array $params * @return mixed * @throws BadMethodCallException */ public function __call(string $name, array $params) { if (method_exists($this->driver, $name)) { return \call_user_func_array([$this->driver, $name], $params); } throw new BadMethodCallException('Method does not exist'); } // -------------------------------------------------------------------------- // ! Select Queries // -------------------------------------------------------------------------- /** * Specifies rows to select in a query * * @param string $fields * @return QueryBuilderInterface */ public function select(string $fields): QueryBuilderInterface { // Split fields by comma $fieldsArray = explode(',', $fields); $fieldsArray = array_map('mb_trim', $fieldsArray); // Split on 'As' foreach ($fieldsArray as $key => $field) { if (stripos($field, 'as') !== FALSE) { $fieldsArray[$key] = preg_split('` as `i', $field); $fieldsArray[$key] = array_map('mb_trim', $fieldsArray[$key]); } } // Quote the identifiers $safeArray = $this->driver->quoteIdent($fieldsArray); unset($fieldsArray); // Join the strings back together for($i = 0, $c = count($safeArray); $i < $c; $i++) { if (\is_array($safeArray[$i])) { $safeArray[$i] = implode(' AS ', $safeArray[$i]); } } $this->state->appendSelectString(implode(', ', $safeArray)); return $this; } /** * Selects the maximum value of a field from a query * * @param string $field * @param string|bool $as * @return QueryBuilderInterface */ public function selectMax(string $field, $as=FALSE): QueryBuilderInterface { // Create the select string $this->state->appendSelectString(' MAX'.$this->_select($field, $as)); return $this; } /** * Selects the minimum value of a field from a query * * @param string $field * @param string|bool $as * @return QueryBuilderInterface */ public function selectMin(string $field, $as=FALSE): QueryBuilderInterface { // Create the select string $this->state->appendSelectString(' MIN'.$this->_select($field, $as)); return $this; } /** * Selects the average value of a field from a query * * @param string $field * @param string|bool $as * @return QueryBuilderInterface */ public function selectAvg(string $field, $as=FALSE): QueryBuilderInterface { // Create the select string $this->state->appendSelectString(' AVG'.$this->_select($field, $as)); return $this; } /** * Selects the sum of a field from a query * * @param string $field * @param string|bool $as * @return QueryBuilderInterface */ public function selectSum(string $field, $as=FALSE): QueryBuilderInterface { // Create the select string $this->state->appendSelectString(' SUM'.$this->_select($field, $as)); return $this; } /** * @todo implement * @param string $fields * @return $this */ public function returning(string $fields = '*'): QueryBuilderInterface { return $this; } /** * Adds the 'distinct' keyword to a query * * @return QueryBuilderInterface */ public function distinct(): QueryBuilderInterface { // Prepend the keyword to the select string $this->state->setSelectString(' DISTINCT' . $this->state->getSelectString()); return $this; } /** * Tell the database to give you the query plan instead of result set * * @return QueryBuilderInterface */ public function explain(): QueryBuilderInterface { $this->explain = TRUE; return $this; } /** * Specify the database table to select from * * @param string $tblname * @return QueryBuilderInterface */ public function from(string $tblname): QueryBuilderInterface { // Split identifiers on spaces $identArray = explode(' ', \mb_trim($tblname)); $identArray = array_map('\\mb_trim', $identArray); // Quote the identifiers $identArray[0] = $this->driver->quoteTable($identArray[0]); $identArray = $this->driver->quoteIdent($identArray); // Paste it back together $this->state->setFromString(implode(' ', $identArray)); return $this; } // -------------------------------------------------------------------------- // ! 'Like' methods // -------------------------------------------------------------------------- /** * Creates a Like clause in the sql statement * * @param string $field * @param mixed $val * @param string $pos * @return QueryBuilderInterface */ public function like(string $field, $val, string $pos='both'): QueryBuilderInterface { return $this->_like($field, $val, $pos); } /** * Generates an OR Like clause * * @param string $field * @param mixed $val * @param string $pos * @return QueryBuilderInterface */ public function orLike(string $field, $val, string $pos='both'): QueryBuilderInterface { return $this->_like($field, $val, $pos, 'LIKE', 'OR'); } /** * Generates a NOT LIKE clause * * @param string $field * @param mixed $val * @param string $pos * @return QueryBuilderInterface */ public function notLike(string $field, $val, string $pos='both'): QueryBuilderInterface { return $this->_like($field, $val, $pos, 'NOT LIKE'); } /** * Generates a OR NOT LIKE clause * * @param string $field * @param mixed $val * @param string $pos * @return QueryBuilderInterface */ public function orNotLike(string $field, $val, string $pos='both'): QueryBuilderInterface { return $this->_like($field, $val, $pos, 'NOT LIKE', 'OR'); } // -------------------------------------------------------------------------- // ! Having methods // -------------------------------------------------------------------------- /** * Generates a 'Having' clause * * @param mixed $key * @param mixed $val * @return QueryBuilderInterface */ public function having($key, $val=[]): QueryBuilderInterface { return $this->_having($key, $val); } /** * Generates a 'Having' clause prefixed with 'OR' * * @param mixed $key * @param mixed $val * @return QueryBuilderInterface */ public function orHaving($key, $val=[]): QueryBuilderInterface { return $this->_having($key, $val, 'OR'); } // -------------------------------------------------------------------------- // ! 'Where' methods // -------------------------------------------------------------------------- /** * Specify condition(s) in the where clause of a query * Note: this function works with key / value, or a * passed array with key / value pairs * * @param mixed $key * @param mixed $val * @param mixed $escape * @return QueryBuilderInterface */ public function where($key, $val=[], $escape=NULL): QueryBuilderInterface { return $this->_whereString($key, $val); } /** * Where clause prefixed with "OR" * * @param string $key * @param mixed $val * @return QueryBuilderInterface */ public function orWhere($key, $val=[]): QueryBuilderInterface { return $this->_whereString($key, $val, 'OR'); } /** * Where clause with 'IN' statement * * @param mixed $field * @param mixed $val * @return QueryBuilderInterface */ public function whereIn($field, $val=[]): QueryBuilderInterface { return $this->_whereIn($field, $val); } /** * Where in statement prefixed with "or" * * @param string $field * @param mixed $val * @return QueryBuilderInterface */ public function orWhereIn($field, $val=[]): QueryBuilderInterface { return $this->_whereIn($field, $val, 'IN', 'OR'); } /** * WHERE NOT IN (FOO) clause * * @param string $field * @param mixed $val * @return QueryBuilderInterface */ public function whereNotIn($field, $val=[]): QueryBuilderInterface { return $this->_whereIn($field, $val, 'NOT IN'); } /** * OR WHERE NOT IN (FOO) clause * * @param string $field * @param mixed $val * @return QueryBuilderInterface */ public function orWhereNotIn($field, $val=[]): QueryBuilderInterface { return $this->_whereIn($field, $val, 'NOT IN', 'OR'); } // -------------------------------------------------------------------------- // ! Other Query Modifier methods // -------------------------------------------------------------------------- /** * Sets values for inserts / updates / deletes * * @param mixed $key * @param mixed $val * @return QueryBuilderInterface */ public function set($key, $val = NULL): QueryBuilderInterface { if (is_scalar($key)) { $pairs = [$key => $val]; } else { $pairs = $key; } $keys = array_keys($pairs); $values = array_values($pairs); $this->state->appendSetArrayKeys($keys); $this->state->appendValues($values); // Use the keys of the array to make the insert/update string // Escape the field names $this->state->setSetArrayKeys( array_map([$this->driver, '_quote'], $this->state->getSetArrayKeys()) ); // Generate the "set" string $setString = implode('=?,', $this->state->getSetArrayKeys()); $setString .= '=?'; $this->state->setSetString($setString); return $this; } /** * Creates a join phrase in a compiled query * * @param string $table * @param string $condition * @param string $type * @return QueryBuilderInterface */ public function join(string $table, string $condition, string $type=''): QueryBuilderInterface { // Prefix and quote table name $table = explode(' ', mb_trim($table)); $table[0] = $this->driver->quoteTable($table[0]); $table = $this->driver->quoteIdent($table); $table = implode(' ', $table); // Parse out the join condition $parsedCondition = $this->parser->compileJoin($condition); $condition = $table . ' ON ' . $parsedCondition; $this->state->appendMap("\n" . strtoupper($type) . ' JOIN ', $condition, 'join'); return $this; } /** * Group the results by the selected field(s) * * @param mixed $field * @return QueryBuilderInterface */ public function groupBy($field): QueryBuilderInterface { if ( ! is_scalar($field)) { $newGroupArray = array_map([$this->driver, 'quoteIdent'], $field); $this->state->setGroupArray( array_merge($this->state->getGroupArray(), $newGroupArray) ); } else { $this->state->appendGroupArray($this->driver->quoteIdent($field)); } $this->state->setGroupString(' GROUP BY ' . implode(',', $this->state->getGroupArray())); return $this; } /** * Order the results by the selected field(s) * * @param string $field * @param string $type * @return QueryBuilderInterface */ public function orderBy(string $field, string $type=''): QueryBuilderInterface { // When ordering by random, do an ascending order if the driver // doesn't support random ordering if (stripos($type, 'rand') !== FALSE) { $rand = $this->driver->getSql()->random(); $type = $rand ?? 'ASC'; } // Set fields for later manipulation $field = $this->driver->quoteIdent($field); $this->state->setOrderArray($field, $type); $orderClauses = []; // Flatten key/val pairs into an array of space-separated pairs foreach($this->state->getOrderArray() as $k => $v) { $orderClauses[] = $k . ' ' . strtoupper($v); } // Set the final string $orderString = ! isset($rand) ? "\nORDER BY ".implode(', ', $orderClauses) : "\nORDER BY".$rand; $this->state->setOrderString($orderString); return $this; } /** * Set a limit on the current sql statement * * @param int $limit * @param int|bool $offset * @return QueryBuilderInterface */ public function limit(int $limit, $offset=FALSE): QueryBuilderInterface { $this->state->setLimit($limit); $this->state->setOffset($offset); return $this; } // -------------------------------------------------------------------------- // ! Query Grouping Methods // -------------------------------------------------------------------------- /** * Adds a paren to the current query for query grouping * * @return QueryBuilderInterface */ public function groupStart(): QueryBuilderInterface { $conj = empty($this->state->getQueryMap()) ? ' WHERE ' : ' '; $this->state->appendMap($conj, '(', 'group_start'); return $this; } /** * Adds a paren to the current query for query grouping, * prefixed with 'NOT' * * @return QueryBuilderInterface */ public function notGroupStart(): QueryBuilderInterface { $conj = empty($this->state->getQueryMap()) ? ' WHERE ' : ' AND '; $this->state->appendMap($conj, ' NOT (', 'group_start'); return $this; } /** * Adds a paren to the current query for query grouping, * prefixed with 'OR' * * @return QueryBuilderInterface */ public function orGroupStart(): QueryBuilderInterface { $this->state->appendMap('', ' OR (', 'group_start'); return $this; } /** * Adds a paren to the current query for query grouping, * prefixed with 'OR NOT' * * @return QueryBuilderInterface */ public function orNotGroupStart(): QueryBuilderInterface { $this->state->appendMap('', ' OR NOT (', 'group_start'); return $this; } /** * Ends a query group * * @return QueryBuilderInterface */ public function groupEnd(): QueryBuilderInterface { $this->state->appendMap('', ')', 'group_end'); return $this; } // -------------------------------------------------------------------------- // ! Query execution methods // -------------------------------------------------------------------------- /** * Select and retrieve all records from the current table, and/or * execute current compiled query * * @param string $table * @param int|bool $limit * @param int|bool $offset * @return PDOStatement */ public function get(string $table='', $limit=FALSE, $offset=FALSE): PDOStatement { // Set the table if ( ! empty($table)) { $this->from($table); } // Set the limit, if it exists if (\is_int($limit)) { $this->limit($limit, $offset); } return $this->_run('get', $table); } /** * Convenience method for get() with a where clause * * @param string $table * @param mixed $where * @param int|bool $limit * @param int|bool $offset * @return PDOStatement */ public function getWhere(string $table, $where=[], $limit=FALSE, $offset=FALSE): PDOStatement { // Create the where clause $this->where($where); // Return the result return $this->get($table, $limit, $offset); } /** * Retrieve the number of rows in the selected table * * @param string $table * @return int */ public function countAll(string $table): int { $sql = 'SELECT * FROM '.$this->driver->quoteTable($table); $res = $this->driver->query($sql); return (int) count($res->fetchAll()); } /** * Retrieve the number of results for the generated query - used * in place of the get() method * * @param string $table * @param boolean $reset * @return int */ public function countAllResults(string $table='', bool $reset = TRUE): int { // Set the table if ( ! empty($table)) { $this->from($table); } $result = $this->_run('get', $table, NULL, NULL, $reset); $rows = $result->fetchAll(); return (int) count($rows); } /** * Creates an insert clause, and executes it * * @param string $table * @param mixed $data * @return PDOStatement */ public function insert(string $table, $data=[]): PDOStatement { if ( ! empty($data)) { $this->set($data); } return $this->_run('insert', $table); } /** * Creates and executes a batch insertion query * * @param string $table * @param array $data * @return PDOStatement */ public function insertBatch(string $table, $data=[]): PDOStatement { // Get the generated values and sql string [$sql, $data] = $this->driver->insertBatch($table, $data); return $sql !== NULL ? $this->_run('', $table, $sql, $data) : NULL; } /** * Creates an update clause, and executes it * * @param string $table * @param mixed $data * @return PDOStatement */ public function update(string $table, $data=[]): PDOStatement { if ( ! empty($data)) { $this->set($data); } return $this->_run('update', $table); } /** * Creates a batch update, and executes it. * Returns the number of affected rows * * @param string $table * @param array $data * @param string $where * @return int|null */ public function updateBatch(string $table, array $data, string $where): ?int { if (empty($table) || empty($data) || empty($where)) { return NULL; } // Get the generated values and sql string [$sql, $data, $affectedRows] = $this->driver->updateBatch($table, $data, $where); $this->_run('', $table, $sql, $data); return $affectedRows; } /** * Deletes data from a table * * @param string $table * @param mixed $where * @return PDOStatement */ public function delete(string $table, $where=''): PDOStatement { // Set the where clause if ( ! empty($where)) { $this->where($where); } return $this->_run('delete', $table); } // -------------------------------------------------------------------------- // ! SQL Returning Methods // -------------------------------------------------------------------------- /** * Returns the generated 'select' sql query * * @param string $table * @param bool $reset * @return string */ public function getCompiledSelect(string $table='', bool $reset=TRUE): string { // Set the table if ( ! empty($table)) { $this->from($table); } return $this->_getCompile('select', $table, $reset); } /** * Returns the generated 'insert' sql query * * @param string $table * @param bool $reset * @return string */ public function getCompiledInsert(string $table, bool $reset=TRUE): string { return $this->_getCompile('insert', $table, $reset); } /** * Returns the generated 'update' sql query * * @param string $table * @param bool $reset * @return string */ public function getCompiledUpdate(string $table='', bool $reset=TRUE): string { return $this->_getCompile('update', $table, $reset); } /** * Returns the generated 'delete' sql query * * @param string $table * @param bool $reset * @return string */ public function getCompiledDelete(string $table='', bool $reset=TRUE): string { return $this->_getCompile('delete', $table, $reset); } // -------------------------------------------------------------------------- // ! Miscellaneous Methods // -------------------------------------------------------------------------- /** * Clear out the class variables, so the next query can be run * * @return void */ public function resetQuery(): void { $this->state = new State(); $this->explain = FALSE; } /** * Method to simplify select_ methods * * @param string $field * @param string|bool $as * @return string */ protected function _select(string $field, $as = FALSE): string { // Escape the identifiers $field = $this->driver->quoteIdent($field); if ( ! \is_string($as)) { // @codeCoverageIgnoreStart return $field; // @codeCoverageIgnoreEnd } $as = $this->driver->quoteIdent($as); return "({$field}) AS {$as} "; } /** * Helper function for returning sql strings * * @param string $type * @param string $table * @param bool $reset * @return string */ protected function _getCompile(string $type, string $table, bool $reset): string { $sql = $this->_compile($type, $table); // Reset the query builder for the next query if ($reset) { $this->resetQuery(); } return $sql; } /** * Simplify 'like' methods * * @param string $field * @param mixed $val * @param string $pos * @param string $like * @param string $conj * @return QueryBuilderInterface */ protected function _like(string $field, $val, string $pos, string $like='LIKE', string $conj='AND'): QueryBuilderInterface { $field = $this->driver->quoteIdent($field); // Add the like string into the order map $like = $field. " {$like} ?"; if ($pos === 'before') { $val = "%{$val}"; } elseif ($pos === 'after') { $val = "{$val}%"; } else { $val = "%{$val}%"; } $conj = empty($this->state->getQueryMap()) ? ' WHERE ' : " {$conj} "; $this->state->appendMap($conj, $like, 'like'); // Add to the values array $this->state->appendWhereValues($val); return $this; } /** * Simplify building having clauses * * @param mixed $key * @param mixed $values * @param string $conj * @return QueryBuilderInterface */ protected function _having($key, $values=[], string $conj='AND'): QueryBuilderInterface { $where = $this->_where($key, $values); // Create key/value placeholders foreach($where as $f => $val) { // Split each key by spaces, in case there // is an operator such as >, <, !=, etc. $fArray = explode(' ', trim($f)); $item = $this->driver->quoteIdent($fArray[0]); // Simple key value, or an operator $item .= (count($fArray) === 1) ? '=?' : " {$fArray[1]} ?"; // Put in the having map $this->state->appendHavingMap([ 'conjunction' => empty($this->state->getHavingMap()) ? ' HAVING ' : " {$conj} ", 'string' => $item ]); } return $this; } /** * Do all the redundant stuff for where/having type methods * * @param mixed $key * @param mixed $val * @return array */ protected function _where($key, $val=[]): array { $where = []; $pairs = []; if (is_scalar($key)) { $pairs[$key] = $val; } else { $pairs = $key; } foreach($pairs as $k => $v) { $where[$k] = $v; $this->state->appendWhereValues($v); } return $where; } /** * Simplify generating where string * * @param mixed $key * @param mixed $values * @param string $defaultConj * @return QueryBuilderInterface */ protected function _whereString($key, $values=[], string $defaultConj='AND'): QueryBuilderInterface { // Create key/value placeholders foreach($this->_where($key, $values) as $f => $val) { $queryMap = $this->state->getQueryMap(); // Split each key by spaces, in case there // is an operator such as >, <, !=, etc. $fArray = explode(' ', trim($f)); $item = $this->driver->quoteIdent($fArray[0]); // Simple key value, or an operator $item .= (count($fArray) === 1) ? '=?' : " {$fArray[1]} ?"; $lastItem = end($queryMap); // Determine the correct conjunction $conjunctionList = array_column($queryMap, 'conjunction'); if (empty($queryMap) || ( ! regexInArray($conjunctionList, "/^ ?\n?WHERE/i"))) { $conj = "\nWHERE "; } elseif ($lastItem['type'] === 'group_start') { $conj = ''; } else { $conj = " {$defaultConj} "; } $this->state->appendMap($conj, $item, 'where'); } return $this; } /** * Simplify where_in methods * * @param mixed $key * @param mixed $val * @param string $in - The (not) in fragment * @param string $conj - The where in conjunction * @return QueryBuilderInterface */ protected function _whereIn($key, $val=[], string $in='IN', string $conj='AND'): QueryBuilderInterface { $key = $this->driver->quoteIdent($key); $params = array_fill(0, count($val), '?'); $this->state->appendWhereValues($val); $conjunction = empty($this->state->getQueryMap()) ? ' WHERE ' : " {$conj} "; $str = $key . " {$in} (".implode(',', $params).') '; $this->state->appendMap($conjunction, $str, 'where_in'); return $this; } /** * Executes the compiled query * * @param string $type * @param string $table * @param string $sql * @param array|null $vals * @param boolean $reset * @return PDOStatement */ protected function _run(string $type, string $table, string $sql=NULL, array $vals=NULL, bool $reset=TRUE): PDOStatement { if ($sql === NULL) { $sql = $this->_compile($type, $table); } if ($vals === NULL) { $vals = array_merge($this->state->getValues(), $this->state->getWhereValues()); } $startTime = microtime(TRUE); $res = empty($vals) ? $this->driver->query($sql) : $this->driver->prepareExecute($sql, $vals); $endTime = microtime(TRUE); $totalTime = number_format($endTime - $startTime, 5); // Add this query to the list of executed queries $this->_appendQuery($vals, $sql, (int) $totalTime); // Reset class state for next query if ($reset) { $this->resetQuery(); } return $res; } /** * Convert the prepared statement into readable sql * * @param array $values * @param string $sql * @param int $totalTime * @return void */ protected function _appendQuery(array $values, string $sql, int $totalTime): void { $evals = \is_array($values) ? $values : []; $esql = str_replace('?', '%s', $sql); // Quote string values foreach($evals as &$v) { $v = ( ! is_numeric($v)) ? htmlentities($this->driver->quote($v), ENT_NOQUOTES, 'utf-8') : $v; } unset($v); // Add the query onto the array of values to pass // as arguments to sprintf array_unshift($evals, $esql); // Add the interpreted query to the list of executed queries $this->queries[] = [ 'time' => $totalTime, 'sql' => sprintf(...$evals) ]; $this->queries['total_time'] += $totalTime; // Set the last query to get rowcounts properly $this->driver->setLastQuery($sql); } /** * Sub-method for generating sql strings * * @codeCoverageIgnore * @param string $type * @param string $table * @return string */ protected function _compileType(string $type='', string $table=''): string { $setArrayKeys = $this->state->getSetArrayKeys(); switch($type) { case 'insert': $paramCount = count($setArrayKeys); $params = array_fill(0, $paramCount, '?'); $sql = "INSERT INTO {$table} (" . implode(',', $setArrayKeys) . ")\nVALUES (".implode(',', $params).')'; break; case 'update': $setString = $this->state->getSetString(); $sql = "UPDATE {$table}\nSET {$setString}"; break; case 'delete': $sql = "DELETE FROM {$table}"; break; // Get queries default: $fromString = $this->state->getFromString(); $selectString = $this->state->getSelectString(); $sql = "SELECT * \nFROM {$fromString}"; // Set the select string if ( ! empty($selectString)) { // Replace the star with the selected fields $sql = str_replace('*', $selectString, $sql); } break; } return $sql; } /** * String together the sql statements for sending to the db * * @param string $type * @param string $table * @return string */ protected function _compile(string $type='', string $table=''): string { // Get the base clause for the query $sql = $this->_compileType($type, $this->driver->quoteTable($table)); $clauses = [ 'queryMap', 'groupString', 'orderString', 'havingMap', ]; // Set each type of subclause foreach($clauses as $clause) { $func = 'get' . ucFirst($clause); $param = $this->state->$func(); if (\is_array($param)) { foreach($param as $q) { $sql .= $q['conjunction'] . $q['string']; } } else { $sql .= $param; } } // Set the limit via the class variables $limit = $this->state->getLimit(); if (is_numeric($limit)) { $sql = $this->driver->getSql()->limit($sql, $limit, $this->state->getOffset()); } // See if the query plan, rather than the // query data should be returned if ($this->explain === TRUE) { $sql = $this->driver->getSql()->explain($sql); } return $sql; } } |