init & add agent & add job

2025-12-18 18:46:43 +00:00 · 2019-01-15 10:02:28 +08:00
parent 71f1f10e2c
commit d0a4b891b5
321 changed files with 24657 additions and 1 deletions
--- a/predis/src/Connection/Factory.php
+++ b/predis/src/Connection/Factory.php
@@ -0,0 +1,188 @@
+<?php
+
+/*
+ * This file is part of the Predis package.
+ *
+ * (c) Daniele Alessandri <suppakilla@gmail.com>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Predis\Connection;
+
+use Predis\Command\RawCommand;
+
+/**
+ * Standard connection factory for creating connections to Redis nodes.
+ *
+ * @author Daniele Alessandri <suppakilla@gmail.com>
+ */
+class Factory implements FactoryInterface
+{
+    private $defaults = array();
+
+    protected $schemes = array(
+        'tcp' => 'Predis\Connection\StreamConnection',
+        'unix' => 'Predis\Connection\StreamConnection',
+        'tls' => 'Predis\Connection\StreamConnection',
+        'redis' => 'Predis\Connection\StreamConnection',
+        'rediss' => 'Predis\Connection\StreamConnection',
+        'http' => 'Predis\Connection\WebdisConnection',
+    );
+
+    /**
+     * Checks if the provided argument represents a valid connection class
+     * implementing Predis\Connection\NodeConnectionInterface. Optionally,
+     * callable objects are used for lazy initialization of connection objects.
+     *
+     * @param mixed $initializer FQN of a connection class or a callable for lazy initialization.
+     *
+     * @throws \InvalidArgumentException
+     *
+     * @return mixed
+     */
+    protected function checkInitializer($initializer)
+    {
+        if (is_callable($initializer)) {
+            return $initializer;
+        }
+
+        $class = new \ReflectionClass($initializer);
+
+        if (!$class->isSubclassOf('Predis\Connection\NodeConnectionInterface')) {
+            throw new \InvalidArgumentException(
+                'A connection initializer must be a valid connection class or a callable object.'
+            );
+        }
+
+        return $initializer;
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    public function define($scheme, $initializer)
+    {
+        $this->schemes[$scheme] = $this->checkInitializer($initializer);
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    public function undefine($scheme)
+    {
+        unset($this->schemes[$scheme]);
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    public function create($parameters)
+    {
+        if (!$parameters instanceof ParametersInterface) {
+            $parameters = $this->createParameters($parameters);
+        }
+
+        $scheme = $parameters->scheme;
+
+        if (!isset($this->schemes[$scheme])) {
+            throw new \InvalidArgumentException("Unknown connection scheme: '$scheme'.");
+        }
+
+        $initializer = $this->schemes[$scheme];
+
+        if (is_callable($initializer)) {
+            $connection = call_user_func($initializer, $parameters, $this);
+        } else {
+            $connection = new $initializer($parameters);
+            $this->prepareConnection($connection);
+        }
+
+        if (!$connection instanceof NodeConnectionInterface) {
+            throw new \UnexpectedValueException(
+                'Objects returned by connection initializers must implement '.
+                "'Predis\Connection\NodeConnectionInterface'."
+            );
+        }
+
+        return $connection;
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    public function aggregate(AggregateConnectionInterface $connection, array $parameters)
+    {
+        foreach ($parameters as $node) {
+            $connection->add($node instanceof NodeConnectionInterface ? $node : $this->create($node));
+        }
+    }
+
+    /**
+     * Assigns a default set of parameters applied to new connections.
+     *
+     * The set of parameters passed to create a new connection have precedence
+     * over the default values set for the connection factory.
+     *
+     * @param array $parameters Set of connection parameters.
+     */
+    public function setDefaultParameters(array $parameters)
+    {
+        $this->defaults = $parameters;
+    }
+
+    /**
+     * Returns the default set of parameters applied to new connections.
+     *
+     * @return array
+     */
+    public function getDefaultParameters()
+    {
+        return $this->defaults;
+    }
+
+    /**
+     * Creates a connection parameters instance from the supplied argument.
+     *
+     * @param mixed $parameters Original connection parameters.
+     *
+     * @return ParametersInterface
+     */
+    protected function createParameters($parameters)
+    {
+        if (is_string($parameters)) {
+            $parameters = Parameters::parse($parameters);
+        } else {
+            $parameters = $parameters ?: array();
+        }
+
+        if ($this->defaults) {
+            $parameters += $this->defaults;
+        }
+
+        return new Parameters($parameters);
+    }
+
+    /**
+     * Prepares a connection instance after its initialization.
+     *
+     * @param NodeConnectionInterface $connection Connection instance.
+     */
+    protected function prepareConnection(NodeConnectionInterface $connection)
+    {
+        $parameters = $connection->getParameters();
+
+        if (isset($parameters->password)) {
+            $connection->addConnectCommand(
+                new RawCommand(array('AUTH', $parameters->password))
+            );
+        }
+
+        if (isset($parameters->database)) {
+            $connection->addConnectCommand(
+                new RawCommand(array('SELECT', $parameters->database))
+            );
+        }
+    }
+}