Add TCP driver

Author: Paweł Niedzielski

src/InfluxDB/Driver/AbstractSocketDriver.php

@@ -0,0 +1,97 @@
+<?php
+/**
+ * @author Stephen "TheCodeAssassin" Hoogendijk
+ */
+
+namespace InfluxDB\Driver;
+
+/**
+ * Class UDP
+ *
+ * @package InfluxDB\Driver
+ */
+abstract class AbstractSocketDriver implements DriverInterface
+{
+    /**
+     * Parameters
+     *
+     * @var array
+     */
+    private $parameters;
+
+    /**
+     * @var array
+     */
+    protected $config;
+
+    /**
+     *
+     * @var resource
+     */
+    protected $stream;
+
+    /**
+     * @param string $host IP/hostname of the InfluxDB host
+     * @param int    $port Port of the InfluxDB process
+     */
+    public function __construct($host, $port)
+    {
+        $this->config['host'] = $host;
+        $this->config['port'] = $port;
+    }
+
+    /**
+     * Close the stream (if created)
+     */
+    public function __destruct()
+    {
+        if (isset($this->stream) && is_resource($this->stream)) {
+            fclose($this->stream);
+        }
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    public function setParameters(array $parameters)
+    {
+        $this->parameters = $parameters;
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    public function getParameters()
+    {
+        return $this->parameters;
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    public function write($data = null)
+    {
+        if (isset($this->stream) === false) {
+            $this->createStream();
+        }
+
+        $this->doWrite($data);
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    abstract public function isSuccess();
+
+    /**
+     * Perform write to socket
+     * @param mixed|null $data
+     */
+    abstract protected function doWrite($data = null);
+
+    /**
+     * Create the resource stream
+     */
+    abstract protected function createStream();
+
+}

src/InfluxDB/Driver/TCP.php

@@ -0,0 +1,44 @@
+<?php
+/**
+ * @author Stephen "TheCodeAssassin" Hoogendijk
+ */
+
+namespace InfluxDB\Driver;
+
+/**
+ * Class UDP
+ *
+ * @package InfluxDB\Driver
+ */
+class TCP extends AbstractSocketDriver implements DriverInterface
+{
+
+    private $result;
+
+    /**
+     * {@inheritdoc}
+     */
+    public function isSuccess()
+    {
+        return (bool) $this->result;
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    protected function createStream()
+    {
+        $host = sprintf('tcp://%s:%d', $this->config['host'], $this->config['port']);
+
+        // stream the data using UDP and suppress any errors
+        $this->stream = @stream_socket_client($host);
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    protected function doWrite($data = null)
+    {
+        $this->result = false !== fwrite($this->stream, $data);
+    }
+}

src/InfluxDB/Driver/UDP.php

@@ -10,76 +10,8 @@
  *
  * @package InfluxDB\Driver
  */
-class UDP implements DriverInterface
+class UDP extends AbstractSocketDriver implements DriverInterface
 {
-    /**
-     * Parameters
-     *
-     * @var array
-     */
-    private $parameters;
-
-    /**
-     * @var array
-     */
-    private $config;
-
-    /**
-     *
-     * @var resource
-     */
-    private $stream;
-
-    /**
-     * @param string $host IP/hostname of the InfluxDB host
-     * @param int    $port Port of the InfluxDB process
-     */
-    public function __construct($host, $port)
-    {
-        $this->config['host'] = $host;
-        $this->config['port'] = $port;
-    }
-
-    /**
-     * Close the stream (if created)
-     */
-    public function __destruct()
-    {
-        if (isset($this->stream) && is_resource($this->stream)) {
-            fclose($this->stream);
-        }
-    }
-
-    /**
-     * {@inheritdoc}
-     */
-    public function setParameters(array $parameters)
-    {
-        $this->parameters = $parameters;
-    }
-
-    /**
-     * {@inheritdoc}
-     */
-    public function getParameters()
-    {
-        return $this->parameters;
-    }
-
-    /**
-     * {@inheritdoc}
-     */
-    public function write($data = null)
-    {
-        if (isset($this->stream) === false) {
-            $this->createStream();
-        }
-
-        @stream_socket_sendto($this->stream, $data);
-
-        return true;
-    }
-
     /**
      * {@inheritdoc}
      */
@@ -89,7 +21,7 @@ public function isSuccess()
     }
 
     /**
-     * Create the resource stream
+     * {@inheritdoc}
      */
     protected function createStream()
     {
@@ -99,4 +31,11 @@ protected function createStream()
         $this->stream = @stream_socket_client($host);
     }
 
+    /**
+     * {@inheritdoc}
+     */
+    protected function doWrite($data = null)
+    {
+        @stream_socket_sendto($this->stream, $data);
+    }
 }