From 059c3fb199f22e66461acc6412f596e211d3833c Mon Sep 17 00:00:00 2001
From: Paul Rock <paul@drteam.rocks>
Date: Sun, 17 Feb 2019 14:14:52 +0300
Subject: [PATCH] more methods moved to socket trait

---
 src/Client.php      | 21 ---------------------
 src/SocketTrait.php | 24 +++++++++++++++++++++++-
 2 files changed, 23 insertions(+), 22 deletions(-)

diff --git a/src/Client.php b/src/Client.php
index b20eba9..da5a165 100644
--- a/src/Client.php
+++ b/src/Client.php
@@ -429,25 +429,4 @@ class Client implements Interfaces\ClientInterface
         return $connected;
     }
 
-    /**
-     * Save socket resource to static variable
-     *
-     * @param   resource $socket
-     * @return  Client
-     */
-    private function setSocket($socket): Client
-    {
-        $this->_socket = $socket;
-        return $this;
-    }
-
-    /**
-     * Return socket resource if is exist
-     *
-     * @return  resource
-     */
-    public function getSocket()
-    {
-        return $this->_socket;
-    }
 }
diff --git a/src/SocketTrait.php b/src/SocketTrait.php
index 1dbc366..d2c261c 100644
--- a/src/SocketTrait.php
+++ b/src/SocketTrait.php
@@ -9,6 +9,7 @@ trait SocketTrait
     /**
      * Initiate socket session
      *
+     * @return  void
      * @throws  \RouterOS\Exceptions\ClientException
      * @throws  \RouterOS\Exceptions\ConfigException
      */
@@ -42,7 +43,7 @@ trait SocketTrait
         }
 
         // Save socket to static variable
-        return $this->setSocket($socket);
+        $this->setSocket($socket);
     }
 
     /**
@@ -54,4 +55,25 @@ trait SocketTrait
     {
         return fclose($this->_socket);
     }
+
+    /**
+     * Save socket resource to static variable
+     *
+     * @param   resource $socket
+     * @return  void
+     */
+    private function setSocket($socket)
+    {
+        $this->_socket = $socket;
+    }
+
+    /**
+     * Return socket resource if is exist
+     *
+     * @return  resource
+     */
+    public function getSocket()
+    {
+        return $this->_socket;
+    }
 }