'lib' rewritten.
This commit is contained in:
Marcin Łojewski
267
lib/Query/DataQuery.php
Normal file
267
lib/Query/DataQuery.php
Normal file
@@ -0,0 +1,267 @@
|
||||
<?php
|
||||
/**
|
||||
* Nextcloud - user_sql
|
||||
*
|
||||
* @copyright 2018 Marcin Łojewski <dev@mlojewski.me>
|
||||
* @author Marcin Łojewski <dev@mlojewski.me>
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU Affero General Public License as
|
||||
* published by the Free Software Foundation, either version 3 of the
|
||||
* License, or (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU Affero General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU Affero General Public License
|
||||
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
namespace OCA\UserSQL\Query;
|
||||
|
||||
use Doctrine\DBAL\Driver\Statement;
|
||||
use OC\DB\Connection;
|
||||
use OC\DB\ConnectionFactory;
|
||||
use OCA\UserSQL\Constant\DB;
|
||||
use OCA\UserSQL\Constant\Query;
|
||||
use OCA\UserSQL\Properties;
|
||||
use OCP\ILogger;
|
||||
|
||||
/**
|
||||
* Used to query a database.
|
||||
*
|
||||
* @author Marcin Łojewski <dev@mlojewski.me>
|
||||
*/
|
||||
class DataQuery
|
||||
{
|
||||
/**
|
||||
* @var string The application name.
|
||||
*/
|
||||
private $appName;
|
||||
/**
|
||||
* @var ILogger The logger instance.
|
||||
*/
|
||||
private $logger;
|
||||
/**
|
||||
* @var Properties The properties array.
|
||||
*/
|
||||
private $properties;
|
||||
/**
|
||||
* @var QueryProvider The query provider.
|
||||
*/
|
||||
private $queryProvider;
|
||||
/**
|
||||
* @var Connection The database connection.
|
||||
*/
|
||||
private $connection;
|
||||
|
||||
/**
|
||||
* The class constructor.
|
||||
*
|
||||
* @param string $AppName The application name.
|
||||
* @param ILogger $logger The logger instance.
|
||||
* @param Properties $properties The properties array.
|
||||
* @param QueryProvider $queryProvider The query provider.
|
||||
*/
|
||||
public function __construct(
|
||||
$AppName, ILogger $logger, Properties $properties,
|
||||
QueryProvider $queryProvider
|
||||
) {
|
||||
$this->appName = $AppName;
|
||||
$this->logger = $logger;
|
||||
$this->properties = $properties;
|
||||
$this->queryProvider = $queryProvider;
|
||||
$this->connection = false;
|
||||
}
|
||||
|
||||
/**
|
||||
* Execute an update query.
|
||||
*
|
||||
* @param string $queryName The query name.
|
||||
* @param array $params The query parameters.
|
||||
*
|
||||
* @see Query
|
||||
* @return bool TRUE on success, FALSE otherwise.
|
||||
*/
|
||||
public function update($queryName, $params = [])
|
||||
{
|
||||
return $this->execQuery($queryName, $params) !== false;
|
||||
}
|
||||
|
||||
/**
|
||||
* Run a given query and return the result.
|
||||
*
|
||||
* @param string $queryName The query to execute.
|
||||
* @param array $params The query parameters to bind.
|
||||
* @param int $limit Results limit. Defaults to -1 (no limit).
|
||||
* @param int $offset Results offset. Defaults to 0.
|
||||
*
|
||||
* @return Statement|bool Result of query or FALSE on failure.
|
||||
*/
|
||||
private function execQuery(
|
||||
$queryName, $params = [], $limit = -1, $offset = 0
|
||||
) {
|
||||
if ($this->connection === false) {
|
||||
$this->connectToDatabase();
|
||||
}
|
||||
|
||||
$query = $this->queryProvider[$queryName];
|
||||
$result = $this->connection->prepare($query, $limit, $offset);
|
||||
|
||||
foreach ($params as $param => $value) {
|
||||
$result->bindValue(":" . $param, $value);
|
||||
}
|
||||
|
||||
$this->logger->debug(
|
||||
"Executing query:" . $query . ", " . implode(",", $params),
|
||||
["app" => $this->appName]
|
||||
);
|
||||
|
||||
if ($result->execute() !== true) {
|
||||
$error = $result->errorInfo();
|
||||
$this->logger->error(
|
||||
"Could not execute the query: " . implode(", ", $error),
|
||||
["app" => $this->appName]
|
||||
);
|
||||
return false;
|
||||
}
|
||||
|
||||
return $result;
|
||||
}
|
||||
|
||||
/**
|
||||
* Connect to the database using Nextcloud's DBAL.
|
||||
*/
|
||||
private function connectToDatabase()
|
||||
{
|
||||
$connectionFactory = new ConnectionFactory(
|
||||
\OC::$server->getSystemConfig()
|
||||
);
|
||||
|
||||
$parameters = array(
|
||||
"host" => $this->properties[DB::HOSTNAME],
|
||||
"password" => $this->properties[DB::PASSWORD],
|
||||
"user" => $this->properties[DB::USERNAME],
|
||||
"dbname" => $this->properties[DB::DATABASE],
|
||||
"tablePrefix" => ""
|
||||
);
|
||||
|
||||
$this->connection = $connectionFactory->getConnection(
|
||||
$this->properties[DB::DRIVER], $parameters
|
||||
);
|
||||
|
||||
$this->logger->debug(
|
||||
"Database connection established.", ["app" => $this->appName]
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
* Fetch a value from the first row and the first column which
|
||||
* the given query returns. Empty result set is consider to be a failure.
|
||||
*
|
||||
* @param string $queryName The query to execute.
|
||||
* @param array $params The query parameters to bind.
|
||||
* @param bool $failure Value returned on database query failure.
|
||||
* Defaults to FALSE.
|
||||
*
|
||||
* @return array|bool Queried value or $failure value on failure.
|
||||
*/
|
||||
public function queryValue($queryName, $params = [], $failure = false)
|
||||
{
|
||||
$result = $this->execQuery($queryName, $params);
|
||||
if ($result === false) {
|
||||
return false;
|
||||
}
|
||||
|
||||
$row = $result->fetch(\PDO::FETCH_COLUMN);
|
||||
if ($row === false) {
|
||||
return $failure;
|
||||
}
|
||||
|
||||
return $row;
|
||||
}
|
||||
|
||||
/**
|
||||
* Fetch values from the first column which the given query returns.
|
||||
*
|
||||
* @param string $queryName The query to execute.
|
||||
* @param array $params The query parameters to bind.
|
||||
* @param int $limit Results limit. Defaults to -1 (no limit).
|
||||
* @param int $offset Results offset. Defaults to 0.
|
||||
*
|
||||
* @return array|bool Queried column or FALSE on failure.
|
||||
*/
|
||||
public function queryColumn(
|
||||
$queryName, $params = [], $limit = -1, $offset = 0
|
||||
) {
|
||||
$result = $this->execQuery($queryName, $params, $limit, $offset);
|
||||
if ($result === false) {
|
||||
return false;
|
||||
}
|
||||
|
||||
$column = $result->fetchAll(\PDO::FETCH_COLUMN);
|
||||
return $column;
|
||||
}
|
||||
|
||||
/**
|
||||
* Fetch entity returned by the given query.
|
||||
*
|
||||
* @param string $queryName The query to execute.
|
||||
* @param string $entityClass The entity class name.
|
||||
* @param array $params The query parameters to bind.
|
||||
*
|
||||
* @return mixed|null The queried entity, NULL if it does not exists or
|
||||
* FALSE on failure.
|
||||
*/
|
||||
public function queryEntity($queryName, $entityClass, $params = [])
|
||||
{
|
||||
$result = $this->execQuery($queryName, $params);
|
||||
if ($result === false) {
|
||||
return false;
|
||||
}
|
||||
|
||||
$result->setFetchMode(\PDO::FETCH_CLASS, $entityClass);
|
||||
$entity = $result->fetch();
|
||||
|
||||
if ($entity === false) {
|
||||
return null;
|
||||
}
|
||||
|
||||
if (empty($entity) === true) {
|
||||
$this->logger->debug(
|
||||
"Empty result for query: " . $queryName,
|
||||
["app" => $this->appName]
|
||||
);
|
||||
return null;
|
||||
}
|
||||
|
||||
return $entity;
|
||||
}
|
||||
|
||||
/**
|
||||
* Fetch entities returned by the given query.
|
||||
*
|
||||
* @param string $queryName The query to execute.
|
||||
* @param string $entityClass The entity class name.
|
||||
* @param array $params The query parameters to bind.
|
||||
* @param int $limit Results limit. Defaults to -1 (no limit).
|
||||
* @param int $offset Results offset. Defaults to 0.
|
||||
*
|
||||
* @return mixed|null The queried entities or FALSE on failure.
|
||||
*/
|
||||
public function queryEntities(
|
||||
$queryName, $entityClass, $params = [], $limit = -1, $offset = 0
|
||||
) {
|
||||
$result = $this->execQuery($queryName, $params, $limit, $offset);
|
||||
if ($result === false) {
|
||||
return false;
|
||||
}
|
||||
|
||||
$result->setFetchMode(\PDO::FETCH_CLASS, $entityClass);
|
||||
$entities = $result->fetchAll();
|
||||
|
||||
return $entities;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user