Adding the ability to use tags servers (#15)

* adding ability to use proxy servers

* added tests for changes

* code style fixes

* trying to fix building

* another try

* fix dependencies and tests

* change readme

* fix readme

* grammar fix

* remove proxy servers and add tags for servers

* code style fixes

* code style fix

* add missing tests

* fix test

* add missing tests

* fix method name in readme

* some refactoring

* refactor variables name with random server hostname

Author: romanpravda

.travis.yml

@@ -23,7 +23,7 @@ before_script:
   - make && make install
   - cd ../../
 
-script: phpunit --coverage-clover ./tests/logs/clover.xml
+script: vendor/bin/phpunit --coverage-clover ./tests/logs/clover.xml
 
 after_script:
  - php vendor/bin/php-coveralls -v

README.md

@@ -72,6 +72,29 @@ By default client will use random server in given list of servers or in specifie
 $client->using('server-2')->select('select * from table');
 ```
 
+## Server tags
+
+```php
+$firstServerOptionsWithTag = (new \Tinderbox\Clickhouse\Common\ServerOptions())->setTag('tag');
+$secondServerOptionsWithAnotherTag = (new \Tinderbox\Clickhouse\Common\ServerOptions())->setTag('another-tag');
+
+$server = new Tinderbox\Clickhouse\Server('127.0.0.1', '8123', 'default', 'user', 'pass', $firstServerOptionsWithTag);
+
+$cluster = new Tinderbox\Clickhouse\Cluster('cluster', [
+    new Tinderbox\Clickhouse\Server('127.0.0.2', '8123', 'default', 'user', 'pass', $secondServerOptionsWithAnotherTag)
+]);
+
+$serverProvider = (new Tinderbox\Clickhouse\ServerProvider())->addServer($server)->addCluster($cluster);
+
+$client = (new Tinderbox\Clickhouse\Client($serverProvider));
+```
+
+To use server with tag, you should call ```usingServerWithTag``` function before execute any query.
+
+```php
+$client->usingServerWithTag('tag');
+```
+
 ## Select queries
 
 Any SELECT query will return instance of `Result`. This class implements interfaces `\ArrayAccess`, `\Countable` и `\Iterator`,

composer.json

@@ -19,16 +19,17 @@
   },
   "require": {
     "php": "~7.1",
-    "guzzlehttp/guzzle": "~6.0"
+    "guzzlehttp/guzzle": "~6.0",
+    "satooshi/php-coveralls": "^2.2"
   },
   "bin": [
     "bin/ccat_linux",
     "bin/ccat_darwin"
   ],
   "require-dev": {
     "mockery/mockery": "^0.9.9",
-    "phpunit/php-code-coverage": "^5.2",
-    "phpunit/phpunit": "~4.0||~5.0||~6.0"
+    "phpunit/php-code-coverage": "^6.0",
+    "phpunit/phpunit": "~7.0"
   },
   "scripts": {
     "test": "phpunit --coverage-text --colors=never"

src/Client.php

@@ -147,6 +147,26 @@ public function usingRandomServer()
         return $this;
     }
 
+    /**
+     * Client will use server with tag as server for queries.
+     *
+     * @var string
+     *
+     * @return $this
+     */
+    public function usingServerWithTag(string $tag)
+    {
+        $this->serverHostname = function () use ($tag) {
+            if ($this->isOnCluster()) {
+                return $this->serverProvider->getRandomServerFromClusterByTag($this->getClusterName(), $tag);
+            } else {
+                return $this->serverProvider->getRandomServerWithTag($tag);
+            }
+        };
+
+        return $this;
+    }
+
     /**
      * Returns true if cluster selected.
      *

src/Cluster.php

@@ -23,6 +23,13 @@ class Cluster
      */
     protected $servers = [];
 
+    /**
+     * Servers in cluster by tags.
+     *
+     * @var \Tinderbox\Clickhouse\Server[][]
+     */
+    protected $serversByTags = [];
+
     /**
      * Cluster constructor.
      *
@@ -86,6 +93,12 @@ public function addServer(string $hostname, Server $server)
         }
 
         $this->servers[$hostname] = $server;
+
+        $serverTags = $server->getOptions()->getTags();
+
+        foreach ($serverTags as $serverTag) {
+            $this->serversByTags[$serverTag][$hostname] = true;
+        }
     }
 
     /**
@@ -98,6 +111,24 @@ public function getServers(): array
         return $this->servers;
     }
 
+    /**
+     * Returns servers in cluster by tag.
+     *
+     * @param string $tag
+     *
+     * @throws ClusterException
+     *
+     * @return \Tinderbox\Clickhouse\Server[]
+     */
+    public function getServersByTag(string $tag): array
+    {
+        if (!isset($this->serversByTags[$tag])) {
+            throw ClusterException::tagNotFound($tag);
+        }
+
+        return $this->serversByTags[$tag];
+    }
+
     /**
      * Returns server by specified hostname.
      *
@@ -121,7 +152,7 @@ public function getServerByHostname(string $hostname): Server
      *
      * @return string
      */
-    public function getName() : string
+    public function getName(): string
     {
         return $this->name;
     }

src/Common/File.php

@@ -8,7 +8,7 @@
 
 class File extends AbstractFile implements FileInterface
 {
-    public function open(bool $gzip = true) : StreamInterface
+    public function open(bool $gzip = true): StreamInterface
     {
         $handle = fopen($this->getSource(), 'r');
 

src/Common/FileFromString.php

@@ -8,7 +8,7 @@
 
 class FileFromString extends AbstractFile implements FileInterface
 {
-    public function open(bool $gzip = true) : StreamInterface
+    public function open(bool $gzip = true): StreamInterface
     {
         $handle = fopen('php://memory', 'r+');
         fwrite($handle, $this->source);

src/Common/MergedFiles.php

@@ -14,7 +14,7 @@ protected function getCcatPath(): string
         return __DIR__.DIRECTORY_SEPARATOR.'..'.DIRECTORY_SEPARATOR.'..'.DIRECTORY_SEPARATOR.'bin'.DIRECTORY_SEPARATOR.'ccat_'.strtolower(PHP_OS);
     }
 
-    public function open(bool $gzip = true) : StreamInterface
+    public function open(bool $gzip = true): StreamInterface
     {
         $descriptorspec = [
             0 => ['pipe', 'r'],

src/Common/ServerOptions.php

@@ -14,6 +14,13 @@ class ServerOptions
      */
     protected $protocol = 'http';
 
+    /**
+     * Tags.
+     *
+     * @var array
+     */
+    protected $tags = [];
+
     /**
      * Sets protocol.
      *
@@ -37,4 +44,46 @@ public function getProtocol(): string
     {
         return $this->protocol;
     }
+
+    /**
+     * Set tags.
+     *
+     * @param array $tags
+     *
+     * @return ServerOptions
+     */
+    public function setTags(array $tags): self
+    {
+        $this->tags = [];
+
+        foreach ($tags as $tag) {
+            $this->addTag($tag);
+        }
+
+        return $this;
+    }
+
+    /**
+     * Adds tag.
+     *
+     * @param string $tag
+     *
+     * @return ServerOptions
+     */
+    public function addTag(string $tag): self
+    {
+        $this->tags[$tag] = true;
+
+        return $this;
+    }
+
+    /**
+     * Returns tags.
+     *
+     * @return array
+     */
+    public function getTags(): array
+    {
+        return array_keys($this->tags);
+    }
 }

src/Common/TempTable.php

@@ -94,7 +94,7 @@ public function getFormat(): string
         return $this->format;
     }
 
-    public function open(bool $gzip = true) : StreamInterface
+    public function open(bool $gzip = true): StreamInterface
     {
         return $this->source->open($gzip);
     }