5 changed files with 214 additions and 2 deletions
-
39db_structure.xml
-
131lib/private/lock/dblockingprovider.php
-
1lib/repair/dropoldtables.php
-
43tests/lib/lock/dblockingprovider.php
-
2version.php
@ -0,0 +1,131 @@ |
|||
<?php |
|||
/** |
|||
* @author Robin Appelman <icewind@owncloud.com> |
|||
* |
|||
* @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 OC\Lock; |
|||
|
|||
use OCP\IDBConnection; |
|||
use OCP\Lock\LockedException; |
|||
|
|||
class DBLockingProvider extends AbstractLockingProvider { |
|||
/** |
|||
* @var \OCP\IDBConnection |
|||
*/ |
|||
private $connection; |
|||
|
|||
/** |
|||
* @param \OCP\IDBConnection $connection |
|||
*/ |
|||
public function __construct(IDBConnection $connection) { |
|||
$this->connection = $connection; |
|||
} |
|||
|
|||
protected function initLockField($path) { |
|||
$this->connection->insertIfNotExist('*PREFIX*locks', ['path' => $path, 'lock' => 0], ['path']); |
|||
} |
|||
|
|||
/** |
|||
* @param string $path |
|||
* @param int $type self::LOCK_SHARED or self::LOCK_EXCLUSIVE |
|||
* @return bool |
|||
*/ |
|||
public function isLocked($path, $type) { |
|||
$query = $this->connection->prepare('SELECT `lock` from `*PREFIX*locks` WHERE `path` = ?'); |
|||
$query->execute([$path]); |
|||
$lockValue = (int)$query->fetchColumn(); |
|||
if ($type === self::LOCK_SHARED) { |
|||
return $lockValue > 0; |
|||
} else if ($type === self::LOCK_EXCLUSIVE) { |
|||
return $lockValue === -1; |
|||
} else { |
|||
return false; |
|||
} |
|||
} |
|||
|
|||
/** |
|||
* @param string $path |
|||
* @param int $type self::LOCK_SHARED or self::LOCK_EXCLUSIVE |
|||
* @throws \OCP\Lock\LockedException |
|||
*/ |
|||
public function acquireLock($path, $type) { |
|||
$this->initLockField($path); |
|||
if ($type === self::LOCK_SHARED) { |
|||
$result = $this->connection->executeUpdate( |
|||
'UPDATE `*PREFIX*locks` SET `lock` = `lock` + 1 WHERE `path` = ? AND `lock` >= 0', |
|||
[$path] |
|||
); |
|||
} else { |
|||
$result = $this->connection->executeUpdate( |
|||
'UPDATE `*PREFIX*locks` SET `lock` = -1 WHERE `path` = ? AND `lock` = 0', |
|||
[$path] |
|||
); |
|||
} |
|||
if ($result !== 1) { |
|||
throw new LockedException($path); |
|||
} |
|||
$this->markAcquire($path, $type); |
|||
} |
|||
|
|||
/** |
|||
* @param string $path |
|||
* @param int $type self::LOCK_SHARED or self::LOCK_EXCLUSIVE |
|||
*/ |
|||
public function releaseLock($path, $type) { |
|||
$this->initLockField($path); |
|||
if ($type === self::LOCK_SHARED) { |
|||
$this->connection->executeUpdate( |
|||
'UPDATE `*PREFIX*locks` SET `lock` = `lock` - 1 WHERE `path` = ? AND `lock` > 0', |
|||
[$path] |
|||
); |
|||
} else { |
|||
$this->connection->executeUpdate( |
|||
'UPDATE `*PREFIX*locks` SET `lock` = 0 WHERE `path` = ? AND `lock` = -1', |
|||
[$path] |
|||
); |
|||
} |
|||
$this->markRelease($path, $type); |
|||
} |
|||
|
|||
/** |
|||
* Change the type of an existing lock |
|||
* |
|||
* @param string $path |
|||
* @param int $targetType self::LOCK_SHARED or self::LOCK_EXCLUSIVE |
|||
* @throws \OCP\Lock\LockedException |
|||
*/ |
|||
public function changeLock($path, $targetType) { |
|||
$this->initLockField($path); |
|||
if ($targetType === self::LOCK_SHARED) { |
|||
$result = $this->connection->executeUpdate( |
|||
'UPDATE `*PREFIX*locks` SET `lock` = 1 WHERE `path` = ? AND `lock` = -1', |
|||
[$path] |
|||
); |
|||
} else { |
|||
$result = $this->connection->executeUpdate( |
|||
'UPDATE `*PREFIX*locks` SET `lock` = -1 WHERE `path` = ? AND `lock` = 1', |
|||
[$path] |
|||
); |
|||
} |
|||
if ($result !== 1) { |
|||
throw new LockedException($path); |
|||
} |
|||
$this->markChange($path, $targetType); |
|||
} |
|||
} |
|||
@ -0,0 +1,43 @@ |
|||
<?php |
|||
/** |
|||
* @author Robin Appelman <icewind@owncloud.com> |
|||
* |
|||
* @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 Test\Lock; |
|||
|
|||
class DBLockingProvider extends LockingProvider { |
|||
|
|||
/** |
|||
* @var \OCP\IDBConnection |
|||
*/ |
|||
private $connection; |
|||
|
|||
/** |
|||
* @return \OCP\Lock\ILockingProvider |
|||
*/ |
|||
protected function getInstance() { |
|||
$this->connection = \OC::$server->getDatabaseConnection(); |
|||
return new \OC\Lock\DBLockingProvider($this->connection); |
|||
} |
|||
|
|||
public function tearDown() { |
|||
$this->connection->executeQuery('DELETE FROM `*PREFIX*locks`'); |
|||
parent::tearDown(); |
|||
} |
|||
} |
|||
Write
Preview
Loading…
Cancel
Save
Reference in new issue