elixir-in-action/chapter5/todo_server.ex

102 lines
2.6 KiB
Elixir
Raw Permalink Normal View History

2020-06-20 13:57:33 -05:00
defmodule TodoServer do
def start do
spawn(fn -> loop(TodoList.new()) end)
end
def add_entry(todo_server, new_entry) do
send(todo_server, {:add_entry, new_entry})
end
def entries(todo_server, date) do
send(todo_server, {:entries, self(), date})
receive do
{:todo_etries, entries} -> entries
after
5000 -> {:error, :timeout}
end
end
def update_entry(todo_server, entry_id, updater_fun) do
send(todo_server, {:update_entry, entry_id, updater_fun})
end
def delete_entry(todo_server, entry_id) do
send(todo_server, {:delete_entry, entry_id})
end
defp loop(todo_list) do
new_todo_list =
receive do
message -> process_message(todo_list, message)
end
loop(new_todo_list)
end
defp process_message(todo_list, {:add_entry, new_entry}) do
TodoList.add_entry(todo_list, new_entry)
end
defp process_message(todo_list, {:entries, caller, date}) do
send(caller, {:todo_etries, TodoList.entries(todo_list, date)})
todo_list
end
defp process_message(todo_list, {:update_entry, entry_id, updater_fun}) do
TodoList.update_entry(todo_list, entry_id, updater_fun)
end
defp process_message(todo_list, {:delete_entry, entry_id}) do
TodoList.delete_entry(todo_list, entry_id)
end
end
defmodule TodoList do
defstruct auto_id: 1, entries: %{}
def new(), do: %TodoList{}
def add_entry(todo_list, entry) do
entry = Map.put(entry, :id, todo_list.auto_id)
new_entries =
Map.put(
todo_list.entries,
todo_list.auto_id,
entry
)
%TodoList{todo_list | entries: new_entries, auto_id: todo_list.auto_id + 1}
end
def entries(todo_list, date) do
todo_list.entries
|> Stream.filter(fn {_, entry} -> entry.date == date end)
|> Enum.map(fn {_, entry} -> entry end)
end
def update_entry(todo_list, %{} = new_entry) do
update_entry(todo_list, new_entry.id, fn _ -> new_entry end)
end
def update_entry(todo_list, entry_id, updater_fun) do
case Map.fetch(todo_list.entries, entry_id) do
:error ->
todo_list
{:ok, old_entry} ->
old_entry_id = old_entry.id
# Make sure that the result of the updater is a map and the
# id remains unchanged.
new_entry = %{id: ^old_entry_id} = updater_fun.(old_entry)
new_entries = Map.put(todo_list.entries, new_entry.id, new_entry)
%TodoList{todo_list | entries: new_entries}
end
end
def delete_entry(todo_list, entry_id) do
%TodoList{todo_list | entries: Map.delete(todo_list.entries, entry_id)}
end
end