338 lines
9.2 KiB
PHP
338 lines
9.2 KiB
PHP
<?php
|
|
/**
|
|
* @author Bernhard Posselt <dev@bernhard-posselt.com>
|
|
* @author Lukas Reschke <lukas@owncloud.com>
|
|
* @author Morris Jobke <hey@morrisjobke.de>
|
|
* @author Thomas Müller <thomas.mueller@tmit.eu>
|
|
*
|
|
* @copyright Copyright (c) 2015, ownCloud, Inc.
|
|
* @license AGPL-3.0
|
|
*
|
|
* This code is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License, version 3,
|
|
* as published by the Free Software Foundation.
|
|
*
|
|
* 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, version 3,
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>
|
|
*
|
|
*/
|
|
|
|
|
|
namespace OCP\AppFramework\Db;
|
|
|
|
use OCP\IDBConnection;
|
|
use OCP\IDb;
|
|
|
|
|
|
/**
|
|
* Simple parent class for inheriting your data access layer from. This class
|
|
* may be subject to change in the future
|
|
*/
|
|
abstract class Mapper {
|
|
|
|
protected $tableName;
|
|
protected $entityClass;
|
|
protected $db;
|
|
|
|
/**
|
|
* @param IDBConnection $db Instance of the Db abstraction layer
|
|
* @param string $tableName the name of the table. set this to allow entity
|
|
* @param string $entityClass the name of the entity that the sql should be
|
|
* mapped to queries without using sql
|
|
*/
|
|
public function __construct(IDBConnection $db, $tableName, $entityClass=null){
|
|
$this->db = $db;
|
|
$this->tableName = '*PREFIX*' . $tableName;
|
|
|
|
// if not given set the entity name to the class without the mapper part
|
|
// cache it here for later use since reflection is slow
|
|
if($entityClass === null) {
|
|
$this->entityClass = str_replace('Mapper', '', get_class($this));
|
|
} else {
|
|
$this->entityClass = $entityClass;
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* @return string the table name
|
|
*/
|
|
public function getTableName(){
|
|
return $this->tableName;
|
|
}
|
|
|
|
|
|
/**
|
|
* Deletes an entity from the table
|
|
* @param Entity $entity the entity that should be deleted
|
|
* @return Entity the deleted entity
|
|
*/
|
|
public function delete(Entity $entity){
|
|
$sql = 'DELETE FROM `' . $this->tableName . '` WHERE `id` = ?';
|
|
$stmt = $this->execute($sql, [$entity->getId()]);
|
|
$stmt->closeCursor();
|
|
return $entity;
|
|
}
|
|
|
|
|
|
/**
|
|
* Creates a new entry in the db from an entity
|
|
* @param Entity $entity the entity that should be created
|
|
* @return Entity the saved entity with the set id
|
|
*/
|
|
public function insert(Entity $entity){
|
|
// get updated fields to save, fields have to be set using a setter to
|
|
// be saved
|
|
$properties = $entity->getUpdatedFields();
|
|
$values = '';
|
|
$columns = '';
|
|
$params = [];
|
|
|
|
// build the fields
|
|
$i = 0;
|
|
foreach($properties as $property => $updated) {
|
|
$column = $entity->propertyToColumn($property);
|
|
$getter = 'get' . ucfirst($property);
|
|
|
|
$columns .= '`' . $column . '`';
|
|
$values .= '?';
|
|
|
|
// only append colon if there are more entries
|
|
if($i < count($properties)-1){
|
|
$columns .= ',';
|
|
$values .= ',';
|
|
}
|
|
|
|
$params[] = $entity->$getter();
|
|
$i++;
|
|
|
|
}
|
|
|
|
$sql = 'INSERT INTO `' . $this->tableName . '`(' .
|
|
$columns . ') VALUES(' . $values . ')';
|
|
|
|
$stmt = $this->execute($sql, $params);
|
|
|
|
$entity->setId((int) $this->db->lastInsertId($this->tableName));
|
|
|
|
$stmt->closeCursor();
|
|
|
|
return $entity;
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
* Updates an entry in the db from an entity
|
|
* @throws \InvalidArgumentException if entity has no id
|
|
* @param Entity $entity the entity that should be created
|
|
* @return Entity the saved entity with the set id
|
|
*/
|
|
public function update(Entity $entity){
|
|
// if entity wasn't changed it makes no sense to run a db query
|
|
$properties = $entity->getUpdatedFields();
|
|
if(count($properties) === 0) {
|
|
return $entity;
|
|
}
|
|
|
|
// entity needs an id
|
|
$id = $entity->getId();
|
|
if($id === null){
|
|
throw new \InvalidArgumentException(
|
|
'Entity which should be updated has no id');
|
|
}
|
|
|
|
// get updated fields to save, fields have to be set using a setter to
|
|
// be saved
|
|
// do not update the id field
|
|
unset($properties['id']);
|
|
|
|
$columns = '';
|
|
$params = [];
|
|
|
|
// build the fields
|
|
$i = 0;
|
|
foreach($properties as $property => $updated) {
|
|
|
|
$column = $entity->propertyToColumn($property);
|
|
$getter = 'get' . ucfirst($property);
|
|
|
|
$columns .= '`' . $column . '` = ?';
|
|
|
|
// only append colon if there are more entries
|
|
if($i < count($properties)-1){
|
|
$columns .= ',';
|
|
}
|
|
|
|
$params[] = $entity->$getter();
|
|
$i++;
|
|
}
|
|
|
|
$sql = 'UPDATE `' . $this->tableName . '` SET ' .
|
|
$columns . ' WHERE `id` = ?';
|
|
$params[] = $id;
|
|
|
|
$stmt = $this->execute($sql, $params);
|
|
$stmt->closeCursor();
|
|
|
|
return $entity;
|
|
}
|
|
|
|
/**
|
|
* Checks if an array is associative
|
|
* @param array $array
|
|
* @return bool true if associative
|
|
*/
|
|
private function isAssocArray(array $array) {
|
|
return array_values($array) !== $array;
|
|
}
|
|
|
|
/**
|
|
* Returns the correct PDO constant based on the value type
|
|
* @param $value
|
|
* @return PDO constant
|
|
*/
|
|
private function getPDOType($value) {
|
|
switch (gettype($value)) {
|
|
case 'integer':
|
|
return \PDO::PARAM_INT;
|
|
case 'boolean':
|
|
return \PDO::PARAM_BOOL;
|
|
default:
|
|
return \PDO::PARAM_STR;
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* Runs an sql query
|
|
* @param string $sql the prepare string
|
|
* @param array $params the params which should replace the ? in the sql query
|
|
* @param int $limit the maximum number of rows
|
|
* @param int $offset from which row we want to start
|
|
* @return \PDOStatement the database query result
|
|
*/
|
|
protected function execute($sql, array $params=[], $limit=null, $offset=null){
|
|
if ($this->db instanceof IDb) {
|
|
$query = $this->db->prepareQuery($sql, $limit, $offset);
|
|
} else {
|
|
$query = $this->db->prepare($sql, $limit, $offset);
|
|
}
|
|
|
|
if ($this->isAssocArray($params)) {
|
|
foreach ($params as $key => $param) {
|
|
$pdoConstant = $this->getPDOType($param);
|
|
$query->bindValue($key, $param, $pdoConstant);
|
|
}
|
|
} else {
|
|
$index = 1; // bindParam is 1 indexed
|
|
foreach ($params as $param) {
|
|
$pdoConstant = $this->getPDOType($param);
|
|
$query->bindValue($index, $param, $pdoConstant);
|
|
$index++;
|
|
}
|
|
}
|
|
|
|
$result = $query->execute();
|
|
|
|
// this is only for backwards compatibility reasons and can be removed
|
|
// in owncloud 10. IDb returns a StatementWrapper from execute, PDO,
|
|
// Doctrine and IDbConnection don't so this needs to be done in order
|
|
// to stay backwards compatible for the things that rely on the
|
|
// StatementWrapper being returned
|
|
if ($result instanceof \OC_DB_StatementWrapper) {
|
|
return $result;
|
|
}
|
|
|
|
return $query;
|
|
}
|
|
|
|
|
|
/**
|
|
* Returns an db result and throws exceptions when there are more or less
|
|
* results
|
|
* @see findEntity
|
|
* @param string $sql the sql query
|
|
* @param array $params the parameters of the sql query
|
|
* @param int $limit the maximum number of rows
|
|
* @param int $offset from which row we want to start
|
|
* @throws DoesNotExistException if the item does not exist
|
|
* @throws MultipleObjectsReturnedException if more than one item exist
|
|
* @return array the result as row
|
|
*/
|
|
protected function findOneQuery($sql, array $params=[], $limit=null, $offset=null){
|
|
$stmt = $this->execute($sql, $params, $limit, $offset);
|
|
$row = $stmt->fetch();
|
|
|
|
if($row === false || $row === null){
|
|
$stmt->closeCursor();
|
|
throw new DoesNotExistException('No matching entry found');
|
|
}
|
|
$row2 = $stmt->fetch();
|
|
$stmt->closeCursor();
|
|
//MDB2 returns null, PDO and doctrine false when no row is available
|
|
if( ! ($row2 === false || $row2 === null )) {
|
|
throw new MultipleObjectsReturnedException('More than one result');
|
|
} else {
|
|
return $row;
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* Creates an entity from a row. Automatically determines the entity class
|
|
* from the current mapper name (MyEntityMapper -> MyEntity)
|
|
* @param array $row the row which should be converted to an entity
|
|
* @return Entity the entity
|
|
*/
|
|
protected function mapRowToEntity($row) {
|
|
return call_user_func($this->entityClass .'::fromRow', $row);
|
|
}
|
|
|
|
|
|
/**
|
|
* Runs a sql query and returns an array of entities
|
|
* @param string $sql the prepare string
|
|
* @param array $params the params which should replace the ? in the sql query
|
|
* @param int $limit the maximum number of rows
|
|
* @param int $offset from which row we want to start
|
|
* @return array all fetched entities
|
|
*/
|
|
protected function findEntities($sql, array $params=[], $limit=null, $offset=null) {
|
|
$stmt = $this->execute($sql, $params, $limit, $offset);
|
|
|
|
$entities = [];
|
|
|
|
while($row = $stmt->fetch()){
|
|
$entities[] = $this->mapRowToEntity($row);
|
|
}
|
|
|
|
$stmt->closeCursor();
|
|
|
|
return $entities;
|
|
}
|
|
|
|
|
|
/**
|
|
* Returns an db result and throws exceptions when there are more or less
|
|
* results
|
|
* @param string $sql the sql query
|
|
* @param array $params the parameters of the sql query
|
|
* @param int $limit the maximum number of rows
|
|
* @param int $offset from which row we want to start
|
|
* @throws DoesNotExistException if the item does not exist
|
|
* @throws MultipleObjectsReturnedException if more than one item exist
|
|
* @return Entity the entity
|
|
*/
|
|
protected function findEntity($sql, array $params=[], $limit=null, $offset=null){
|
|
return $this->mapRowToEntity($this->findOneQuery($sql, $params, $limit, $offset));
|
|
}
|
|
|
|
|
|
}
|