LockRegistry.php 6.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164
  1. <?php
  2. /*
  3. * This file is part of the Symfony package.
  4. *
  5. * (c) Fabien Potencier <fabien@symfony.com>
  6. *
  7. * For the full copyright and license information, please view the LICENSE
  8. * file that was distributed with this source code.
  9. */
  10. namespace Symfony\Component\Cache;
  11. use Psr\Log\LoggerInterface;
  12. use Symfony\Contracts\Cache\CacheInterface;
  13. use Symfony\Contracts\Cache\ItemInterface;
  14. /**
  15. * LockRegistry is used internally by existing adapters to protect against cache stampede.
  16. *
  17. * It does so by wrapping the computation of items in a pool of locks.
  18. * Foreach each apps, there can be at most 20 concurrent processes that
  19. * compute items at the same time and only one per cache-key.
  20. *
  21. * @author Nicolas Grekas <p@tchwork.com>
  22. */
  23. final class LockRegistry
  24. {
  25. private static $openedFiles = [];
  26. private static $lockedFiles;
  27. private static $signalingException;
  28. private static $signalingCallback;
  29. /**
  30. * The number of items in this list controls the max number of concurrent processes.
  31. */
  32. private static $files = [
  33. __DIR__.\DIRECTORY_SEPARATOR.'Adapter'.\DIRECTORY_SEPARATOR.'AbstractAdapter.php',
  34. __DIR__.\DIRECTORY_SEPARATOR.'Adapter'.\DIRECTORY_SEPARATOR.'AbstractTagAwareAdapter.php',
  35. __DIR__.\DIRECTORY_SEPARATOR.'Adapter'.\DIRECTORY_SEPARATOR.'AdapterInterface.php',
  36. __DIR__.\DIRECTORY_SEPARATOR.'Adapter'.\DIRECTORY_SEPARATOR.'ApcuAdapter.php',
  37. __DIR__.\DIRECTORY_SEPARATOR.'Adapter'.\DIRECTORY_SEPARATOR.'ArrayAdapter.php',
  38. __DIR__.\DIRECTORY_SEPARATOR.'Adapter'.\DIRECTORY_SEPARATOR.'ChainAdapter.php',
  39. __DIR__.\DIRECTORY_SEPARATOR.'Adapter'.\DIRECTORY_SEPARATOR.'CouchbaseBucketAdapter.php',
  40. __DIR__.\DIRECTORY_SEPARATOR.'Adapter'.\DIRECTORY_SEPARATOR.'CouchbaseCollectionAdapter.php',
  41. __DIR__.\DIRECTORY_SEPARATOR.'Adapter'.\DIRECTORY_SEPARATOR.'DoctrineDbalAdapter.php',
  42. __DIR__.\DIRECTORY_SEPARATOR.'Adapter'.\DIRECTORY_SEPARATOR.'FilesystemAdapter.php',
  43. __DIR__.\DIRECTORY_SEPARATOR.'Adapter'.\DIRECTORY_SEPARATOR.'FilesystemTagAwareAdapter.php',
  44. __DIR__.\DIRECTORY_SEPARATOR.'Adapter'.\DIRECTORY_SEPARATOR.'MemcachedAdapter.php',
  45. __DIR__.\DIRECTORY_SEPARATOR.'Adapter'.\DIRECTORY_SEPARATOR.'NullAdapter.php',
  46. __DIR__.\DIRECTORY_SEPARATOR.'Adapter'.\DIRECTORY_SEPARATOR.'ParameterNormalizer.php',
  47. __DIR__.\DIRECTORY_SEPARATOR.'Adapter'.\DIRECTORY_SEPARATOR.'PdoAdapter.php',
  48. __DIR__.\DIRECTORY_SEPARATOR.'Adapter'.\DIRECTORY_SEPARATOR.'PhpArrayAdapter.php',
  49. __DIR__.\DIRECTORY_SEPARATOR.'Adapter'.\DIRECTORY_SEPARATOR.'PhpFilesAdapter.php',
  50. __DIR__.\DIRECTORY_SEPARATOR.'Adapter'.\DIRECTORY_SEPARATOR.'ProxyAdapter.php',
  51. __DIR__.\DIRECTORY_SEPARATOR.'Adapter'.\DIRECTORY_SEPARATOR.'Psr16Adapter.php',
  52. __DIR__.\DIRECTORY_SEPARATOR.'Adapter'.\DIRECTORY_SEPARATOR.'RedisAdapter.php',
  53. __DIR__.\DIRECTORY_SEPARATOR.'Adapter'.\DIRECTORY_SEPARATOR.'RedisTagAwareAdapter.php',
  54. __DIR__.\DIRECTORY_SEPARATOR.'Adapter'.\DIRECTORY_SEPARATOR.'TagAwareAdapter.php',
  55. __DIR__.\DIRECTORY_SEPARATOR.'Adapter'.\DIRECTORY_SEPARATOR.'TagAwareAdapterInterface.php',
  56. __DIR__.\DIRECTORY_SEPARATOR.'Adapter'.\DIRECTORY_SEPARATOR.'TraceableAdapter.php',
  57. __DIR__.\DIRECTORY_SEPARATOR.'Adapter'.\DIRECTORY_SEPARATOR.'TraceableTagAwareAdapter.php',
  58. ];
  59. /**
  60. * Defines a set of existing files that will be used as keys to acquire locks.
  61. *
  62. * @return array The previously defined set of files
  63. */
  64. public static function setFiles(array $files): array
  65. {
  66. $previousFiles = self::$files;
  67. self::$files = $files;
  68. foreach (self::$openedFiles as $file) {
  69. if ($file) {
  70. flock($file, \LOCK_UN);
  71. fclose($file);
  72. }
  73. }
  74. self::$openedFiles = self::$lockedFiles = [];
  75. return $previousFiles;
  76. }
  77. public static function compute(callable $callback, ItemInterface $item, bool &$save, CacheInterface $pool, \Closure $setMetadata = null, LoggerInterface $logger = null)
  78. {
  79. if ('\\' === \DIRECTORY_SEPARATOR && null === self::$lockedFiles) {
  80. // disable locking on Windows by default
  81. self::$files = self::$lockedFiles = [];
  82. }
  83. $key = self::$files ? abs(crc32($item->getKey())) % \count(self::$files) : -1;
  84. if ($key < 0 || self::$lockedFiles || !$lock = self::open($key)) {
  85. return $callback($item, $save);
  86. }
  87. self::$signalingException ??= unserialize("O:9:\"Exception\":1:{s:16:\"\0Exception\0trace\";a:0:{}}");
  88. self::$signalingCallback ??= function () { throw self::$signalingException; };
  89. while (true) {
  90. try {
  91. $locked = false;
  92. // race to get the lock in non-blocking mode
  93. $locked = flock($lock, \LOCK_EX | \LOCK_NB, $wouldBlock);
  94. if ($locked || !$wouldBlock) {
  95. $logger && $logger->info(sprintf('Lock %s, now computing item "{key}"', $locked ? 'acquired' : 'not supported'), ['key' => $item->getKey()]);
  96. self::$lockedFiles[$key] = true;
  97. $value = $callback($item, $save);
  98. if ($save) {
  99. if ($setMetadata) {
  100. $setMetadata($item);
  101. }
  102. $pool->save($item->set($value));
  103. $save = false;
  104. }
  105. return $value;
  106. }
  107. // if we failed the race, retry locking in blocking mode to wait for the winner
  108. $logger && $logger->info('Item "{key}" is locked, waiting for it to be released', ['key' => $item->getKey()]);
  109. flock($lock, \LOCK_SH);
  110. } finally {
  111. flock($lock, \LOCK_UN);
  112. unset(self::$lockedFiles[$key]);
  113. }
  114. try {
  115. $value = $pool->get($item->getKey(), self::$signalingCallback, 0);
  116. $logger && $logger->info('Item "{key}" retrieved after lock was released', ['key' => $item->getKey()]);
  117. $save = false;
  118. return $value;
  119. } catch (\Exception $e) {
  120. if (self::$signalingException !== $e) {
  121. throw $e;
  122. }
  123. $logger && $logger->info('Item "{key}" not found while lock was released, now retrying', ['key' => $item->getKey()]);
  124. }
  125. }
  126. return null;
  127. }
  128. private static function open(int $key)
  129. {
  130. if (null !== $h = self::$openedFiles[$key] ?? null) {
  131. return $h;
  132. }
  133. set_error_handler(function () {});
  134. try {
  135. $h = fopen(self::$files[$key], 'r+');
  136. } finally {
  137. restore_error_handler();
  138. }
  139. return self::$openedFiles[$key] = $h ?: @fopen(self::$files[$key], 'r');
  140. }
  141. }