Reconnect if client closes connection

atipugin · Nov 19, 2023 · 56374c5 · 56374c5
1 parent 1a24b2b
commit 56374c5
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/lib/sonic/connection.rb b/lib/sonic/connection.rb
@@ -23,20 +23,36 @@ def disconnect
     end
 
     def read
-      data = socket.gets.chomp
+      data = socket.gets&.chomp
+
+      raise ServerError, data if data.nil?
+      raise ServerError, data if data.start_with?('ENDED ')
       raise ServerError, data if data.start_with?('ERR ')
 
       data
     end
 
     def write(data)
-      socket.puts(data)
+      begin
+        socket.puts(data)
+      rescue Errno::EPIPE
+        reconnect
+        socket.puts(data)
+      end
     end
 
+
     private
 
     def socket
       @socket ||= TCPSocket.open(@host, @port)
     end
+
+    def reconnect
+      @socket.close if @socket
+      @socket = TCPSocket.open(@host, @port)
+
+      connect
+    end
   end
 end
diff --git a/sonic-ruby-0.2.3.gem b/sonic-ruby-0.2.3.gem