Add better error handling in echo server0.2.1

2 files changed, 11 insertions, 5 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
diff --git a/mix.exs b/mix.exs
index 16818d9..e96830b 100644
--- a/mix.exs
+++ b/mix.exs
@@ -6,7 +6,7 @@ defmodule Octonetcat.Mixfile do
       app: :octonetcat,
       description: "Demo Echo Server Application",
       package: package(),
-      version: "0.2.0",
+      version: "0.2.1",
       elixir: "~> 1.5",
       start_permanent: Mix.env == :prod,
       docs: docs(),