Bug fix for lookup

aditimalladi · Oct 22, 2018 · c75bd1d · c75bd1d
1 parent 50a38e6
commit c75bd1d
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/lib/chief.ex b/lib/chief.ex
@@ -75,6 +75,13 @@ defmodule Chief do
   def handle_call({:get_succ, node}, _from, state) do
     node_list = state[:node_list]
     node_i = Enum.find_index(node_list, fn i -> i==node end)
-    {:reply, Enum.fetch!(node_list, node_i+1), state}
+    cond do
+      node_i == nil ->
+        {:reply, node, state}
+      node_i == length(node_list) - 1 ->
+        {:reply, Enum.fetch!(node_list, 0), state}
+      true ->
+        {:reply, Enum.fetch!(node_list, node_i+1), state}
+    end
   end
 end