Allow adding "user" information to segments

Author: legoscia

lib/aws_ex_ray/segment.ex

@@ -23,7 +23,8 @@
       metadata:   map,
       error:      map | nil,
       http:       map,
-      aws:        map | nil
+      aws:        map | nil,
+      user:       String.t | nil
     }
 
     defstruct id:         "",
@@ -36,7 +37,8 @@
               annotation: %{},
               metadata:   %{},
               http: %{},
-              aws: nil
+              aws: nil,
+              user: nil
 
     @spec new(trace :: Trace.t, name :: String.t) :: t
     def new(trace, name) do
@@ -105,6 +107,11 @@
       end)
     end
 
+    @spec set_user(seg :: t, user :: String.t) :: t
+    def set_user(seg, user) do
+      Map.put(seg, :user, user)
+    end
+
     @spec set_http_request(seg :: t, req :: HTTPRequest.t) :: t
     def set_http_request(seg, req) do
       put_in(seg.http.request, req)

lib/aws_ex_ray/segment/formatter.ex

@@ -29,6 +29,7 @@
       |> embed_http(seg)
       |> embed_error(seg)
       |> embed_aws(seg)
+      |> embed_user(seg)
     end
 
     defp embed_error(m, seg) do
@@ -107,4 +108,13 @@
       end
     end
 
+    defp embed_user(m, seg) do
+      case seg.user do
+        nil ->
+          m
+        user ->
+          Map.put(m, :user, user)
+      end
+    end
+
   end