Use the uid_lower column
This can use a proper index Signed-off-by: Roeland Jago Douma <roeland@famdouma.nl>
This commit is contained in:
parent
7a7d424e91
commit
8edbeb159e
|
@ -58,6 +58,7 @@ declare(strict_types=1);
|
||||||
namespace OC\User;
|
namespace OC\User;
|
||||||
|
|
||||||
use OC\Cache\CappedMemoryCache;
|
use OC\Cache\CappedMemoryCache;
|
||||||
|
use OCP\IDBConnection;
|
||||||
use OCP\User\Backend\ABackend;
|
use OCP\User\Backend\ABackend;
|
||||||
use OCP\User\Backend\ICheckPasswordBackend;
|
use OCP\User\Backend\ICheckPasswordBackend;
|
||||||
use OCP\User\Backend\ICountUsersBackend;
|
use OCP\User\Backend\ICountUsersBackend;
|
||||||
|
@ -87,6 +88,9 @@ class Database extends ABackend
|
||||||
/** @var EventDispatcher */
|
/** @var EventDispatcher */
|
||||||
private $eventDispatcher;
|
private $eventDispatcher;
|
||||||
|
|
||||||
|
/** @var IDBConnection */
|
||||||
|
private $dbConn;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* \OC\User\Database constructor.
|
* \OC\User\Database constructor.
|
||||||
*
|
*
|
||||||
|
@ -97,6 +101,15 @@ class Database extends ABackend
|
||||||
$this->eventDispatcher = $eventDispatcher ? $eventDispatcher : \OC::$server->getEventDispatcher();
|
$this->eventDispatcher = $eventDispatcher ? $eventDispatcher : \OC::$server->getEventDispatcher();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* FIXME: This function should not be required!
|
||||||
|
*/
|
||||||
|
private function fixDI() {
|
||||||
|
if ($this->dbConn === null) {
|
||||||
|
$this->dbConn = \OC::$server->getDatabaseConnection();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Create a new user
|
* Create a new user
|
||||||
*
|
*
|
||||||
|
@ -108,15 +121,21 @@ class Database extends ABackend
|
||||||
* itself, not in its subclasses.
|
* itself, not in its subclasses.
|
||||||
*/
|
*/
|
||||||
public function createUser(string $uid, string $password): bool {
|
public function createUser(string $uid, string $password): bool {
|
||||||
|
$this->fixDI();
|
||||||
|
|
||||||
if (!$this->userExists($uid)) {
|
if (!$this->userExists($uid)) {
|
||||||
$event = new GenericEvent($password);
|
$event = new GenericEvent($password);
|
||||||
$this->eventDispatcher->dispatch('OCP\PasswordPolicy::validate', $event);
|
$this->eventDispatcher->dispatch('OCP\PasswordPolicy::validate', $event);
|
||||||
$query = \OC_DB::prepare('INSERT INTO `*PREFIX*users` ( `uid`, `password` ) VALUES( ?, ? )');
|
|
||||||
try {
|
$qb = $this->dbConn->getQueryBuilder();
|
||||||
$result = $query->execute([$uid, \OC::$server->getHasher()->hash($password)]);
|
$qb->insert('users')
|
||||||
} catch (\Exception $e) {
|
->values([
|
||||||
$result = false;
|
'uid' => $qb->createNamedParameter($uid),
|
||||||
}
|
'password' => $qb->createNamedParameter(\OC::$server->getHasher()->hash($password)),
|
||||||
|
'uid_lower' => $qb->createNamedParameter(mb_strtolower($uid)),
|
||||||
|
]);
|
||||||
|
|
||||||
|
$result = $qb->execute();
|
||||||
|
|
||||||
// Clear cache
|
// Clear cache
|
||||||
unset($this->cache[$uid]);
|
unset($this->cache[$uid]);
|
||||||
|
@ -136,6 +155,8 @@ class Database extends ABackend
|
||||||
* Deletes a user
|
* Deletes a user
|
||||||
*/
|
*/
|
||||||
public function deleteUser($uid) {
|
public function deleteUser($uid) {
|
||||||
|
$this->fixDI();
|
||||||
|
|
||||||
// Delete user-group-relation
|
// Delete user-group-relation
|
||||||
$query = \OC_DB::prepare('DELETE FROM `*PREFIX*users` WHERE `uid` = ?');
|
$query = \OC_DB::prepare('DELETE FROM `*PREFIX*users` WHERE `uid` = ?');
|
||||||
$result = $query->execute([$uid]);
|
$result = $query->execute([$uid]);
|
||||||
|
@ -157,6 +178,8 @@ class Database extends ABackend
|
||||||
* Change the password of a user
|
* Change the password of a user
|
||||||
*/
|
*/
|
||||||
public function setPassword(string $uid, string $password): bool {
|
public function setPassword(string $uid, string $password): bool {
|
||||||
|
$this->fixDI();
|
||||||
|
|
||||||
if ($this->userExists($uid)) {
|
if ($this->userExists($uid)) {
|
||||||
$event = new GenericEvent($password);
|
$event = new GenericEvent($password);
|
||||||
$this->eventDispatcher->dispatch('OCP\PasswordPolicy::validate', $event);
|
$this->eventDispatcher->dispatch('OCP\PasswordPolicy::validate', $event);
|
||||||
|
@ -179,6 +202,8 @@ class Database extends ABackend
|
||||||
* Change the display name of a user
|
* Change the display name of a user
|
||||||
*/
|
*/
|
||||||
public function setDisplayName(string $uid, string $displayName): bool {
|
public function setDisplayName(string $uid, string $displayName): bool {
|
||||||
|
$this->fixDI();
|
||||||
|
|
||||||
if ($this->userExists($uid)) {
|
if ($this->userExists($uid)) {
|
||||||
$query = \OC_DB::prepare('UPDATE `*PREFIX*users` SET `displayname` = ? WHERE LOWER(`uid`) = LOWER(?)');
|
$query = \OC_DB::prepare('UPDATE `*PREFIX*users` SET `displayname` = ? WHERE LOWER(`uid`) = LOWER(?)');
|
||||||
$query->execute([$displayName, $uid]);
|
$query->execute([$displayName, $uid]);
|
||||||
|
@ -210,9 +235,9 @@ class Database extends ABackend
|
||||||
* @return array an array of all displayNames (value) and the corresponding uids (key)
|
* @return array an array of all displayNames (value) and the corresponding uids (key)
|
||||||
*/
|
*/
|
||||||
public function getDisplayNames($search = '', $limit = null, $offset = null) {
|
public function getDisplayNames($search = '', $limit = null, $offset = null) {
|
||||||
$connection = \OC::$server->getDatabaseConnection();
|
$this->fixDI();
|
||||||
|
|
||||||
$query = $connection->getQueryBuilder();
|
$query = $this->dbConn->getQueryBuilder();
|
||||||
|
|
||||||
$query->select('uid', 'displayname')
|
$query->select('uid', 'displayname')
|
||||||
->from('users', 'u')
|
->from('users', 'u')
|
||||||
|
@ -222,9 +247,9 @@ class Database extends ABackend
|
||||||
$query->expr()->eq('configkey', $query->expr()->literal('email')))
|
$query->expr()->eq('configkey', $query->expr()->literal('email')))
|
||||||
)
|
)
|
||||||
// sqlite doesn't like re-using a single named parameter here
|
// sqlite doesn't like re-using a single named parameter here
|
||||||
->where($query->expr()->iLike('uid', $query->createPositionalParameter('%' . $connection->escapeLikeParameter($search) . '%')))
|
->where($query->expr()->iLike('uid', $query->createPositionalParameter('%' . $this->dbConn->escapeLikeParameter($search) . '%')))
|
||||||
->orWhere($query->expr()->iLike('displayname', $query->createPositionalParameter('%' . $connection->escapeLikeParameter($search) . '%')))
|
->orWhere($query->expr()->iLike('displayname', $query->createPositionalParameter('%' . $this->dbConn->escapeLikeParameter($search) . '%')))
|
||||||
->orWhere($query->expr()->iLike('configvalue', $query->createPositionalParameter('%' . $connection->escapeLikeParameter($search) . '%')))
|
->orWhere($query->expr()->iLike('configvalue', $query->createPositionalParameter('%' . $this->dbConn->escapeLikeParameter($search) . '%')))
|
||||||
->orderBy($query->func()->lower('displayname'), 'ASC')
|
->orderBy($query->func()->lower('displayname'), 'ASC')
|
||||||
->orderBy($query->func()->lower('uid'), 'ASC')
|
->orderBy($query->func()->lower('uid'), 'ASC')
|
||||||
->setMaxResults($limit)
|
->setMaxResults($limit)
|
||||||
|
@ -250,10 +275,20 @@ class Database extends ABackend
|
||||||
* returns the user id or false
|
* returns the user id or false
|
||||||
*/
|
*/
|
||||||
public function checkPassword(string $uid, string $password) {
|
public function checkPassword(string $uid, string $password) {
|
||||||
$query = \OC_DB::prepare('SELECT `uid`, `password` FROM `*PREFIX*users` WHERE LOWER(`uid`) = LOWER(?)');
|
$this->fixDI();
|
||||||
$result = $query->execute([$uid]);
|
|
||||||
|
$qb = $this->dbConn->getQueryBuilder();
|
||||||
|
$qb->select('uid', 'password')
|
||||||
|
->from('users')
|
||||||
|
->where(
|
||||||
|
$qb->expr()->eq(
|
||||||
|
'uid_lower', $qb->createNamedParameter(mb_strtolower($uid))
|
||||||
|
)
|
||||||
|
);
|
||||||
|
$result = $qb->execute();
|
||||||
|
$row = $result->fetch();
|
||||||
|
$result->closeCursor();
|
||||||
|
|
||||||
$row = $result->fetchRow();
|
|
||||||
if ($row) {
|
if ($row) {
|
||||||
$storedHash = $row['password'];
|
$storedHash = $row['password'];
|
||||||
$newHash = '';
|
$newHash = '';
|
||||||
|
@ -276,6 +311,8 @@ class Database extends ABackend
|
||||||
* @return boolean true if user was found, false otherwise
|
* @return boolean true if user was found, false otherwise
|
||||||
*/
|
*/
|
||||||
private function loadUser($uid) {
|
private function loadUser($uid) {
|
||||||
|
$this->fixDI();
|
||||||
|
|
||||||
$uid = (string)$uid;
|
$uid = (string)$uid;
|
||||||
if (!isset($this->cache[$uid])) {
|
if (!isset($this->cache[$uid])) {
|
||||||
//guests $uid could be NULL or ''
|
//guests $uid could be NULL or ''
|
||||||
|
@ -284,23 +321,25 @@ class Database extends ABackend
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
$query = \OC_DB::prepare('SELECT `uid`, `displayname` FROM `*PREFIX*users` WHERE LOWER(`uid`) = LOWER(?)');
|
$qb = $this->dbConn->getQueryBuilder();
|
||||||
$result = $query->execute([$uid]);
|
$qb->select('uid', 'displayname')
|
||||||
|
->from('users')
|
||||||
if ($result === false) {
|
->where(
|
||||||
Util::writeLog('core', \OC_DB::getErrorMessage(), Util::ERROR);
|
$qb->expr()->eq(
|
||||||
return false;
|
'uid_lower', $qb->createNamedParameter(mb_strtolower($uid))
|
||||||
}
|
)
|
||||||
|
);
|
||||||
|
$result = $qb->execute();
|
||||||
|
$row = $result->fetch();
|
||||||
|
$result->closeCursor();
|
||||||
|
|
||||||
$this->cache[$uid] = false;
|
$this->cache[$uid] = false;
|
||||||
|
|
||||||
// "uid" is primary key, so there can only be a single result
|
// "uid" is primary key, so there can only be a single result
|
||||||
if ($row = $result->fetchRow()) {
|
if ($row !== false) {
|
||||||
$this->cache[$uid]['uid'] = $row['uid'];
|
$this->cache[$uid]['uid'] = $row['uid'];
|
||||||
$this->cache[$uid]['displayname'] = $row['displayname'];
|
$this->cache[$uid]['displayname'] = $row['displayname'];
|
||||||
$result->closeCursor();
|
|
||||||
} else {
|
} else {
|
||||||
$result->closeCursor();
|
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -361,6 +400,8 @@ class Database extends ABackend
|
||||||
* @return int|bool
|
* @return int|bool
|
||||||
*/
|
*/
|
||||||
public function countUsers() {
|
public function countUsers() {
|
||||||
|
$this->fixDI();
|
||||||
|
|
||||||
$query = \OC_DB::prepare('SELECT COUNT(*) FROM `*PREFIX*users`');
|
$query = \OC_DB::prepare('SELECT COUNT(*) FROM `*PREFIX*users`');
|
||||||
$result = $query->execute();
|
$result = $query->execute();
|
||||||
if ($result === false) {
|
if ($result === false) {
|
||||||
|
|
Loading…
Reference in New Issue