Merge f37c6b4 into 34796e6

lib/maxwell/conn.ex

@@ -47,7 +47,8 @@ defmodule Maxwell.Conn do
     req_body: iodata | Map.t,
     status: non_neg_integer | nil,
     resp_headers: %{binary => binary},
-    resp_body: iodata | Map.t
+    resp_body: iodata | Map.t,
+    private: Map.t
   }
 
   defstruct state: :unsent,
@@ -60,7 +61,8 @@ defmodule Maxwell.Conn do
     opts: [],
     status: nil,
     resp_headers: %{},
-    resp_body: ""
+    resp_body: "",
+    private: %{}
 
   alias Maxwell.{Conn, Query}
 
@@ -414,6 +416,36 @@ defmodule Maxwell.Conn do
   def get_resp_body(%Conn{resp_body: body}, keys) when is_list(keys),        do: get_in(body, keys)
   def get_resp_body(%Conn{resp_body: body}, key), do: body[key]
 
+
+  @doc """
+  Set a private value. If it already exists, it is updated.
+
+  ### Examples
+
+      iex> %Maxwell.Conn{private: %{}}
+      |> put_private(:user_id, "zhongwencool")
+      %Maxwell.Conn{private: %{user_id: "zhongwencool"}}
+  """
+  @spec put_private(Conn.t, Atom.t, term()) :: Conn.t
+  def put_private(%Conn{private: private} = conn, key, value) do
+    new_private = Map.put(private, key, value)
+    %{conn | private: new_private}
+  end
+
+  @doc """
+  Get a private value
+
+  ### Examples
+
+      iex> %Maxwell.Conn{private: %{user_id: "zhongwencool"}}
+      |> get_private(:user_id)
+      "zhongwencool"
+  """
+  @spec get_private(Conn.t, Atom.t) :: term()
+  def get_private(%Conn{private: private}, key) do
+    Map.get(private, key)
+  end
+
   defimpl Inspect, for: Conn do
     def inspect(conn, opts) do
       Inspect.Any.inspect(conn, opts)

test/maxwell/conn_test.exs

@@ -234,4 +234,13 @@ defmodule ConnTest do
       get_resp_body(%Conn{state: :unsent, resp_body: %{"foo" => "bar"}}, "foo")
     end
   end
+
+  test "put_private/3" do
+    assert put_private(%Conn{}, :user_id, "zhongwencool") == %Conn{private: %{user_id: "zhongwencool"}}
+  end
+
+  test "get_private/2" do
+    assert get_private(%Conn{}, :user_id) == nil
+    assert get_private(%Conn{private: %{user_id: "zhongwencool"}}, :user_id) == "zhongwencool"
+  end
 end