Manager::getCurrentNode() - Code Metrics - mariano/disque-php - Measure and Improve Code Quality continuously with Scrutinizer

Manager::getCurrentNode() A
last analyzed 2017-05-30 02:39 UTC

↳ Parent: Manager

Complexity

Conditions	1
Paths	1

Size

Total Lines	4
Code Lines	2

Duplication

Lines	0
Ratio	0 %

Code Coverage

Tests	0
CRAP Score	2

Importance

Changes

Metric	Value
dl	0
loc	4
ccs	0
cts	4
cp	0
rs	10
c	0
b	0
f	0
cc	1
eloc	2
nc	1
nop	0
crap	2

<?php
namespace Disque\Connection;

use Disque\Connection\Node\Node;
use Disque\Connection\Node\NodePrioritizerInterface;
use Disque\Connection\Node\ConservativeJobCountPrioritizer;
use Disque\Command\CommandInterface;
use Disque\Command\GetJob;
use Disque\Command\Response\HelloResponse;
use Disque\Command\Response\JobsResponse;
use Disque\Connection\Factory\ConnectionFactoryInterface;
use Disque\Connection\Factory\SocketFactory;

/**
 * The connection manager connects to Disque nodes and chooses the best of them
 *
 * If there are multiple nodes to connect, the first connection is always
 * random. The manager then switches to the best nodes according to its
 * NodePriority strategy.
 *
 * If the manager knows the credentials of only one node, it will automatically
 * discover other nodes in the cluster and connect to them if needed (unless
 * they are password-protected).
 */
class Manager implements ManagerInterface
{
    /**
     * Servers we can connect to initially, without knowing the cluster
     *
     * After connecting to one, the server returns a list of other nodes
     * in the cluster so we can connect to them automatically, unless
     * the discovered nodes are secured with a password.
     *
     * 'serverAddress' => Credentials
     *
     * @var Credentials[]
     */
    protected $credentials = [];

    /**
     * A strategy to prioritize nodes and find the best one to switch to
     *
     * The default strategy is the ConservativeJobCountPrioritizer. It
     * prioritizes nodes by their job count, but prefers the current node
     * in order to avoid switching until there is a clearly better node.
     *
     * @var NodePrioritizerInterface
     */
    protected $priorityStrategy;

    /**
     * List of nodes, ie Disque instances available in the cluster
     *
     * 'nodeId' => Node
     *
     * @var Node[]
     */
    protected $nodes = [];

    /**
     * Node prefixes and their corresponding node ID
     *
     * Node prefix consists of the first 8 bytes from the node ID. Because job
     * IDs contain the node prefix, it can be used to identify on which node
     * a job lives.
     *
     * 'nodePrefix' => 'nodeId'
     *
     * @var array
     */
    protected $nodePrefixes = [];

    /**
     * The ID of the node we are currently connected to
     *
     * @var string
     */
    protected $nodeId;

    /**
     * @var ConnectionFactoryInterface
     */
    private $connectionFactory;

    public function __construct()
    {
        $this->connectionFactory = new SocketFactory();
        $this->priorityStrategy = new ConservativeJobCountPrioritizer();
    }

    /**
     * @inheritdoc
     */
    public function getConnectionFactory()
    {
        return $this->connectionFactory;
    }

    /**
     * @inheritdoc
     */
    public function setConnectionFactory(
        ConnectionFactoryInterface $connectionFactory
    ) {
        $this->connectionFactory = $connectionFactory;
    }

    /**
     * @inheritdoc
     */
    public function getCredentials()
    {
        return $this->credentials;
    }

    /**
     * @inheritdoc
     */
    public function addServer(Credentials $credentials)
    {
        $address = $credentials->getAddress();
        $this->credentials[$address] = $credentials;
    }

    /**
     * @inheritdoc
     */
    public function getPriorityStrategy()
    {
        return $this->priorityStrategy;
    }

    /**
     * @inheritdoc
     */
    public function setPriorityStrategy(NodePrioritizerInterface $priorityStrategy)
    {
        $this->priorityStrategy = $priorityStrategy;
    }

    /**
     * @inheritdoc
     */
    public function isConnected()
    {
        return (
            isset($this->nodeId) &&
            $this->getCurrentNode()->isConnected()
        );
    }

    /**
     * @inheritdoc
     */
    public function connect()
    {
        // If the manager was already connected, connect to a node from the last
        // HELLO response. This information is newer than the credentials
        // supplied by the user at the beginning.
        if ($this->wasAlreadyConnected()) {
            try {
                $this->switchNodeIfNeeded();
            } catch (ConnectionException $e) {
                // Ignore the error, we'll try reconnecting with credentials below
            }
        }

        // Use the user-supplied credentials in case this is the initial
        // connection.
        // If the reconnection attempt above didn't work, fall back
        // to the user-supplied credentials, too.
        if ( ! $this->isConnected()) {
            $currentNode = $this->findAvailableConnection();
            $this->switchToNode($currentNode);
        }

        return $this->getCurrentNode();
    }

    /**
     * @inheritdoc
     */
    public function execute(CommandInterface $command)
    {
        $this->shouldBeConnected();
        $command = $this->preprocessExecution($command);

        $response = $this->getCurrentNode()->execute($command);

        $response = $this->postprocessExecution($command, $response);
        return $response;
    }

    /**
     * @inheritdoc
     */
    public function getCurrentNode()
    {
        return $this->nodes[$this->nodeId];
    }

    /**
     * Get a functional connection to any known node
     *
     * Disque suggests the first connection should be chosen randomly
     * We go through the user-supplied credentials randomly and try to connect.
     *
     * @return Node A connected node
     *
     * @throws ConnectionException
     */
    protected function findAvailableConnection()
    {
        $servers = $this->credentials;
        shuffle($servers);
        $previous = null;

        foreach ($servers as $server) {
            try {
                $node = $this->getNodeConnection($server);
            } catch (ConnectionException $e) {
                $previous = $e;
                continue;
            }

            if ($node->isConnected()) {
                return $node;
            }
        }

        throw new ConnectionException('No servers available', 0, $previous);
    }

    /**
     * Connect to the node given in the credentials
     *
     * @param Credentials $server
     *
     * @return Node A connected node
     *
     * @throws ConnectionException
     * @throws AuthenticationException
     */
    protected function getNodeConnection(Credentials $server)
    {
        $node = $this->createNode($server);
        $node->connect();
        return $node;
    }

    /**
     * Reset node counters that should be reset upon node switch
     */
    protected function resetNodeCounters()
    {
        foreach($this->nodes as $node) {
            $node->resetJobCount();
        }
    }

    /**
     * Hook into the command execution and do anything before it's executed
     *
     * Eg. start measuring node latency etc.
     *
     * @param CommandInterface $command
     *
     * @return CommandInterface $command
     */
    protected function preprocessExecution(CommandInterface $command)
    {
        return $command;
    }

    /**
     * Postprocess the command execution, eg. update node stats
     *
     * @param CommandInterface $command
     * @param mixed            $response
     *
     * @return mixed
     * @throws ConnectionException
     */
    protected function postprocessExecution(
        CommandInterface $command,
        $response
    ) {
        if ($command instanceof GetJob) {
            $this->updateNodeStats($command->parse($response));
            $this->switchNodeIfNeeded();
        }

        return $response;
    }

    /**
     * Update node counters indicating how many jobs the node has produced
     *
     * @param array $jobs Jobs
     */
    protected function updateNodeStats(array $jobs)
    {
        foreach ($jobs as $job) {
            $jobId = $job[JobsResponse::KEY_ID];
            $nodeId = $this->getNodeIdFromJobId($jobId);
            if (!isset($nodeId) || !isset($this->nodes[$nodeId])) {
                continue;
            }

            $node = $this->nodes[$nodeId];
            $node->addJobCount(1);
        }
    }

    /**
     * Decide if we should switch to a better node
     *
     * @throws ConnectionException
     */
    private function switchNodeIfNeeded()
    {
        $sortedNodes = $this->priorityStrategy->sort(
            $this->nodes,
            $this->nodeId
        );

        $previous = null;

        // Try to connect by priority, continue on error, return on success
        foreach($sortedNodes as $nodeCandidate) {
            // If the first recommended node is our current node and it has
            // a working connection, return early.
            // If its connection is not working, let's try and reconnect further
            // below, or find the first best connected node.
            if ($nodeCandidate->getId() === $this->nodeId &&
                $nodeCandidate->isConnected()) {
                return;
            }

            try {
                if ($nodeCandidate->isConnected()) {
                    // Say a new HELLO to the node, the cluster might have changed
                    $nodeCandidate->sayHello();
                } else {
                    $nodeCandidate->connect();
                }
            } catch (ConnectionException $e) {
                $previous = $e;
                continue;
            }

            $this->switchToNode($nodeCandidate);
            return;
        }

        throw new ConnectionException('Could not switch to any node', 0, $previous);
    }

    /**
     * Get a node ID based off a Job ID
     *
     * @param string       $jobId Job ID
     * @return string|null        Node ID
     */
    private function getNodeIdFromJobId($jobId)
    {
        $nodePrefix = $this->getNodePrefixFromJobId($jobId);
        if (
            isset($this->nodePrefixes[$nodePrefix]) &&
            array_key_exists($this->nodePrefixes[$nodePrefix], $this->nodes)
        ) {
            return $this->nodePrefixes[$nodePrefix];
        }

        return null;
    }

    /**
     * Get the node prefix from the job ID
     *
     * @param string  $jobId
     * @return string        Node prefix
     */
    private function getNodePrefixFromJobId($jobId)
    {
        $nodePrefix = substr(
            $jobId,
            JobsResponse::ID_NODE_PREFIX_START,
            Node::PREFIX_LENGTH
        );

        return $nodePrefix;
    }

    /**
     * We should be connected
     *
     * @return void
     * @throws ConnectionException
     */
    protected function shouldBeConnected()
    {
        // If we lost the connection, first let's try and reconnect
        if (!$this->isConnected()) {
            try {
                $this->switchNodeIfNeeded();
            } catch (ConnectionException $e) {
                throw new ConnectionException('Not connected. ' . $e->getMessage(), 0, $e);
            }
        }
    }

    /**
     * Create a new Node object
     *
     * @param Credentials $credentials
     *
     * @return Node An unconnected Node
     */
    private function createNode(Credentials $credentials)
    {
        $host = $credentials->getHost();
        $port = $credentials->getPort();
        $connection = $this->connectionFactory->create($host, $port);

        return new Node($credentials, $connection);
    }

    /**
     * Switch to the given node and map the cluster from its HELLO
     *
     * @param Node $node
     */
    private function switchToNode(Node $node)
    {
        $nodeId = $node->getId();
        // Return early if we're trying to switch to the current node.
        if (($this->nodeId === $nodeId)) {
            // But return early only if the current node is connected to Disque.
            // If it is disconnected, we want to overwrite it with the node
            // from the method argument, because that one is connected.
            if ($this->getCurrentNode()->isConnected()) {
                return;
            }

            // Copy the stats from the now-disconnected node object
            $this->copyNodeStats($this->getCurrentNode(), $node);
        }

        $this->resetNodeCounters();

        $this->nodeId = $nodeId;
        $this->nodes[$nodeId] = $node;
        $this->revealClusterFromHello($node);
    }

    /**
     * Reveal the whole Disque cluster from a node HELLO response
     *
     * The HELLO response from a Disque node contains addresses of all other
     * nodes in the cluster. We want to learn about them and save them, so that
     * we can switch to them later, if needed.
     *
     * @param Node $node The current node
     */
    private function revealClusterFromHello(Node $node)
    {
        $hello = $node->getHello();
        $revealedNodes = [];

        foreach ($hello[HelloResponse::NODES] as $node) {
            $id = $node[HelloResponse::NODE_ID];
            $revealedNode = $this->revealNodeFromHello($id, $node);

            // Update or set the node's priority as determined by Disque
            $priority = $node[HelloResponse::NODE_PRIORITY];
            $revealedNode->setPriority($priority);

            $revealedNodes[$id] = $revealedNode;
        }

        $this->nodes = $revealedNodes;
    }

    /**
     * Reveal a single node from a HELLO response, or use an existing node
     *
     * @param string $nodeId The node ID
     * @param array  $node   Node information as returned by the HELLO command
     *
     * @return Node $node A node in the current cluster
     */
    private function revealNodeFromHello($nodeId, array $node)
    {
        /**
         * Add the node prefix to the pool. We create the prefix manually
         * from the node ID rather than asking the Node object. Newly created
         * Nodes aren't connected and thus don't know their ID or prefix.
         *
         * @see Node::sayHello()
         */
        $prefix = substr($nodeId, Node::PREFIX_START, Node::PREFIX_LENGTH);
        $this->nodePrefixes[$prefix] = $nodeId;

        // If the node already exists in our pool, use it, don't overwrite it
        // with a new one. We would lose its stats and connection.
        if (isset($this->nodes[$nodeId])) {
            return $this->nodes[$nodeId];
        }

        $host = $node[HelloResponse::NODE_HOST];
        $port = $node[HelloResponse::NODE_PORT];
        $credentials = new Credentials($host, $port);

        $address = $credentials->getAddress();
        // If there are user-supplied credentials for this node, use them.
        // They may contain a password
        if (isset($this->credentials[$address])) {
            $credentials = $this->credentials[$address];
        }

        // Instantiate a new Node object for the newly revealed node
        return $this->createNode($credentials);
    }

    /**
     * Check if the manager held a connection to Disque already
     *
     * @return bool
     */
    private function wasAlreadyConnected()
    {
        return ( ! empty($this->nodes));
    }

    /**
     * Copy node stats from the old to the new node
     *
     * @param Node $oldNode
     * @param Node $newNode
     */
    private function copyNodeStats(Node $oldNode, Node $newNode)
    {
        $oldNodeJobCount = $oldNode->getTotalJobCount();
        $newNode->addJobCount($oldNodeJobCount);
    }
}

1			<?php
2			namespace Disque\Connection;
3
4			use Disque\Connection\Node\Node;
5			use Disque\Connection\Node\NodePrioritizerInterface;
6			use Disque\Connection\Node\ConservativeJobCountPrioritizer;
7			use Disque\Command\CommandInterface;
8			use Disque\Command\GetJob;
9			use Disque\Command\Response\HelloResponse;
10			use Disque\Command\Response\JobsResponse;
11			use Disque\Connection\Factory\ConnectionFactoryInterface;
12			use Disque\Connection\Factory\SocketFactory;
13
14			/**
15			* The connection manager connects to Disque nodes and chooses the best of them
16			*
17			* If there are multiple nodes to connect, the first connection is always
18			* random. The manager then switches to the best nodes according to its
19			* NodePriority strategy.
20			*
21			* If the manager knows the credentials of only one node, it will automatically
22			* discover other nodes in the cluster and connect to them if needed (unless
23			* they are password-protected).
24			*/
25			class Manager implements ManagerInterface
26			{
27			/**
28			* Servers we can connect to initially, without knowing the cluster
29			*
30			* After connecting to one, the server returns a list of other nodes
31			* in the cluster so we can connect to them automatically, unless
32			* the discovered nodes are secured with a password.
33			*
34			* 'serverAddress' => Credentials
35			*
36			* @var Credentials[]
37			*/
38			protected $credentials = [];
39
40			/**
41			* A strategy to prioritize nodes and find the best one to switch to
42			*
43			* The default strategy is the ConservativeJobCountPrioritizer. It
44			* prioritizes nodes by their job count, but prefers the current node
45			* in order to avoid switching until there is a clearly better node.
46			*
47			* @var NodePrioritizerInterface
48			*/
49			protected $priorityStrategy;
50
51			/**
52			* List of nodes, ie Disque instances available in the cluster
53			*
54			* 'nodeId' => Node
55			*
56			* @var Node[]
57			*/
58			protected $nodes = [];
59
60			/**
61			* Node prefixes and their corresponding node ID
62			*
63			* Node prefix consists of the first 8 bytes from the node ID. Because job
64			* IDs contain the node prefix, it can be used to identify on which node
65			* a job lives.
66			*
67			* 'nodePrefix' => 'nodeId'
68			*
69			* @var array
70			*/
71			protected $nodePrefixes = [];
72
73			/**
74			* The ID of the node we are currently connected to
75			*
76			* @var string
77			*/
78			protected $nodeId;
79
80			/**
81			* @var ConnectionFactoryInterface
82			*/
83			private $connectionFactory;
84
85			public function __construct()
86			{
87			$this->connectionFactory = new SocketFactory();
88			$this->priorityStrategy = new ConservativeJobCountPrioritizer();
89			}
90
91			/**
92			* @inheritdoc
93			*/
94			public function getConnectionFactory()
95			{
96			return $this->connectionFactory;
97			}
98
99			/**
100			* @inheritdoc
101			*/
102			public function setConnectionFactory(
103			ConnectionFactoryInterface $connectionFactory
104			) {
105			$this->connectionFactory = $connectionFactory;
106			}
107
108			/**
109			* @inheritdoc
110			*/
111			public function getCredentials()
112			{
113			return $this->credentials;
114			}
115
116			/**
117			* @inheritdoc
118			*/
119			public function addServer(Credentials $credentials)
120			{
121			$address = $credentials->getAddress();
122			$this->credentials[$address] = $credentials;
123			}
124
125			/**
126			* @inheritdoc
127			*/
128			public function getPriorityStrategy()
129			{
130			return $this->priorityStrategy;
131			}
132
133			/**
134			* @inheritdoc
135			*/
136			public function setPriorityStrategy(NodePrioritizerInterface $priorityStrategy)
137			{
138			$this->priorityStrategy = $priorityStrategy;
139			}
140
141			/**
142			* @inheritdoc
143			*/
144			public function isConnected()
145			{
146			return (
147			isset($this->nodeId) &&
148			$this->getCurrentNode()->isConnected()
149			);
150			}
151
152			/**
153			* @inheritdoc
154			*/
155			public function connect()
156			{
157			// If the manager was already connected, connect to a node from the last
158			// HELLO response. This information is newer than the credentials
159			// supplied by the user at the beginning.
160			if ($this->wasAlreadyConnected()) {
161			try {
162			$this->switchNodeIfNeeded();
163			} catch (ConnectionException $e) {
164			// Ignore the error, we'll try reconnecting with credentials below
165			}
166			}
167
168			// Use the user-supplied credentials in case this is the initial
169			// connection.
170			// If the reconnection attempt above didn't work, fall back
171			// to the user-supplied credentials, too.
172			if ( ! $this->isConnected()) {
173			$currentNode = $this->findAvailableConnection();
174			$this->switchToNode($currentNode);
175			}
176
177			return $this->getCurrentNode();
178			}
179
180			/**
181			* @inheritdoc
182			*/
183			public function execute(CommandInterface $command)
184			{
185			$this->shouldBeConnected();
186			$command = $this->preprocessExecution($command);
187
188			$response = $this->getCurrentNode()->execute($command);
189
190			$response = $this->postprocessExecution($command, $response);
191			return $response;
192			}
193
194			/**
195			* @inheritdoc
196			*/
197			public function getCurrentNode()
198			{
199			return $this->nodes[$this->nodeId];
200			}
201
202			/**
203			* Get a functional connection to any known node
204			*
205			* Disque suggests the first connection should be chosen randomly
206			* We go through the user-supplied credentials randomly and try to connect.
207			*
208			* @return Node A connected node
209			*
210			* @throws ConnectionException
211			*/
212			protected function findAvailableConnection()
213			{
214			$servers = $this->credentials;
215			shuffle($servers);
216			$previous = null;
217
218			foreach ($servers as $server) {
219			try {
220			$node = $this->getNodeConnection($server);
221			} catch (ConnectionException $e) {
222			$previous = $e;
223			continue;
224			}
225
226			if ($node->isConnected()) {
227			return $node;
228			}
229			}
230
231			throw new ConnectionException('No servers available', 0, $previous);
232			}
233
234			/**
235			* Connect to the node given in the credentials
236			*
237			* @param Credentials $server
238			*
239			* @return Node A connected node
240			*
241			* @throws ConnectionException
242			* @throws AuthenticationException
243			*/
244			protected function getNodeConnection(Credentials $server)
245			{
246			$node = $this->createNode($server);
247			$node->connect();
248			return $node;
249			}
250
251			/**
252			* Reset node counters that should be reset upon node switch
253			*/
254			protected function resetNodeCounters()
255			{
256			foreach($this->nodes as $node) {
257			$node->resetJobCount();
258			}
259			}
260
261			/**
262			* Hook into the command execution and do anything before it's executed
263			*
264			* Eg. start measuring node latency etc.
265			*
266			* @param CommandInterface $command
267			*
268			* @return CommandInterface $command
269			*/
270			protected function preprocessExecution(CommandInterface $command)
271			{
272			return $command;
273			}
274
275			/**
276			* Postprocess the command execution, eg. update node stats
277			*
278			* @param CommandInterface $command
279			* @param mixed $response
280			*
281			* @return mixed
282			* @throws ConnectionException
283			*/
284			protected function postprocessExecution(
285			CommandInterface $command,
286			$response
287			) {
288			if ($command instanceof GetJob) {
289			$this->updateNodeStats($command->parse($response));
290			$this->switchNodeIfNeeded();
291			}
292
293			return $response;
294			}
295
296			/**
297			* Update node counters indicating how many jobs the node has produced
298			*
299			* @param array $jobs Jobs
300			*/
301			protected function updateNodeStats(array $jobs)
302			{
303			foreach ($jobs as $job) {
304			$jobId = $job[JobsResponse::KEY_ID];
305			$nodeId = $this->getNodeIdFromJobId($jobId);
306			if (!isset($nodeId) \|\| !isset($this->nodes[$nodeId])) {
307			continue;
308			}
309
310			$node = $this->nodes[$nodeId];
311			$node->addJobCount(1);
312			}
313			}
314
315			/**
316			* Decide if we should switch to a better node
317			*
318			* @throws ConnectionException
319			*/
320			private function switchNodeIfNeeded()
321			{
322			$sortedNodes = $this->priorityStrategy->sort(
323			$this->nodes,
324			$this->nodeId
325			);
326
327			$previous = null;
328
329			// Try to connect by priority, continue on error, return on success
330			foreach($sortedNodes as $nodeCandidate) {
331			// If the first recommended node is our current node and it has
332			// a working connection, return early.
333			// If its connection is not working, let's try and reconnect further
334			// below, or find the first best connected node.
335			if ($nodeCandidate->getId() === $this->nodeId &&
336			$nodeCandidate->isConnected()) {
337			return;
338			}
339
340			try {
341			if ($nodeCandidate->isConnected()) {
342			// Say a new HELLO to the node, the cluster might have changed
343			$nodeCandidate->sayHello();
344			} else {
345			$nodeCandidate->connect();
346			}
347			} catch (ConnectionException $e) {
348			$previous = $e;
349			continue;
350			}
351
352			$this->switchToNode($nodeCandidate);
353			return;
354			}
355
356			throw new ConnectionException('Could not switch to any node', 0, $previous);
357			}
358
359			/**
360			* Get a node ID based off a Job ID
361			*
362			* @param string $jobId Job ID
363			* @return string\|null Node ID
364			*/
365			private function getNodeIdFromJobId($jobId)
366			{
367			$nodePrefix = $this->getNodePrefixFromJobId($jobId);
368			if (
369			isset($this->nodePrefixes[$nodePrefix]) &&
370			array_key_exists($this->nodePrefixes[$nodePrefix], $this->nodes)
371			) {
372			return $this->nodePrefixes[$nodePrefix];
373			}
374
375			return null;
376			}
377
378			/**
379			* Get the node prefix from the job ID
380			*
381			* @param string $jobId
382			* @return string Node prefix
383			*/
384			private function getNodePrefixFromJobId($jobId)
385			{
386			$nodePrefix = substr(
387			$jobId,
388			JobsResponse::ID_NODE_PREFIX_START,
389			Node::PREFIX_LENGTH
390			);
391
392			return $nodePrefix;
393			}
394
395			/**
396			* We should be connected
397			*
398			* @return void
399			* @throws ConnectionException
400			*/
401			protected function shouldBeConnected()
402			{
403			// If we lost the connection, first let's try and reconnect
404			if (!$this->isConnected()) {
405			try {
406			$this->switchNodeIfNeeded();
407			} catch (ConnectionException $e) {
408			throw new ConnectionException('Not connected. ' . $e->getMessage(), 0, $e);
409			}
410			}
411			}
412
413			/**
414			* Create a new Node object
415			*
416			* @param Credentials $credentials
417			*
418			* @return Node An unconnected Node
419			*/
420			private function createNode(Credentials $credentials)
421			{
422			$host = $credentials->getHost();
423			$port = $credentials->getPort();
424			$connection = $this->connectionFactory->create($host, $port);
425
426			return new Node($credentials, $connection);
427			}
428
429			/**
430			* Switch to the given node and map the cluster from its HELLO
431			*
432			* @param Node $node
433			*/
434			private function switchToNode(Node $node)
435			{
436			$nodeId = $node->getId();
437			// Return early if we're trying to switch to the current node.
438			if (($this->nodeId === $nodeId)) {
439			// But return early only if the current node is connected to Disque.
440			// If it is disconnected, we want to overwrite it with the node
441			// from the method argument, because that one is connected.
442			if ($this->getCurrentNode()->isConnected()) {
443			return;
444			}
445
446			// Copy the stats from the now-disconnected node object
447			$this->copyNodeStats($this->getCurrentNode(), $node);
448			}
449
450			$this->resetNodeCounters();
451
452			$this->nodeId = $nodeId;
453			$this->nodes[$nodeId] = $node;
454			$this->revealClusterFromHello($node);
455			}
456
457			/**
458			* Reveal the whole Disque cluster from a node HELLO response
459			*
460			* The HELLO response from a Disque node contains addresses of all other
461			* nodes in the cluster. We want to learn about them and save them, so that
462			* we can switch to them later, if needed.
463			*
464			* @param Node $node The current node
465			*/
466			private function revealClusterFromHello(Node $node)
467			{
468			$hello = $node->getHello();
469			$revealedNodes = [];
470
471			foreach ($hello[HelloResponse::NODES] as $node) {
472			$id = $node[HelloResponse::NODE_ID];
473			$revealedNode = $this->revealNodeFromHello($id, $node);
474
475			// Update or set the node's priority as determined by Disque
476			$priority = $node[HelloResponse::NODE_PRIORITY];
477			$revealedNode->setPriority($priority);
478
479			$revealedNodes[$id] = $revealedNode;
480			}
481
482			$this->nodes = $revealedNodes;
483			}
484
485			/**
486			* Reveal a single node from a HELLO response, or use an existing node
487			*
488			* @param string $nodeId The node ID
489			* @param array $node Node information as returned by the HELLO command
490			*
491			* @return Node $node A node in the current cluster
492			*/
493			private function revealNodeFromHello($nodeId, array $node)
494			{
495			/**
496			* Add the node prefix to the pool. We create the prefix manually
497			* from the node ID rather than asking the Node object. Newly created
498			* Nodes aren't connected and thus don't know their ID or prefix.
499			*
500			* @see Node::sayHello()
501			*/
502			$prefix = substr($nodeId, Node::PREFIX_START, Node::PREFIX_LENGTH);
503			$this->nodePrefixes[$prefix] = $nodeId;
504
505			// If the node already exists in our pool, use it, don't overwrite it
506			// with a new one. We would lose its stats and connection.
507			if (isset($this->nodes[$nodeId])) {
508			return $this->nodes[$nodeId];
509			}
510
511			$host = $node[HelloResponse::NODE_HOST];
512			$port = $node[HelloResponse::NODE_PORT];
513			$credentials = new Credentials($host, $port);
514
515			$address = $credentials->getAddress();
516			// If there are user-supplied credentials for this node, use them.
517			// They may contain a password
518			if (isset($this->credentials[$address])) {
519			$credentials = $this->credentials[$address];
520			}
521
522			// Instantiate a new Node object for the newly revealed node
523			return $this->createNode($credentials);
524			}
525
526			/**
527			* Check if the manager held a connection to Disque already
528			*
529			* @return bool
530			*/
531			private function wasAlreadyConnected()
532			{
533			return ( ! empty($this->nodes));
534			}
535
536			/**
537			* Copy node stats from the old to the new node
538			*
539			* @param Node $oldNode
540			* @param Node $newNode
541			*/
542			private function copyNodeStats(Node $oldNode, Node $newNode)
543			{
544			$oldNodeJobCount = $oldNode->getTotalJobCount();
545			$newNode->addJobCount($oldNodeJobCount);
546			}
547			}

mariano / disque-php

Manager::getCurrentNode() A last analyzed 2017-05-30 02:39 UTC

Complexity

Size

Duplication

Code Coverage

Importance

Duplication Side-by-Side

Filter issues like

Manager::getCurrentNode() A
last analyzed 2017-05-30 02:39 UTC