Add better error handling in echo server0.2.1

1 files changed, 10 insertions, 4 deletions

diff --git a/lib/octonetcat/servers/echo.ex b/lib/octonetcat/servers/echo.ex
index 0d7487f..e416d9d 100644
--- a/lib/octonetcat/servers/echo.ex
+++ b/lib/octonetcat/servers/echo.ex
@@ -6,17 +6,23 @@ defmodule Octonetcat.Echo do
   dies on failure or disconnect.
   """
 
+  require Logger
+
   def serve(socket) do
     socket
-    |> read_line!()
+    |> read_line()
     |> write_line!(socket)
 
     serve(socket)
   end
 
-  def read_line!(socket) do
-    {:ok, line} = :gen_tcp.recv(socket, 0)
-    line
+  def read_line(socket) do
+    case :gen_tcp.recv(socket, 0) do
+      {:ok, line} -> line
+      {:error, reason} ->
+        Logger.error(reason)
+        Process.exit(self(), :normal)
+    end
   end
 
   def write_line!(line, socket) do