Adding multihost - locking

Adding Unit-Tests for it
This commit is contained in:
Philipp Holzer 2018-07-04 23:37:22 +02:00
parent b07dfbb03f
commit aac94d1d74
No known key found for this signature in database
GPG Key ID: 58160D7D6AF942B6
22 changed files with 741 additions and 154 deletions

View File

@ -41,7 +41,7 @@ define('FRIENDICA_PLATFORM', 'Friendica');
define('FRIENDICA_CODENAME', 'The Tazmans Flax-lily'); define('FRIENDICA_CODENAME', 'The Tazmans Flax-lily');
define('FRIENDICA_VERSION', '2018.08-dev'); define('FRIENDICA_VERSION', '2018.08-dev');
define('DFRN_PROTOCOL_VERSION', '2.23'); define('DFRN_PROTOCOL_VERSION', '2.23');
define('DB_UPDATE_VERSION', 1272); define('DB_UPDATE_VERSION', 1273);
define('NEW_UPDATE_ROUTINE_VERSION', 1170); define('NEW_UPDATE_ROUTINE_VERSION', 1170);
/** /**

View File

@ -0,0 +1,83 @@
<?php
namespace Friendica\Core\Cache;
use Friendica\Core\Cache;
/**
* @brief Implementation of the IMemoryCacheDriver mainly for testing purpose
*
* Class ArrayCache
*
* @package Friendica\Core\Cache
*/
class ArrayCache implements IMemoryCacheDriver
{
use TraitCompareDelete;
/** @var array Array with the cached data */
protected $cachedData = array();
/**
* (@inheritdoc)
*/
public function get($key)
{
if (isset($this->cachedData[$key])) {
return $this->cachedData[$key];
}
return null;
}
/**
* (@inheritdoc)
*/
public function set($key, $value, $ttl = Cache::FIVE_MINUTES)
{
$this->cachedData[$key] = $value;
return true;
}
/**
* (@inheritdoc)
*/
public function delete($key)
{
unset($this->cachedData[$key]);
return true;
}
/**
* (@inheritdoc)
*/
public function clear()
{
$this->cachedData = [];
return true;
}
/**
* (@inheritdoc)
*/
public function add($key, $value, $ttl = Cache::FIVE_MINUTES)
{
if (isset($this->cachedData[$key])) {
return false;
} else {
return $this->set($key, $value, $ttl);
}
}
/**
* (@inheritdoc)
*/
public function compareSet($key, $oldValue, $newValue, $ttl = Cache::FIVE_MINUTES)
{
if ($this->get($key) === $oldValue) {
return $this->set($key, $newValue);
} else {
return false;
}
}
}

View File

@ -33,11 +33,11 @@ class DatabaseCacheDriver implements ICacheDriver
return null; return null;
} }
public function set($key, $value, $duration = Cache::MONTH) public function set($key, $value, $ttl = Cache::FIVE_MINUTES)
{ {
$fields = [ $fields = [
'v' => serialize($value), 'v' => serialize($value),
'expires' => DateTimeFormat::utc('now + ' . $duration . ' seconds'), 'expires' => DateTimeFormat::utc('now + ' . $ttl . ' seconds'),
'updated' => DateTimeFormat::utcNow() 'updated' => DateTimeFormat::utcNow()
]; ];

View File

@ -12,7 +12,7 @@ use Friendica\Core\Cache;
interface ICacheDriver interface ICacheDriver
{ {
/** /**
* Fetches cached data according to the key * @brief Fetches cached data according to the key
* *
* @param string $key The key to the cached data * @param string $key The key to the cached data
* *
@ -21,28 +21,27 @@ interface ICacheDriver
public function get($key); public function get($key);
/** /**
* Stores data in the cache identified by the key. The input $value can have multiple formats. * @brief Stores data in the cache identified by the key. The input $value can have multiple formats.
* *
* @param string $key The cache key * @param string $key The cache key
* @param mixed $value The value to store * @param mixed $value The value to store
* @param integer $duration The cache lifespan, must be one of the Cache constants * @param integer $ttl The cache lifespan, must be one of the Cache constants
* *
* @return bool * @return bool
*/ */
public function set($key, $value, $duration = Cache::MONTH); public function set($key, $value, $ttl = Cache::FIVE_MINUTES);
/** /**
* Delete a key from the cache * @brief Delete a key from the cache
* *
* @param string $key * @param string $key The cache key
* *
* @return bool * @return bool
*/ */
public function delete($key); public function delete($key);
/** /**
* Remove outdated data from the cache * @brief Remove outdated data from the cache
* *
* @return bool * @return bool
*/ */

View File

@ -0,0 +1,45 @@
<?php
namespace Friendica\Core\Cache;
use Friendica\Core\Cache;
/**
* @brief This interface defines methods for Memory-Caches only
*
* Interface IMemoryCacheDriver
*
* @package Friendica\Core\Cache
*/
interface IMemoryCacheDriver extends ICacheDriver
{
/**
* @brief Sets a value if it's not already stored
*
* @param string $key The cache key
* @param mixed $value The old value we know from the cache
* @param int $ttl The cache lifespan, must be one of the Cache constants
* @return bool
*/
public function add($key, $value, $ttl = Cache::FIVE_MINUTES);
/**
* @brief Compares if the old value is set and sets the new value
*
* @param string $key The cache key
* @param mixed $oldValue The old value we know from the cache
* @param mixed $newValue The new value we want to set
* @param int $ttl The cache lifespan, must be one of the Cache constants
*
* @return bool
*/
public function compareSet($key, $oldValue, $newValue, $ttl = Cache::FIVE_MINUTES);
/**
* @brief Compares if the old value is set and removes it
*
* @param string $key The cache key
* @param mixed $value The old value we know and want to delete
* @return bool
*/
public function compareDelete($key, $value);
}

View File

@ -10,10 +10,13 @@ use Friendica\Core\Cache;
* *
* @author Hypolite Petovan <mrpetovan@gmail.com> * @author Hypolite Petovan <mrpetovan@gmail.com>
*/ */
class MemcacheCacheDriver extends BaseObject implements ICacheDriver class MemcacheCacheDriver extends BaseObject implements IMemoryCacheDriver
{ {
use TraitCompareSet;
use TraitCompareDelete;
/** /**
* @var Memcache * @var \Memcache
*/ */
private $memcache; private $memcache;
@ -30,6 +33,9 @@ class MemcacheCacheDriver extends BaseObject implements ICacheDriver
} }
} }
/**
* (@inheritdoc)
*/
public function get($key) public function get($key)
{ {
$return = null; $return = null;
@ -54,17 +60,31 @@ class MemcacheCacheDriver extends BaseObject implements ICacheDriver
return $return; return $return;
} }
public function set($key, $value, $duration = Cache::MONTH) /**
* (@inheritdoc)
*/
public function set($key, $value, $ttl = Cache::FIVE_MINUTES)
{ {
// We store with the hostname as key to avoid problems with other applications // We store with the hostname as key to avoid problems with other applications
return $this->memcache->set( if ($ttl > 0) {
self::getApp()->get_hostname() . ":" . $key, return $this->memcache->set(
serialize($value), self::getApp()->get_hostname() . ":" . $key,
MEMCACHE_COMPRESSED, serialize($value),
time() + $duration MEMCACHE_COMPRESSED,
); time() + $ttl
);
} else {
return $this->memcache->set(
self::getApp()->get_hostname() . ":" . $key,
serialize($value),
MEMCACHE_COMPRESSED
);
}
} }
/**
* (@inheritdoc)
*/
public function delete($key) public function delete($key)
{ {
return $this->memcache->delete($key); return $this->memcache->delete($key);
@ -72,6 +92,14 @@ class MemcacheCacheDriver extends BaseObject implements ICacheDriver
public function clear() public function clear()
{ {
return true; return $this->memcache->flush();
}
/**
* (@inheritdoc)
*/
public function add($key, $value, $ttl = Cache::FIVE_MINUTES)
{
return $this->memcache->add(self::getApp()->get_hostname() . ":" . $key, $value, $ttl);
} }
} }

View File

@ -10,8 +10,11 @@ use Friendica\Core\Cache;
* *
* @author Hypolite Petovan <mrpetovan@gmail.com> * @author Hypolite Petovan <mrpetovan@gmail.com>
*/ */
class MemcachedCacheDriver extends BaseObject implements ICacheDriver class MemcachedCacheDriver extends BaseObject implements IMemoryCacheDriver
{ {
use TraitCompareSet;
use TraitCompareDelete;
/** /**
* @var Memcached * @var Memcached
*/ */
@ -46,14 +49,22 @@ class MemcachedCacheDriver extends BaseObject implements ICacheDriver
return $return; return $return;
} }
public function set($key, $value, $duration = Cache::MONTH) public function set($key, $value, $ttl = Cache::FIVE_MINUTES)
{ {
// We store with the hostname as key to avoid problems with other applications // We store with the hostname as key to avoid problems with other applications
return $this->memcached->set( if ($ttl > 0) {
self::getApp()->get_hostname() . ':' . $key, return $this->memcached->set(
$value, self::getApp()->get_hostname() . ':' . $key,
time() + $duration $value,
); time() + $ttl
);
} else {
return $this->memcached->set(
self::getApp()->get_hostname() . ':' . $key,
$value
);
}
} }
public function delete($key) public function delete($key)
@ -67,4 +78,17 @@ class MemcachedCacheDriver extends BaseObject implements ICacheDriver
{ {
return true; return true;
} }
/**
* @brief Sets a value if it's not already stored
*
* @param string $key The cache key
* @param mixed $value The old value we know from the cache
* @param int $ttl The cache lifespan, must be one of the Cache constants
* @return bool
*/
public function add($key, $value, $ttl = Cache::FIVE_MINUTES)
{
return $this->memcached->add(self::getApp()->get_hostname() . ":" . $key, $value, $ttl);
}
} }

View File

@ -11,7 +11,7 @@ use Friendica\Core\Cache;
* @author Hypolite Petovan <mrpetovan@gmail.com> * @author Hypolite Petovan <mrpetovan@gmail.com>
* @author Roland Haeder <roland@mxchange.org> * @author Roland Haeder <roland@mxchange.org>
*/ */
class RedisCacheDriver extends BaseObject implements ICacheDriver class RedisCacheDriver extends BaseObject implements IMemoryCacheDriver
{ {
/** /**
* @var Redis * @var Redis
@ -55,14 +55,21 @@ class RedisCacheDriver extends BaseObject implements ICacheDriver
return $return; return $return;
} }
public function set($key, $value, $duration = Cache::MONTH) public function set($key, $value, $ttl = Cache::FIVE_MINUTES)
{ {
// We store with the hostname as key to avoid problems with other applications // We store with the hostname as key to avoid problems with other applications
return $this->redis->set( if ($ttl > 0) {
self::getApp()->get_hostname() . ":" . $key, return $this->redis->setex(
serialize($value), self::getApp()->get_hostname() . ":" . $key,
time() + $duration time() + $ttl,
); serialize($value)
);
} else {
return $this->redis->set(
self::getApp()->get_hostname() . ":" . $key,
serialize($value)
);
}
} }
public function delete($key) public function delete($key)
@ -74,4 +81,75 @@ class RedisCacheDriver extends BaseObject implements ICacheDriver
{ {
return true; return true;
} }
/**
* @brief Sets a value if it's not already stored
*
* @param string $key The cache key
* @param mixed $value The old value we know from the cache
* @param int $ttl The cache lifespan, must be one of the Cache constants
* @return bool
*/
public function add($key, $value, $ttl = Cache::FIVE_MINUTES)
{
if (!is_int($value)) {
$value = serialize($value);
}
return $this->redis->setnx(self::getApp()->get_hostname() . ":" . $key, $value);
}
/**
* @brief Compares if the old value is set and sets the new value
*
* @param string $key The cache key
* @param mixed $oldValue The old value we know
* @param mixed $newValue The new value we want to set
* @param int $ttl The cache lifespan, must be one of the Cache constants
* @return bool
*/
public function compareSet($key, $oldValue, $newValue, $ttl = Cache::FIVE_MINUTES)
{
if (!is_int($newValue)) {
$newValue = serialize($newValue);
}
$this->redis->watch(self::getApp()->get_hostname() . ":" . $key);
// If the old value isn't what we expected, somebody else changed the key meanwhile
if ($this->get($key) === $oldValue) {
if ($ttl > 0) {
$result = $this->redis->multi()
->setex(self::getApp()->get_hostname() . ":" . $ttl, $key, $newValue)
->exec();
} else {
$result = $this->redis->multi()
->set(self::getApp()->get_hostname() . ":" . $key, $newValue)
->exec();
}
return $result !== false;
}
$this->redis->unwatch();
return false;
}
/**
* @brief Compares if the old value is set and removes it
*
* @param string $key The cache key
* @param mixed $value The old value we know and want to delete
* @return bool
*/
public function compareDelete($key, $value)
{
$this->redis->watch(self::getApp()->get_hostname() . ":" . $key);
// If the old value isn't what we expected, somebody else changed the key meanwhile
if ($this->get($key) === $value) {
$result = $this->redis->multi()
->del(self::getApp()->get_hostname() . ":" . $key)
->exec();
return $result !== false;
}
$this->redis->unwatch();
return false;
}
} }

View File

@ -0,0 +1,45 @@
<?php
namespace Friendica\Core\Cache;
use Friendica\Core\Cache;
/**
* Trait TraitCompareSetDelete
*
* @brief This Trait is to compensate non native "exclusive" sets/deletes in caches
*
* @package Friendica\Core\Cache
*/
trait TraitCompareDelete
{
abstract public function get($key);
abstract public function set($key, $value, $ttl = Cache::FIVE_MINUTES);
abstract public function delete($key);
abstract public function add($key, $value, $ttl = Cache::FIVE_MINUTES);
/**
* @brief NonNative - Compares if the old value is set and removes it
*
* @param string $key The cache key
* @param mixed $value The old value we know and want to delete
* @return bool
*/
public function compareDelete($key, $value) {
if ($this->add($key . "_lock", true)) {
if ($this->get($key) === $value) {
$this->delete($key);
$this->delete($key . "_lock");
return true;
} else {
$this->delete($key . "_lock");
return false;
}
} else {
return false;
}
}
}

View File

@ -0,0 +1,48 @@
<?php
namespace Friendica\Core\Cache;
use Friendica\Core\Cache;
/**
* Trait TraitCompareSetDelete
*
* @brief This Trait is to compensate non native "exclusive" sets/deletes in caches
*
* @package Friendica\Core\Cache
*/
trait TraitCompareSet
{
abstract public function get($key);
abstract public function set($key, $value, $ttl = Cache::FIVE_MINUTES);
abstract public function delete($key);
abstract public function add($key, $value, $ttl = Cache::FIVE_MINUTES);
/**
* @brief NonNative - Compares if the old value is set and sets the new value
*
* @param string $key The cache key
* @param mixed $oldValue The old value we know from the cache
* @param mixed $newValue The new value we want to set
* @param int $ttl The cache lifespan, must be one of the Cache constants
*
* @return bool
*/
public function compareSet($key, $oldValue, $newValue, $ttl = Cache::FIVE_MINUTES) {
if ($this->add($key . "_lock", true)) {
if ($this->get($key) === $oldValue) {
$this->set($key, $newValue, $ttl);
$this->delete($key . "_lock");
return true;
} else {
$this->delete($key . "_lock");
return false;
}
} else {
return false;
}
}
}

View File

@ -10,6 +10,7 @@ namespace Friendica\Core;
*/ */
use Friendica\Core\Cache\CacheDriverFactory; use Friendica\Core\Cache\CacheDriverFactory;
use Friendica\Core\Cache\IMemoryCacheDriver;
/** /**
* @brief This class contain Functions for preventing parallel execution of functions * @brief This class contain Functions for preventing parallel execution of functions
@ -29,17 +30,23 @@ class Lock
switch ($lock_driver) { switch ($lock_driver) {
case 'memcache': case 'memcache':
$cache_driver = CacheDriverFactory::create('memcache'); $cache_driver = CacheDriverFactory::create('memcache');
self::$driver = new Lock\CacheLockDriver($cache_driver); if ($cache_driver instanceof IMemoryCacheDriver) {
self::$driver = new Lock\CacheLockDriver($cache_driver);
}
break; break;
case 'memcached': case 'memcached':
$cache_driver = CacheDriverFactory::create('memcached'); $cache_driver = CacheDriverFactory::create('memcached');
self::$driver = new Lock\CacheLockDriver($cache_driver); if ($cache_driver instanceof IMemoryCacheDriver) {
self::$driver = new Lock\CacheLockDriver($cache_driver);
}
break; break;
case 'redis': case 'redis':
$cache_driver = CacheDriverFactory::create('redis'); $cache_driver = CacheDriverFactory::create('redis');
self::$driver = new Lock\CacheLockDriver($cache_driver); if ($cache_driver instanceof IMemoryCacheDriver) {
self::$driver = new Lock\CacheLockDriver($cache_driver);
}
break; break;
case 'database': case 'database':
@ -85,7 +92,9 @@ class Lock
if ($cache_driver != 'database') { if ($cache_driver != 'database') {
try { try {
$lock_driver = CacheDriverFactory::create($cache_driver); $lock_driver = CacheDriverFactory::create($cache_driver);
self::$driver = new Lock\CacheLockDriver($lock_driver); if ($lock_driver instanceof IMemoryCacheDriver) {
self::$driver = new Lock\CacheLockDriver($lock_driver);
}
return; return;
} catch (\Exception $exception) { } catch (\Exception $exception) {
logger('Using Cache driver for locking failed: ' . $exception->getMessage()); logger('Using Cache driver for locking failed: ' . $exception->getMessage());

View File

@ -1,6 +1,7 @@
<?php <?php
namespace Friendica\Core\Lock; namespace Friendica\Core\Lock;
use Friendica\BaseObject;
/** /**
* Class AbstractLockDriver * Class AbstractLockDriver
@ -9,7 +10,7 @@ namespace Friendica\Core\Lock;
* *
* @brief Basic class for Locking with common functions (local acquired locks, releaseAll, ..) * @brief Basic class for Locking with common functions (local acquired locks, releaseAll, ..)
*/ */
abstract class AbstractLockDriver implements ILockDriver abstract class AbstractLockDriver extends BaseObject implements ILockDriver
{ {
/** /**
* @var array The local acquired locks * @var array The local acquired locks
@ -23,7 +24,7 @@ abstract class AbstractLockDriver implements ILockDriver
* @return bool Returns true if the lock is set * @return bool Returns true if the lock is set
*/ */
protected function hasAcquiredLock($key) { protected function hasAcquiredLock($key) {
return isset($this->acquireLock[$key]); return isset($this->acquireLock[$key]) && $this->acquiredLocks[$key] === true;
} }
/** /**
@ -50,7 +51,7 @@ abstract class AbstractLockDriver implements ILockDriver
* @return void * @return void
*/ */
public function releaseAll() { public function releaseAll() {
foreach ($this->acquiredLocks as $acquiredLock) { foreach ($this->acquiredLocks as $acquiredLock => $hasLock) {
$this->releaseLock($acquiredLock); $this->releaseLock($acquiredLock);
} }
} }

View File

@ -2,7 +2,7 @@
namespace Friendica\Core\Lock; namespace Friendica\Core\Lock;
use Friendica\Core\Cache\ICacheDriver; use Friendica\Core\Cache\IMemoryCacheDriver;
class CacheLockDriver extends AbstractLockDriver class CacheLockDriver extends AbstractLockDriver
{ {
@ -14,9 +14,9 @@ class CacheLockDriver extends AbstractLockDriver
/** /**
* CacheLockDriver constructor. * CacheLockDriver constructor.
* *
* @param ICacheDriver $cache The CacheDriver for this type of lock * @param IMemoryCacheDriver $cache The CacheDriver for this type of lock
*/ */
public function __construct(ICacheDriver $cache) public function __construct(IMemoryCacheDriver $cache)
{ {
$this->cache = $cache; $this->cache = $cache;
} }
@ -35,32 +35,31 @@ class CacheLockDriver extends AbstractLockDriver
$got_lock = false; $got_lock = false;
$start = time(); $start = time();
$cachekey = get_app()->get_hostname() . ";lock:" . $key; $cachekey = self::getCacheKey($key);
do { do {
$lock = $this->cache->get($cachekey); $lock = $this->cache->get($cachekey);
// When we do want to lock something that was already locked by us.
if (!is_bool($lock)) { if ((int)$lock == getmypid()) {
$pid = (int)$lock;
// When the process id isn't used anymore, we can safely claim the lock for us.
// Or we do want to lock something that was already locked by us.
if (!posix_kill($pid, 0) || ($pid == getmypid())) {
$lock = false;
}
}
if (is_bool($lock)) {
$this->cache->set($cachekey, getmypid(), 300);
$got_lock = true; $got_lock = true;
} }
// When we do want to lock something new
if (is_null($lock)) {
// At first initialize it with "0"
$this->cache->add($cachekey, 0);
// Now the value has to be "0" because otherwise the key was used by another process meanwhile
if ($this->cache->compareSet($cachekey, 0, getmypid(), 300)) {
$got_lock = true;
$this->markAcquire($key);
}
}
if (!$got_lock && ($timeout > 0)) { if (!$got_lock && ($timeout > 0)) {
usleep(rand(10000, 200000)); usleep(rand(10000, 200000));
} }
} while (!$got_lock && ((time() - $start) < $timeout)); } while (!$got_lock && ((time() - $start) < $timeout));
$this->markAcquire($key);
return $got_lock; return $got_lock;
} }
@ -68,22 +67,33 @@ class CacheLockDriver extends AbstractLockDriver
* @brief Removes a lock if it was set by us * @brief Removes a lock if it was set by us
* *
* @param string $key Name of the lock * @param string $key Name of the lock
*
* @return mixed
*/ */
public function releaseLock($key) public function releaseLock($key)
{ {
$cachekey = get_app()->get_hostname() . ";lock:" . $key; $cachekey = self::getCacheKey($key);
$lock = $this->cache->get($cachekey);
if (!is_bool($lock)) {
if ((int)$lock == getmypid()) {
$this->cache->delete($cachekey);
}
}
$this->cache->compareDelete($cachekey, getmypid());
$this->markRelease($key); $this->markRelease($key);
}
return; /**
* @brief Checks, if a key is currently locked to a process
*
* @param string $key The name of the lock
* @return bool
*/
public function isLocked($key)
{
$cachekey = self::getCacheKey($key);
$lock = $this->cache->get($cachekey);
return isset($lock) && ($lock !== false);
}
/**
* @param string $key The original key
* @return string The cache key used for the cache
*/
private static function getCacheKey($key) {
return self::getApp()->get_hostname() . ";lock:" . $key;
} }
} }

View File

@ -4,6 +4,7 @@ namespace Friendica\Core\Lock;
use dba; use dba;
use Friendica\Database\DBM; use Friendica\Database\DBM;
use Friendica\Util\DateTimeFormat;
/** /**
* Locking driver that stores the locks in the database * Locking driver that stores the locks in the database
@ -11,12 +12,7 @@ use Friendica\Database\DBM;
class DatabaseLockDriver extends AbstractLockDriver class DatabaseLockDriver extends AbstractLockDriver
{ {
/** /**
* @brief Sets a lock for a given name * (@inheritdoc)
*
* @param string $key The Name of the lock
* @param integer $timeout Seconds until we give up
*
* @return boolean Was the lock successful?
*/ */
public function acquireLock($key, $timeout = 120) public function acquireLock($key, $timeout = 120)
{ {
@ -25,26 +21,25 @@ class DatabaseLockDriver extends AbstractLockDriver
do { do {
dba::lock('locks'); dba::lock('locks');
$lock = dba::selectFirst('locks', ['locked', 'pid'], ['name' => $key]); $lock = dba::selectFirst('locks', ['locked', 'pid'], ['`name` = ? AND `expires` >= ?', $key, DateTimeFormat::utcNow()]);
if (DBM::is_result($lock)) { if (DBM::is_result($lock)) {
if ($lock['locked']) { if ($lock['locked']) {
// When the process id isn't used anymore, we can safely claim the lock for us.
if (!posix_kill($lock['pid'], 0)) {
$lock['locked'] = false;
}
// We want to lock something that was already locked by us? So we got the lock. // We want to lock something that was already locked by us? So we got the lock.
if ($lock['pid'] == getmypid()) { if ($lock['pid'] == getmypid()) {
$got_lock = true; $got_lock = true;
$this->markAcquire($key);
} }
} }
if (!$lock['locked']) { if (!$lock['locked']) {
dba::update('locks', ['locked' => true, 'pid' => getmypid()], ['name' => $key]); dba::update('locks', ['locked' => true, 'pid' => getmypid(), 'expires' => DateTimeFormat::utc('now + 300seconds')], ['name' => $key]);
$got_lock = true; $got_lock = true;
$this->markAcquire($key);
} }
} else { } else {
dba::insert('locks', ['name' => $key, 'locked' => true, 'pid' => getmypid()]); dba::insert('locks', ['name' => $key, 'locked' => true, 'pid' => getmypid(), 'expires' => DateTimeFormat::utc('now + 300seconds')]);
$got_lock = true; $got_lock = true;
$this->markAcquire($key);
} }
dba::unlock(); dba::unlock();
@ -54,36 +49,42 @@ class DatabaseLockDriver extends AbstractLockDriver
} }
} while (!$got_lock && ((time() - $start) < $timeout)); } while (!$got_lock && ((time() - $start) < $timeout));
$this->markAcquire($key);
return $got_lock; return $got_lock;
} }
/** /**
* @brief Removes a lock if it was set by us * (@inheritdoc)
*
* @param string $key Name of the lock
*
* @return mixed
*/ */
public function releaseLock($key) public function releaseLock($key)
{ {
dba::delete('locks', ['locked' => false, 'pid' => 0], ['name' => $key, 'pid' => getmypid()]); dba::delete('locks', ['name' => $key, 'pid' => getmypid()]);
$this->releaseLock($key); $this->markRelease($key);
return; return;
} }
/** /**
* @brief Removes all lock that were set by us * (@inheritdoc)
*
* @return void
*/ */
public function releaseAll() public function releaseAll()
{ {
dba::delete('locks', ['locked' => false, 'pid' => 0], ['pid' => getmypid()]); dba::delete('locks', ['pid' => getmypid()]);
$this->acquiredLocks = []; $this->acquiredLocks = [];
} }
/**
* (@inheritdoc)
*/
public function isLocked($key)
{
$lock = dba::selectFirst('locks', ['locked'], ['`name` = ? AND `expires` >= ?', $key, DateTimeFormat::utcNow()]);
if (DBM::is_result($lock)) {
return $lock['locked'] !== false;
} else {
return false;
}
}
} }

View File

@ -9,6 +9,14 @@ namespace Friendica\Core\Lock;
*/ */
interface ILockDriver interface ILockDriver
{ {
/**
* @brief Checks, if a key is currently locked to a or my process
*
* @param string $key The name of the lock
* @return bool
*/
public function isLocked($key);
/** /**
* *
* @brief Acquires a lock for a given name * @brief Acquires a lock for a given name

View File

@ -4,6 +4,8 @@ namespace Friendica\Core\Lock;
class SemaphoreLockDriver extends AbstractLockDriver class SemaphoreLockDriver extends AbstractLockDriver
{ {
private static $semaphore = [];
public function __construct() public function __construct()
{ {
if (!function_exists('sem_get')) { if (!function_exists('sem_get')) {
@ -42,10 +44,15 @@ class SemaphoreLockDriver extends AbstractLockDriver
*/ */
public function acquireLock($key, $timeout = 120) public function acquireLock($key, $timeout = 120)
{ {
$this->acquiredLocks[$key] = sem_get(self::semaphoreKey($key)); self::$semaphore[$key] = sem_get(self::semaphoreKey($key));
if ($this->acquiredLocks[$key]) { if (self::$semaphore[$key]) {
return sem_acquire($this->acquiredLocks[$key], ($timeout == 0)); if (sem_acquire(self::$semaphore[$key], ($timeout == 0))) {
$this->markAcquire($key);
return true;
}
} }
return false;
} }
/** /**
@ -57,12 +64,24 @@ class SemaphoreLockDriver extends AbstractLockDriver
*/ */
public function releaseLock($key) public function releaseLock($key)
{ {
if (empty($this->acquiredLocks[$key])) { if (empty(self::$semaphore[$key])) {
return false; return false;
} else { } else {
$success = @sem_release($this->acquiredLocks[$key]); $success = @sem_release(self::$semaphore[$key]);
unset($this->acquiredLocks[$key]); unset(self::$semaphore[$key]);
$this->markRelease($key);
return $success; return $success;
} }
} }
/**
* @brief Checks, if a key is currently locked to a process
*
* @param string $key The name of the lock
* @return bool
*/
public function isLocked($key)
{
return @sem_get(self::$semaphore[$key]) !== false;
}
} }

View File

@ -4,10 +4,9 @@
*/ */
namespace Friendica\Database; namespace Friendica\Database;
use dba;
use Friendica\Core\Config; use Friendica\Core\Config;
use Friendica\Core\L10n; use Friendica\Core\L10n;
use Friendica\Database\DBM;
use dba;
require_once 'boot.php'; require_once 'boot.php';
require_once 'include/dba.php'; require_once 'include/dba.php';
@ -1285,9 +1284,11 @@ class DBStructure
"name" => ["type" => "varchar(128)", "not null" => "1", "default" => "", "comment" => ""], "name" => ["type" => "varchar(128)", "not null" => "1", "default" => "", "comment" => ""],
"locked" => ["type" => "boolean", "not null" => "1", "default" => "0", "comment" => ""], "locked" => ["type" => "boolean", "not null" => "1", "default" => "0", "comment" => ""],
"pid" => ["type" => "int unsigned", "not null" => "1", "default" => "0", "comment" => "Process ID"], "pid" => ["type" => "int unsigned", "not null" => "1", "default" => "0", "comment" => "Process ID"],
], "expires" => ["type" => "datetime", "not null" => "1", "default" => NULL_DATE, "comment" => "datetime of cache expiration"],
],
"indexes" => [ "indexes" => [
"PRIMARY" => ["id"], "PRIMARY" => ["id"],
"name_expires" => ["name", "expires"]
] ]
]; ];
$database["mail"] = [ $database["mail"] = [

View File

@ -14,7 +14,7 @@ user:
uid: 42 uid: 42
username: Test user username: Test user
nickname: selfcontact nickname: selfcontact
verified: true verified: 1
password: $2y$10$DLRNTRmJgKe1cSrFJ5Jb0edCqvXlA9sh/RHdSnfxjbR.04yZRm4Qm password: $2y$10$DLRNTRmJgKe1cSrFJ5Jb0edCqvXlA9sh/RHdSnfxjbR.04yZRm4Qm
theme: frio theme: frio
@ -24,12 +24,12 @@ contact:
uid: 42 uid: 42
name: Self contact name: Self contact
nick: selfcontact nick: selfcontact
self: true self: 1
nurl: http://localhost/profile/selfcontact nurl: http://localhost/profile/selfcontact
url: http://localhost/profile/selfcontact url: http://localhost/profile/selfcontact
about: User used in tests about: User used in tests
pending: false pending: 0
blocked: false blocked: 0
rel: 1 rel: 1
network: dfrn network: dfrn
- -
@ -39,11 +39,11 @@ contact:
# the fallback to api_get_nick() in api_get_user() # the fallback to api_get_nick() in api_get_user()
name: othercontact name: othercontact
nick: othercontact nick: othercontact
self: false self: 0
nurl: http://localhost/profile/othercontact nurl: http://localhost/profile/othercontact
url: http://localhost/profile/othercontact url: http://localhost/profile/othercontact
pending: false pending: 0
blocked: false blocked: 0
rel: 0 rel: 0
network: dfrn network: dfrn
- -
@ -51,150 +51,150 @@ contact:
uid: 0 uid: 0
name: Friend contact name: Friend contact
nick: friendcontact nick: friendcontact
self: false self: 0
nurl: http://localhost/profile/friendcontact nurl: http://localhost/profile/friendcontact
url: http://localhost/profile/friendcontact url: http://localhost/profile/friendcontact
pending: false pending: 0
blocked: false blocked: 0
rel: 2 rel: 2
network: dfrn network: dfrn
item: item:
- -
id: 1 id: 1
visible: true visible: 1
contact-id: 42 contact-id: 42
author-id: 42 author-id: 42
owner-id: 45 owner-id: 45
uid: 42 uid: 42
verb: http://activitystrea.ms/schema/1.0/post verb: http://activitystrea.ms/schema/1.0/post
unseen: true unseen: 1
body: Parent status body: Parent status
parent: 1 parent: 1
author-link: http://localhost/profile/selfcontact author-link: http://localhost/profile/selfcontact
wall: true wall: 1
starred: true starred: 1
origin: true origin: 1
allow_cid: '' allow_cid: ''
allow_gid: '' allow_gid: ''
deny_cid: '' deny_cid: ''
deny_gid: '' deny_gid: ''
- -
id: 2 id: 2
visible: true visible: 1
contact-id: 42 contact-id: 42
author-id: 42 author-id: 42
owner-id: 45 owner-id: 45
uid: 42 uid: 42
verb: http://activitystrea.ms/schema/1.0/post verb: http://activitystrea.ms/schema/1.0/post
unseen: false unseen: 0
body: Reply body: Reply
parent: 1 parent: 1
author-link: http://localhost/profile/selfcontact author-link: http://localhost/profile/selfcontact
wall: true wall: 1
starred: false starred: 0
origin: true origin: 1
- -
id: 3 id: 3
visible: true visible: 1
contact-id: 43 contact-id: 43
author-id: 43 author-id: 43
owner-id: 42 owner-id: 42
uid: 42 uid: 42
verb: http://activitystrea.ms/schema/1.0/post verb: http://activitystrea.ms/schema/1.0/post
unseen: false unseen: 0
body: Other user status body: Other user status
parent: 3 parent: 3
author-link: http://localhost/profile/othercontact author-link: http://localhost/profile/othercontact
wall: true wall: 1
starred: false starred: 0
origin: true origin: 1
- -
id: 4 id: 4
visible: true visible: 1
contact-id: 44 contact-id: 44
author-id: 44 author-id: 44
owner-id: 42 owner-id: 42
uid: 42 uid: 42
verb: http://activitystrea.ms/schema/1.0/post verb: http://activitystrea.ms/schema/1.0/post
unseen: false unseen: 0
body: Friend user reply body: Friend user reply
parent: 1 parent: 1
author-link: http://localhost/profile/othercontact author-link: http://localhost/profile/othercontact
wall: true wall: 1
starred: false starred: 0
origin: true origin: 1
- -
id: 5 id: 5
visible: true visible: 1
contact-id: 42 contact-id: 42
author-id: 42 author-id: 42
owner-id: 42 owner-id: 42
uid: 42 uid: 42
verb: http://activitystrea.ms/schema/1.0/post verb: http://activitystrea.ms/schema/1.0/post
unseen: false unseen: 0
body: '[share]Shared status[/share]' body: '[share]Shared status[/share]'
parent: 1 parent: 1
author-link: http://localhost/profile/othercontact author-link: http://localhost/profile/othercontact
wall: true wall: 1
starred: false starred: 0
origin: true origin: 1
allow_cid: '' allow_cid: ''
allow_gid: '' allow_gid: ''
deny_cid: '' deny_cid: ''
deny_gid: '' deny_gid: ''
- -
id: 6 id: 6
visible: true visible: 1
contact-id: 44 contact-id: 44
author-id: 44 author-id: 44
owner-id: 42 owner-id: 42
uid: 42 uid: 42
verb: http://activitystrea.ms/schema/1.0/post verb: http://activitystrea.ms/schema/1.0/post
unseen: false unseen: 0
body: Friend user status body: Friend user status
parent: 6 parent: 6
author-link: http://localhost/profile/othercontact author-link: http://localhost/profile/othercontact
wall: true wall: 1
starred: false starred: 0
origin: true origin: 1
thread: thread:
- -
iid: 1 iid: 1
visible: true visible: 1
contact-id: 42 contact-id: 42
author-id: 42 author-id: 42
owner-id: 42 owner-id: 42
uid: 42 uid: 42
wall: true wall: 1
- -
iid: 3 iid: 3
visible: true visible: 1
contact-id: 43 contact-id: 43
author-id: 43 author-id: 43
owner-id: 43 owner-id: 43
uid: 0 uid: 0
wall: true wall: 1
- -
iid: 6 iid: 6
visible: true visible: 1
contact-id: 44 contact-id: 44
author-id: 44 author-id: 44
owner-id: 44 owner-id: 44
uid: 0 uid: 0
wall: true wall: 1
group: group:
- -
id: 1 id: 1
uid: 42 uid: 42
visible: true visible: 1
name: Visible list name: Visible list
- -
id: 2 id: 2
uid: 42 uid: 42
visible: false visible: 0
name: Private list name: Private list
search: search:

View File

@ -0,0 +1,27 @@
<?php
namespace Friendica\Test\src\Core\Lock;
use Friendica\Core\Cache\ArrayCache;
use Friendica\Core\Lock\CacheLockDriver;
class CacheLockDriverTest extends LockTest
{
/**
* @var \Friendica\Core\Cache\IMemoryCacheDriver
*/
private $cache;
protected function getInstance()
{
$this->cache = new ArrayCache();
return new CacheLockDriver($this->cache);
}
public function tearDown()
{
$this->cache->clear();
parent::tearDown();
}
}

View File

@ -0,0 +1,67 @@
<?php
namespace Friendica\Test\src\Core\Lock;
use dba;
use Friendica\Core\Lock\DatabaseLockDriver;
use Friendica\Database\DBStructure;
use PHPUnit\DbUnit\DataSet\YamlDataSet;
use PHPUnit\DbUnit\TestCaseTrait;
use PHPUnit_Extensions_Database_DB_IDatabaseConnection;
class DatabaseLockDriverTest extends LockTest
{
use TestCaseTrait;
/**
* Get database connection.
*
* This function is executed before each test in order to get a database connection that can be used by tests.
* If no prior connection is available, it tries to create one using the USER, PASS and DB environment variables.
*
* If it could not connect to the database, the test is skipped.
*
* @return PHPUnit_Extensions_Database_DB_IDatabaseConnection
* @see https://phpunit.de/manual/5.7/en/database.html
*/
protected function getConnection()
{
if (!dba::$connected) {
dba::connect('localhost', getenv('USER'), getenv('PASS'), getenv('DB'));
if (dba::$connected) {
$app = get_app();
// We need to do this in order to disable logging
$app->module = 'install';
// Create database structure
DBStructure::update(false, true, true);
} else {
$this->markTestSkipped('Could not connect to the database.');
}
}
return $this->createDefaultDBConnection(dba::get_db(), getenv('DB'));
}
/**
* Get dataset to populate the database with.
* @return YamlDataSet
* @see https://phpunit.de/manual/5.7/en/database.html
*/
protected function getDataSet()
{
return new YamlDataSet(__DIR__ . '/../../../datasets/api.yml');
}
protected function getInstance()
{
return new DatabaseLockDriver();
}
public function tearDown()
{
dba::delete('locks', [ 'id > 0']);
parent::tearDown();
}
}

View File

@ -0,0 +1,80 @@
<?php
namespace Friendica\Test\src\Core\Lock;
use Friendica\App;
use Friendica\Core\Config;
use PHPUnit\Framework\TestCase;
abstract class LockTest extends TestCase
{
/**
* @var \Friendica\Core\Lock\ILockDriver
*/
protected $instance;
abstract protected function getInstance();
protected function setUp()
{
global $a;
parent::setUp();
$this->instance = $this->getInstance();
// Reusable App object
$this->app = new App(__DIR__.'/../');
$a = $this->app;
// Default config
Config::set('config', 'hostname', 'localhost');
Config::set('system', 'throttle_limit_day', 100);
Config::set('system', 'throttle_limit_week', 100);
Config::set('system', 'throttle_limit_month', 100);
Config::set('system', 'theme', 'system_theme');
}
public function testLock() {
$this->instance->acquireLock('foo', 1);
$this->assertTrue($this->instance->isLocked('foo'));
$this->assertFalse($this->instance->isLocked('bar'));
}
public function testDoubleLock() {
$this->instance->acquireLock('foo', 1);
$this->assertTrue($this->instance->isLocked('foo'));
// We already locked it
$this->assertTrue($this->instance->acquireLock('foo', 1));
}
public function testReleaseLock() {
$this->instance->acquireLock('foo', 1);
$this->assertTrue($this->instance->isLocked('foo'));
$this->instance->releaseLock('foo');
$this->assertFalse($this->instance->isLocked('foo'));
}
public function testReleaseAll() {
$this->instance->acquireLock('foo', 1);
$this->instance->acquireLock('bar', 1);
$this->instance->acquireLock('#/$%§', 1);
$this->instance->releaseAll();
$this->assertFalse($this->instance->isLocked('foo'));
$this->assertFalse($this->instance->isLocked('bar'));
$this->assertFalse($this->instance->isLocked('#/$%§'));
}
public function testReleaseAfterUnlock() {
$this->instance->acquireLock('foo', 1);
$this->instance->acquireLock('bar', 1);
$this->instance->acquireLock('#/$%§', 1);
$this->instance->releaseLock('foo');
$this->instance->releaseAll();
$this->assertFalse($this->instance->isLocked('bar'));
$this->assertFalse($this->instance->isLocked('#/$%§'));
}
}

View File

@ -0,0 +1,14 @@
<?php
namespace Friendica\Test\src\Core\Lock;
use Friendica\Core\Lock\SemaphoreLockDriver;
class SemaphoreLockDriverTest extends LockTest
{
protected function getInstance()
{
return new SemaphoreLockDriver();
}
}