extract and fix lock methods

master
Jean-Christian Paul Denis 2023-05-13 00:57:32 +02:00
parent 647f3da685
commit fed9454327
Signed by: JcDenis
GPG Key ID: 1B5B8C5B90B6C951
2 changed files with 165 additions and 39 deletions

148
src/Lock.php 100644
View File

@ -0,0 +1,148 @@
<?php
/**
* @brief periodical, a plugin for Dotclear 2
*
* @package Dotclear
* @subpackage Plugin
*
* @author Jean-Christian Denis and contributors
*
* @copyright Jean-Christian Denis
* @copyright GPL-2.0 https://www.gnu.org/licenses/gpl-2.0.html
*/
declare(strict_types=1);
namespace Dotclear\Plugin\periodical;
use Dotclear\Helper\File\{
Files,
Path
};
class Lock
{
/**
* Locked files resource stack.
*
* @var array<string,resource>
*/
protected static $lock_stack = [];
/**
* Locked files status stack.
*
* @var array<string,bool>
*/
protected static $lock_disposable = [];
/**
* Last lock attempt error
*
* @var string
*/
protected static $lock_error = '';
/**
* Lock file.
*
* @param string $file The file path
* @param bool $disposable File only use to lock
*
* @return null|string Clean file path on success, empty string on error, null if already locked
*/
public static function lock(string $file, bool $disposable = false): ?string
{
# Real path
$file = Path::real($file, false);
if (false === $file) {
self::$lock_error = __("Can't get file path");
return '';
}
# not a dir
if (is_dir($file)) {
self::$lock_error = __("Can't lock a directory");
return '';
}
# already marked as locked
if (isset(self::$lock_stack[$file]) || $disposable && file_exists($file)) {
return null;
}
# Need flock function
if (!function_exists('flock')) {
self::$lock_error = __("Can't call php function named flock");
return '';
}
# Make dir
if (!is_dir(dirname($file))) {
Files::makeDir(dirname($file), true);
}
# Open new file
if (!file_exists($file)) {
$resource = @fopen($file, 'w');
if ($resource === false) {
self::$lock_error = __("Can't create file");
return '';
}
fwrite($resource, '1', strlen('1'));
//fclose($resource);
} else {
# Open existsing file
$resource = @fopen($file, 'r+');
if ($resource === false) {
self::$lock_error = __("Can't open file");
return '';
}
}
# Lock file
if (!flock($resource, LOCK_EX | LOCK_NB)) {
self::$lock_error = __("Can't lock file");
return '';
}
self::$lock_stack[$file] = $resource;
self::$lock_disposable[$file] = $disposable;
return $file;
}
/**
* Unlock file.
*
* @param string $file The file to unlock
*/
public static function unlock(string $file): void
{
if (isset(self::$lock_stack[$file])) {
fclose(self::$lock_stack[$file]);
if (!empty(self::$lock_disposable[$file]) && file_exists($file)) {
@unlink($file);
}
unset(
self::$lock_stack[$file],
self::$lock_disposable[$file]
);
}
}
/**
* Get last error from lock method.
*
* @return string The last lock error
*/
public static function getlastLockError(): string
{
return self::$lock_error;
}
}

View File

@ -38,7 +38,7 @@ use Exception;
*/ */
class Utils class Utils
{ {
/** @var null|resource Lock update process */ /** @var null|string $lock File lock for update */
private static $lock = null; private static $lock = null;
/** /**
@ -401,56 +401,34 @@ class Utils
} }
/** /**
* Lock a file to see if an update is ongoing * Lock a file to see if an update is ongoing.
*
* @return bool True if file is locked
*/ */
public static function lockUpdate(): bool public static function lockUpdate(): bool
{ {
try { try {
// Need flock function # Cache writable ?
if (!function_exists('flock')) {
throw new Exception("Can't call php function named flock");
}
// Cache writable ?
if (!is_writable(DC_TPL_CACHE)) { if (!is_writable(DC_TPL_CACHE)) {
throw new Exception("Can't write in cache fodler"); throw new Exception("Can't write in cache fodler");
} }
// Set file path # Set file path
$f_md5 = md5((string) dcCore::app()->blog?->id); $f_md5 = md5((string) dcCore::app()->blog?->id);
$cached_file = sprintf( $file = sprintf(
'%s/%s/%s/%s/%s.txt', '%s/%s/%s/%s/%s.txt',
DC_TPL_CACHE, DC_TPL_CACHE,
'periodical', My::id(),
substr($f_md5, 0, 2), substr($f_md5, 0, 2),
substr($f_md5, 2, 2), substr($f_md5, 2, 2),
$f_md5 $f_md5
); );
// Real path
$cached_file = Path::real($cached_file, false); $file = Lock::lock($file);
if (is_bool($cached_file)) { if (is_null($file) || empty($file)) {
throw new Exception("Can't write in cache fodler"); return false;
} }
// Make dir
if (!is_dir(dirname($cached_file))) { self::$lock = $file;
Files::makeDir(dirname($cached_file), true);
}
// Make file
if (!file_exists($cached_file)) {
!$fp = @fopen($cached_file, 'w');
if ($fp === false) {
throw new Exception("Can't create file");
}
fwrite($fp, '1', strlen('1'));
fclose($fp);
}
// Open file
if (!($fp = @fopen($cached_file, 'r+'))) {
throw new Exception("Can't open file");
}
// Lock file
if (!flock($fp, LOCK_EX)) {
throw new Exception("Can't lock file");
}
self::$lock = $fp;
return true; return true;
} catch (Exception $e) { } catch (Exception $e) {
@ -459,12 +437,12 @@ class Utils
} }
/** /**
* Unlock update process * Unlock file of update process.
*/ */
public static function unlockUpdate(): void public static function unlockUpdate(): void
{ {
if (!is_null(self::$lock)) { if (!is_null(self::$lock)) {
@fclose(self::$lock); Lock::unlock(self::$lock);
self::$lock = null; self::$lock = null;
} }
} }