| 1: | <?php |
| 2: | |
| 3: | declare(strict_types=1); |
| 4: | |
| 5: | /** |
| 6: | * Copyright OpenSearch Contributors |
| 7: | * SPDX-License-Identifier: Apache-2.0 |
| 8: | * |
| 9: | * OpenSearch PHP client |
| 10: | * |
| 11: | * @link https://github.com/opensearch-project/opensearch-php/ |
| 12: | * @copyright Copyright (c) Elasticsearch B.V (https://www.elastic.co) |
| 13: | * @license http://www.apache.org/licenses/LICENSE-2.0 Apache License, Version 2.0 |
| 14: | * @license https://www.gnu.org/licenses/lgpl-2.1.html GNU Lesser General Public License, Version 2.1 |
| 15: | * |
| 16: | * Licensed to Elasticsearch B.V under one or more agreements. |
| 17: | * Elasticsearch B.V licenses this file to you under the Apache 2.0 License or |
| 18: | * the GNU Lesser General Public License, Version 2.1, at your option. |
| 19: | * See the LICENSE file in the project root for more information. |
| 20: | */ |
| 21: | |
| 22: | namespace OpenSearch\ConnectionPool\Selectors; |
| 23: | |
| 24: | use OpenSearch\Connections\ConnectionInterface; |
| 25: | |
| 26: | class StickyRoundRobinSelector implements SelectorInterface |
| 27: | { |
| 28: | /** |
| 29: | * @var int |
| 30: | */ |
| 31: | private $current = 0; |
| 32: | |
| 33: | /** |
| 34: | * @var int |
| 35: | */ |
| 36: | private $currentCounter = 0; |
| 37: | |
| 38: | /** |
| 39: | * Use current connection unless it is dead, otherwise round-robin |
| 40: | * |
| 41: | * @param ConnectionInterface[] $connections Array of connections to choose from |
| 42: | */ |
| 43: | public function select(array $connections): ConnectionInterface |
| 44: | { |
| 45: | /** |
| 46: | * @var ConnectionInterface[] $connections |
| 47: | */ |
| 48: | if ($connections[$this->current]->isAlive()) { |
| 49: | return $connections[$this->current]; |
| 50: | } |
| 51: | |
| 52: | $this->currentCounter += 1; |
| 53: | $this->current = $this->currentCounter % count($connections); |
| 54: | |
| 55: | return $connections[$this->current]; |
| 56: | } |
| 57: | } |
| 58: |