Handle terminate calls in pool workers

Before passing terminate off to the parent implementation, be sure to send
a message to the scheduler about leaving the pool.

1 files changed, 6 insertions, 0 deletions

diff --git a/lib/poolparty/pool/worker.ex b/lib/poolparty/pool/worker.ex
index 2131f8e..6b79fc3 100644
--- a/lib/poolparty/pool/worker.ex
+++ b/lib/poolparty/pool/worker.ex
@@ -24,4 +24,10 @@ defmodule PoolParty.Pool.Worker do
     {:noreply, state}
   end
 
+  def terminate(reason, state) do
+    Logger.debug("[#{__MODULE__}]: Pool Worker terminating")
+    PoolParty.Scheduler.leave(self())
+    super(reason, state)
+  end
+
 end