PhpRedisConnector.php 9.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283
  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 (! empty($config['tcp_keepalive'])) {
  123. $client->setOption(Redis::OPT_TCP_KEEPALIVE, $config['tcp_keepalive']);
  124. }
  125. if (defined('Redis::OPT_PACK_IGNORE_NUMBERS') &&
  126. array_key_exists('pack_ignore_numbers', $config)) {
  127. $client->setOption(Redis::OPT_PACK_IGNORE_NUMBERS, $config['pack_ignore_numbers']);
  128. }
  129. });
  130. }
  131. /**
  132. * Establish a connection with the Redis host.
  133. *
  134. * @param \Redis $client
  135. * @param array $config
  136. * @return void
  137. */
  138. protected function establishConnection($client, array $config)
  139. {
  140. $persistent = $config['persistent'] ?? false;
  141. $parameters = [
  142. $this->formatHost($config),
  143. $config['port'],
  144. Arr::get($config, 'timeout', 0.0),
  145. $persistent ? Arr::get($config, 'persistent_id', null) : null,
  146. Arr::get($config, 'retry_interval', 0),
  147. ];
  148. if (version_compare(phpversion('redis'), '3.1.3', '>=')) {
  149. $parameters[] = Arr::get($config, 'read_timeout', 0.0);
  150. }
  151. if (version_compare(phpversion('redis'), '5.3.0', '>=') && ! is_null($context = Arr::get($config, 'context'))) {
  152. $parameters[] = $context;
  153. }
  154. $client->{$persistent ? 'pconnect' : 'connect'}(...$parameters);
  155. }
  156. /**
  157. * Create a new redis cluster instance.
  158. *
  159. * @param array $servers
  160. * @param array $options
  161. * @return \RedisCluster
  162. */
  163. protected function createRedisClusterInstance(array $servers, array $options)
  164. {
  165. $parameters = [
  166. null,
  167. array_values($servers),
  168. $options['timeout'] ?? 0,
  169. $options['read_timeout'] ?? 0,
  170. isset($options['persistent']) && $options['persistent'],
  171. ];
  172. if (version_compare(phpversion('redis'), '4.3.0', '>=')) {
  173. $parameters[] = $options['password'] ?? null;
  174. }
  175. if (version_compare(phpversion('redis'), '5.3.2', '>=') && ! is_null($context = Arr::get($options, 'context'))) {
  176. $parameters[] = $context;
  177. }
  178. return tap(new RedisCluster(...$parameters), function ($client) use ($options) {
  179. if (! empty($options['prefix'])) {
  180. $client->setOption(Redis::OPT_PREFIX, $options['prefix']);
  181. }
  182. if (! empty($options['scan'])) {
  183. $client->setOption(Redis::OPT_SCAN, $options['scan']);
  184. }
  185. if (! empty($options['failover'])) {
  186. $client->setOption(RedisCluster::OPT_SLAVE_FAILOVER, $options['failover']);
  187. }
  188. if (array_key_exists('serializer', $options)) {
  189. $client->setOption(Redis::OPT_SERIALIZER, $options['serializer']);
  190. }
  191. if (array_key_exists('compression', $options)) {
  192. $client->setOption(Redis::OPT_COMPRESSION, $options['compression']);
  193. }
  194. if (array_key_exists('compression_level', $options)) {
  195. $client->setOption(Redis::OPT_COMPRESSION_LEVEL, $options['compression_level']);
  196. }
  197. if (! empty($options['tcp_keepalive'])) {
  198. $client->setOption(Redis::OPT_TCP_KEEPALIVE, $options['tcp_keepalive']);
  199. }
  200. });
  201. }
  202. /**
  203. * Format the host using the scheme if available.
  204. *
  205. * @param array $options
  206. * @return string
  207. */
  208. protected function formatHost(array $options)
  209. {
  210. if (isset($options['scheme'])) {
  211. return Str::start($options['host'], "{$options['scheme']}://");
  212. }
  213. return $options['host'];
  214. }
  215. /**
  216. * Parse a "friendly" backoff algorithm name into an integer.
  217. *
  218. * @param mixed $algorithm
  219. * @return int
  220. *
  221. * @throws \InvalidArgumentException
  222. */
  223. protected function parseBackoffAlgorithm(mixed $algorithm)
  224. {
  225. if (is_int($algorithm)) {
  226. return $algorithm;
  227. }
  228. return match ($algorithm) {
  229. 'default' => Redis::BACKOFF_ALGORITHM_DEFAULT,
  230. 'decorrelated_jitter' => Redis::BACKOFF_ALGORITHM_DECORRELATED_JITTER,
  231. 'equal_jitter' => Redis::BACKOFF_ALGORITHM_EQUAL_JITTER,
  232. 'exponential' => Redis::BACKOFF_ALGORITHM_EXPONENTIAL,
  233. 'uniform' => Redis::BACKOFF_ALGORITHM_UNIFORM,
  234. 'constant' => Redis::BACKOFF_ALGORITHM_CONSTANT,
  235. default => throw new InvalidArgumentException("Algorithm [{$algorithm}] is not a valid PhpRedis backoff algorithm.")
  236. };
  237. }
  238. }