Convert colorMapper to QBMapper

This commit is contained in:
Matias De lellis
2021-09-20 20:14:17 -03:00
parent 443f6bf144
commit 3365a914c7
3 changed files with 33 additions and 16 deletions

View File

@@ -5,6 +5,10 @@ use JsonSerializable;
use OCP\AppFramework\Db\Entity;
/**
* @method void setColor(string $color)
* @method strinf getColor()
*/
class Color extends Entity implements JsonSerializable {
protected $color;

View File

@@ -2,36 +2,42 @@
namespace OCA\QuickNotes\Db;
use OCP\IDBConnection;
use OCP\AppFramework\Db\Mapper;
use OCP\AppFramework\Db\QBMapper;
use OCP\AppFramework\Db\DoesNotExistException;
use OCP\DB\QueryBuilder\IQueryBuilder;
use OCA\QuickNotes\Db\Color;
class ColorMapper extends Mapper {
class ColorMapper extends QBMapper {
public function __construct(IDBConnection $db) {
parent::__construct($db, 'quicknotes_colors', '\OCA\QuickNotes\Db\Color');
parent::__construct($db, 'quicknotes_colors', Color::class);
}
public function find($id): Color {
$sql = 'SELECT * FROM *PREFIX*quicknotes_colors WHERE id = ?';
return $this->findEntity($sql, [$id]);
}
public function findAll(): array {
$sql = 'SELECT * FROM *PREFIX*quicknotes_colors';
return $this->findEntities($sql, []);
public function find(int $id): Color {
$qb = $this->db->getQueryBuilder();
$qb->select('*')
->from($this->tableName)
->where(
$qb->expr()->eq('id', $qb->createNamedParameter($id, IQueryBuilder::PARAM_INT))
);
return $this->findEntity($qb);
}
public function findByColor(string $color): Color {
$sql = 'SELECT * FROM *PREFIX*quicknotes_colors WHERE color = ?';
return $this->findEntity($sql, [$color]);
$qb = $this->db->getQueryBuilder();
$qb->select('*')
->from($this->tableName)
->where(
$qb->expr()->eq('color', $qb->createNamedParameter($color, IQueryBuilder::PARAM_STRING))
);
return $this->findEntity($qb);
}
public function colorExists(string $color): bool {
$sql = 'SELECT * FROM *PREFIX*quicknotes_colors WHERE color = ?';
try {
$this->findEntity($sql, [$color]);
$this->findByColor($color);
} catch (DoesNotExistException $e) {
return false;
}

View File

@@ -19,12 +19,19 @@ class NoteShare extends Entity implements JsonSerializable {
protected $sharedUser;
protected $sharedGroup;
protected $userId;
public function setUserId (string $userId): void {
$this->userId = $userId;
}
public function jsonSerialize() {
return [
'id' => $this->id,
'note_id' => $this->noteId,
'shared_user' => $this->sharedUser,
'shared_group' => $this->sharedGroup
'shared_group' => $this->sharedGroup,
'user_id' => $this->userId
];
}