瀏覽代碼

Default arguments when no register_gen_reset was called

Tim Buchwaldt 7 年之前
父節點
當前提交
67e4768add
共有 1 個文件被更改,包括 5 次插入1 次删除
  1. 5 1
      lib/gen_reset.ex

+ 5 - 1
lib/gen_reset.ex

@@ -8,7 +8,11 @@ defmodule GenReset do
   defmacro __using__(_args) do
     quote do
       def handle_call(:"$gen_reset", _from, state) do
-        case init(:erlang.get(:"$gen_reset_initial_args")) do
+        args = case :erlang.get(:"$gen_reset_initial_args") do
+          :undefined -> nil
+          rest -> rest
+        end
+        case init(args) do
           {:ok, state} ->
             {:reply, :ok, state}