ConnectionParameters.php 3.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111
  1. <?php
  2. namespace Predis;
  3. class ConnectionParameters {
  4. private $_parameters;
  5. private static $_sharedOptions;
  6. public function __construct($parameters = null) {
  7. $parameters = $parameters ?: array();
  8. $this->_parameters = is_array($parameters)
  9. ? $this->filter($parameters)
  10. : $this->parseURI($parameters);
  11. }
  12. private static function paramsExtractor($params, $kv) {
  13. @list($k, $v) = explode('=', $kv);
  14. $params[$k] = $v;
  15. return $params;
  16. }
  17. private static function getSharedOptions() {
  18. if (isset(self::$_sharedOptions)) {
  19. return self::$_sharedOptions;
  20. }
  21. $optEmpty = new Options\Option();
  22. $optBoolean = new Options\CustomOption(array(
  23. 'validate' => function($value) { return (bool) $value; },
  24. 'default' => function() { return false; },
  25. ));
  26. self::$_sharedOptions = array(
  27. 'scheme' => new Options\CustomOption(array(
  28. 'default' => function() { return 'tcp'; },
  29. )),
  30. 'host' => new Options\CustomOption(array(
  31. 'default' => function() { return '127.0.0.1'; },
  32. )),
  33. 'port' => new Options\CustomOption(array(
  34. 'validate' => function($value) { return (int) $value; },
  35. 'default' => function() { return 6379; },
  36. )),
  37. 'path' => $optEmpty,
  38. 'database' => $optEmpty,
  39. 'password' => $optEmpty,
  40. 'connection_async' => $optBoolean,
  41. 'connection_persistent' => $optBoolean,
  42. 'connection_timeout' => new Options\CustomOption(array(
  43. 'default' => function() { return 5; },
  44. )),
  45. 'read_write_timeout' => $optEmpty,
  46. 'alias' => $optEmpty,
  47. 'weight' => $optEmpty,
  48. );
  49. return self::$_sharedOptions;
  50. }
  51. protected function parseURI($uri) {
  52. if (!is_string($uri)) {
  53. throw new \InvalidArgumentException('URI must be a string');
  54. }
  55. if (stripos($uri, 'unix') === 0) {
  56. // Hack to support URIs for UNIX sockets with minimal effort.
  57. $uri = str_ireplace('unix:///', 'unix://localhost/', $uri);
  58. }
  59. $parsed = @parse_url($uri);
  60. if ($parsed == false || !isset($parsed['host'])) {
  61. throw new ClientException("Invalid URI: $uri");
  62. }
  63. if (array_key_exists('query', $parsed)) {
  64. $query = explode('&', $parsed['query']);
  65. $parsed = array_reduce($query, 'self::paramsExtractor', $parsed);
  66. }
  67. return $this->filter($parsed);
  68. }
  69. protected function filter($parameters) {
  70. $handlers = self::getSharedOptions();
  71. foreach ($parameters as $parameter => $value) {
  72. if (isset($handlers[$parameter])) {
  73. $parameters[$parameter] = $handlers[$parameter]($value);
  74. }
  75. }
  76. return $parameters;
  77. }
  78. private function tryInitializeValue($parameter) {
  79. if (isset(self::$_sharedOptions[$parameter])) {
  80. $value = self::$_sharedOptions[$parameter]->getDefault();
  81. $this->_parameters[$parameter] = $value;
  82. return $value;
  83. }
  84. }
  85. public function __get($parameter) {
  86. if (isset($this->_parameters[$parameter])) {
  87. return $this->_parameters[$parameter];
  88. }
  89. return $this->tryInitializeValue($parameter);
  90. }
  91. public function __isset($parameter) {
  92. if (isset($this->_parameters[$parameter])) {
  93. return true;
  94. }
  95. $value = $this->tryInitializeValue($parameter);
  96. return isset($value);
  97. }
  98. }