PhpRedisConnector.php 9.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275
  1. <?php
  2. namespace Illuminate\Redis\Connectors;
  3. use Illuminate\Contracts\Redis\Connector;
  4. use Illuminate\Redis\Connections\PhpRedisClusterConnection;
  5. use Illuminate\Redis\Connections\PhpRedisConnection;
  6. use Illuminate\Support\Arr;
  7. use Illuminate\Support\Facades\Redis as RedisFacade;
  8. use Illuminate\Support\Str;
  9. use InvalidArgumentException;
  10. use LogicException;
  11. use Redis;
  12. use RedisCluster;
  13. class PhpRedisConnector implements Connector
  14. {
  15. /**
  16. * Create a new connection.
  17. *
  18. * @param array $config
  19. * @param array $options
  20. * @return \Illuminate\Redis\Connections\PhpRedisConnection
  21. */
  22. public function connect(array $config, array $options)
  23. {
  24. $formattedOptions = Arr::pull($config, 'options', []);
  25. if (isset($config['prefix'])) {
  26. $formattedOptions['prefix'] = $config['prefix'];
  27. }
  28. $connector = function () use ($config, $options, $formattedOptions) {
  29. return $this->createClient(array_merge(
  30. $config, $options, $formattedOptions
  31. ));
  32. };
  33. return new PhpRedisConnection($connector(), $connector, $config);
  34. }
  35. /**
  36. * Create a new clustered PhpRedis connection.
  37. *
  38. * @param array $config
  39. * @param array $clusterOptions
  40. * @param array $options
  41. * @return \Illuminate\Redis\Connections\PhpRedisClusterConnection
  42. */
  43. public function connectToCluster(array $config, array $clusterOptions, array $options)
  44. {
  45. $options = array_merge($options, $clusterOptions, Arr::pull($config, 'options', []));
  46. return new PhpRedisClusterConnection($this->createRedisClusterInstance(
  47. array_map($this->buildClusterConnectionString(...), $config), $options
  48. ));
  49. }
  50. /**
  51. * Build a single cluster seed string from an array.
  52. *
  53. * @param array $server
  54. * @return string
  55. */
  56. protected function buildClusterConnectionString(array $server)
  57. {
  58. return $this->formatHost($server).':'.$server['port'];
  59. }
  60. /**
  61. * Create the Redis client instance.
  62. *
  63. * @param array $config
  64. * @return \Redis
  65. *
  66. * @throws \LogicException
  67. */
  68. protected function createClient(array $config)
  69. {
  70. return tap(new Redis, function ($client) use ($config) {
  71. if ($client instanceof RedisFacade) {
  72. throw new LogicException(
  73. extension_loaded('redis')
  74. ? 'Please remove or rename the Redis facade alias in your "app" configuration file in order to avoid collision with the PHP Redis extension.'
  75. : 'Please make sure the PHP Redis extension is installed and enabled.'
  76. );
  77. }
  78. $this->establishConnection($client, $config);
  79. if (array_key_exists('max_retries', $config)) {
  80. $client->setOption(Redis::OPT_MAX_RETRIES, $config['max_retries']);
  81. }
  82. if (array_key_exists('backoff_algorithm', $config)) {
  83. $client->setOption(Redis::OPT_BACKOFF_ALGORITHM, $this->parseBackoffAlgorithm($config['backoff_algorithm']));
  84. }
  85. if (array_key_exists('backoff_base', $config)) {
  86. $client->setOption(Redis::OPT_BACKOFF_BASE, $config['backoff_base']);
  87. }
  88. if (array_key_exists('backoff_cap', $config)) {
  89. $client->setOption(Redis::OPT_BACKOFF_CAP, $config['backoff_cap']);
  90. }
  91. if (! empty($config['password'])) {
  92. if (isset($config['username']) && $config['username'] !== '' && is_string($config['password'])) {
  93. $client->auth([$config['username'], $config['password']]);
  94. } else {
  95. $client->auth($config['password']);
  96. }
  97. }
  98. if (isset($config['database'])) {
  99. $client->select((int) $config['database']);
  100. }
  101. if (! empty($config['prefix'])) {
  102. $client->setOption(Redis::OPT_PREFIX, $config['prefix']);
  103. }
  104. if (! empty($config['read_timeout'])) {
  105. $client->setOption(Redis::OPT_READ_TIMEOUT, $config['read_timeout']);
  106. }
  107. if (! empty($config['scan'])) {
  108. $client->setOption(Redis::OPT_SCAN, $config['scan']);
  109. }
  110. if (! empty($config['name'])) {
  111. $client->client('SETNAME', $config['name']);
  112. }
  113. if (array_key_exists('serializer', $config)) {
  114. $client->setOption(Redis::OPT_SERIALIZER, $config['serializer']);
  115. }
  116. if (array_key_exists('compression', $config)) {
  117. $client->setOption(Redis::OPT_COMPRESSION, $config['compression']);
  118. }
  119. if (array_key_exists('compression_level', $config)) {
  120. $client->setOption(Redis::OPT_COMPRESSION_LEVEL, $config['compression_level']);
  121. }
  122. if (defined('Redis::OPT_PACK_IGNORE_NUMBERS') &&
  123. array_key_exists('pack_ignore_numbers', $config)) {
  124. $client->setOption(Redis::OPT_PACK_IGNORE_NUMBERS, $config['pack_ignore_numbers']);
  125. }
  126. });
  127. }
  128. /**
  129. * Establish a connection with the Redis host.
  130. *
  131. * @param \Redis $client
  132. * @param array $config
  133. * @return void
  134. */
  135. protected function establishConnection($client, array $config)
  136. {
  137. $persistent = $config['persistent'] ?? false;
  138. $parameters = [
  139. $this->formatHost($config),
  140. $config['port'],
  141. Arr::get($config, 'timeout', 0.0),
  142. $persistent ? Arr::get($config, 'persistent_id', null) : null,
  143. Arr::get($config, 'retry_interval', 0),
  144. ];
  145. if (version_compare(phpversion('redis'), '3.1.3', '>=')) {
  146. $parameters[] = Arr::get($config, 'read_timeout', 0.0);
  147. }
  148. if (version_compare(phpversion('redis'), '5.3.0', '>=') && ! is_null($context = Arr::get($config, 'context'))) {
  149. $parameters[] = $context;
  150. }
  151. $client->{$persistent ? 'pconnect' : 'connect'}(...$parameters);
  152. }
  153. /**
  154. * Create a new redis cluster instance.
  155. *
  156. * @param array $servers
  157. * @param array $options
  158. * @return \RedisCluster
  159. */
  160. protected function createRedisClusterInstance(array $servers, array $options)
  161. {
  162. $parameters = [
  163. null,
  164. array_values($servers),
  165. $options['timeout'] ?? 0,
  166. $options['read_timeout'] ?? 0,
  167. isset($options['persistent']) && $options['persistent'],
  168. ];
  169. if (version_compare(phpversion('redis'), '4.3.0', '>=')) {
  170. $parameters[] = $options['password'] ?? null;
  171. }
  172. if (version_compare(phpversion('redis'), '5.3.2', '>=') && ! is_null($context = Arr::get($options, 'context'))) {
  173. $parameters[] = $context;
  174. }
  175. return tap(new RedisCluster(...$parameters), function ($client) use ($options) {
  176. if (! empty($options['prefix'])) {
  177. $client->setOption(Redis::OPT_PREFIX, $options['prefix']);
  178. }
  179. if (! empty($options['scan'])) {
  180. $client->setOption(Redis::OPT_SCAN, $options['scan']);
  181. }
  182. if (! empty($options['failover'])) {
  183. $client->setOption(RedisCluster::OPT_SLAVE_FAILOVER, $options['failover']);
  184. }
  185. if (array_key_exists('serializer', $options)) {
  186. $client->setOption(Redis::OPT_SERIALIZER, $options['serializer']);
  187. }
  188. if (array_key_exists('compression', $options)) {
  189. $client->setOption(Redis::OPT_COMPRESSION, $options['compression']);
  190. }
  191. if (array_key_exists('compression_level', $options)) {
  192. $client->setOption(Redis::OPT_COMPRESSION_LEVEL, $options['compression_level']);
  193. }
  194. });
  195. }
  196. /**
  197. * Format the host using the scheme if available.
  198. *
  199. * @param array $options
  200. * @return string
  201. */
  202. protected function formatHost(array $options)
  203. {
  204. if (isset($options['scheme'])) {
  205. return Str::start($options['host'], "{$options['scheme']}://");
  206. }
  207. return $options['host'];
  208. }
  209. /**
  210. * Parse a "friendly" backoff algorithm name into an integer.
  211. *
  212. * @param mixed $algorithm
  213. * @return int
  214. *
  215. * @throws \InvalidArgumentException
  216. */
  217. protected function parseBackoffAlgorithm(mixed $algorithm)
  218. {
  219. if (is_int($algorithm)) {
  220. return $algorithm;
  221. }
  222. return match ($algorithm) {
  223. 'default' => Redis::BACKOFF_ALGORITHM_DEFAULT,
  224. 'decorrelated_jitter' => Redis::BACKOFF_ALGORITHM_DECORRELATED_JITTER,
  225. 'equal_jitter' => Redis::BACKOFF_ALGORITHM_EQUAL_JITTER,
  226. 'exponential' => Redis::BACKOFF_ALGORITHM_EXPONENTIAL,
  227. 'uniform' => Redis::BACKOFF_ALGORITHM_UNIFORM,
  228. 'constant' => Redis::BACKOFF_ALGORITHM_CONSTANT,
  229. default => throw new InvalidArgumentException("Algorithm [{$algorithm}] is not a valid PhpRedis backoff algorithm.")
  230. };
  231. }
  232. }