110 lines
3.2 KiB
PHP
110 lines
3.2 KiB
PHP
|
<?php
|
||
|
|
|
||
|
|
namespace Friendica\Test\Util;
|
||
|
|
|
||
|
|
use Friendica\Core\Cache;
|
||
|
|
use Friendica\Core\Lock\DatabaseLockDriver;
|
||
|
|
|
||
|
|
trait DbaLockMockTrait
|
||
|
|
{
|
||
|
|
use DBAMockTrait;
|
||
|
|
use DateTimeFormatMockTrait;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Mocking acquireLock with DBA-backend
|
||
|
|
* @see DatabaseLockDriver::acquireLock()
|
||
|
|
*
|
||
|
|
* @param mixed $key The key to lock
|
||
|
|
* @param int $ttl The TimeToLive
|
||
|
|
*
|
||
|
|
* @param bool $locked Was the lock already set?
|
||
|
|
* @param null $pid The PID which was set
|
||
|
|
* @param bool $rowExists True, if a row already exists in the lock table
|
||
|
|
* @param null $time The current timestamp
|
||
|
|
* @param null|int $times How often the method will get used
|
||
|
|
*/
|
||
|
|
public function mockAcquireLock($key, $ttl = Cache::FIVE_MINUTES, $locked = false, $pid = null, $rowExists = true, $time = null, $times = null)
|
||
|
|
{
|
||
|
|
if ($time === null) {
|
||
|
|
$time = time();
|
||
|
|
}
|
||
|
|
|
||
|
|
if ($pid === null) {
|
||
|
|
$pid = getmypid();
|
||
|
|
}
|
||
|
|
|
||
|
|
$this->mockDbaLock('locks', true, $times);
|
||
|
|
|
||
|
|
$this->mockUtcNow($time, $times);
|
||
|
|
$result = ['locked' => $locked, 'pid' => $pid];
|
||
|
|
$this->mockSelectFirst('locks', ['locked', 'pid'], ['`name` = ? AND `expires` >= ?', $key, $time], $result, $times);
|
||
|
|
$this->mockIsResult($result, $rowExists, $times);
|
||
|
|
|
||
|
|
if ($rowExists) {
|
||
|
|
if (!$locked ) {
|
||
|
|
$this->mockUtc('now + ' . $ttl . 'seconds', $time, $times);
|
||
|
|
$this->mockDBAUpdate('locks', ['locked' => true, 'pid' => $pid, 'expires' => $time], ['name' => $key], [], true, $times);
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
$this->mockUtc('now + ' . $ttl . 'seconds', $time, $times);
|
||
|
|
$this->mockDBAInsert('locks', ['name' => $key, 'locked' => true, 'pid' => $pid, 'expires' => $time], false, true, $times);
|
||
|
|
}
|
||
|
|
|
||
|
|
$this->mockDbaUnlock($times);
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Mocking isLocked with DBA-backend
|
||
|
|
* @see DatabaseLockDriver::isLocked()
|
||
|
|
*
|
||
|
|
* @param mixed $key The key of the lock
|
||
|
|
* @param null|bool $return True, if the key is already locked
|
||
|
|
* @param null $time The current timestamp
|
||
|
|
* @param null|int $times How often the method will get used
|
||
|
|
*/
|
||
|
|
public function mockIsLocked($key, $return = true, $time = null, $times = null)
|
||
|
|
{
|
||
|
|
if ($time === null) {
|
||
|
|
$time = time();
|
||
|
|
}
|
||
|
|
|
||
|
|
$this->mockUtcNow($time, $times);
|
||
|
|
$return = ((isset($return)) ? ['locked' => $return] : null);
|
||
|
|
$this->mockSelectFirst('locks', ['locked'], ['`name` = ? AND `expires` >= ?', $key, $time], $return, $times);
|
||
|
|
$this->mockIsResult($return, (isset($return) && $return), $times);
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Mocking releaseAll with DBA-backend
|
||
|
|
* @see DatabaseLockDriver::releaseAll()
|
||
|
|
*
|
||
|
|
* @param null $pid The PID which was set
|
||
|
|
* @param null|int $times How often the method will get used
|
||
|
|
*/
|
||
|
|
public function mockReleaseAll($pid = null, $times = null)
|
||
|
|
{
|
||
|
|
if ($pid === null) {
|
||
|
|
$pid = getmypid();
|
||
|
|
}
|
||
|
|
|
||
|
|
$this->mockDBADelete('locks', ['pid' => $pid], true, $times);
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Mocking ReleaseLock with DBA-backend
|
||
|
|
* @see DatabaseLockDriver::releaseLock()
|
||
|
|
*
|
||
|
|
* @param mixed $key The key to release
|
||
|
|
* @param null|int $pid The PID which was set
|
||
|
|
* @param null|int $times How often the method will get used
|
||
|
|
*/
|
||
|
|
public function mockReleaseLock($key, $pid = null, $times = null)
|
||
|
|
{
|
||
|
|
if ($pid === null) {
|
||
|
|
$pid = getmypid();
|
||
|
|
}
|
||
|
|
|
||
|
|
$this->mockDBADelete('locks', ['name' => $key, 'pid' => $pid], true, $times);
|
||
|
|
}
|
||
|
|
}
|