parent
a8552a1b24
commit
4ea7cbb0f5
@ -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(); |
||||
} |
||||
} |
||||
Loading…
Reference in new issue