Add add_metadata functions for segment and subsegment

Author: legoscia

lib/aws_ex_ray/segment.ex

@@ -94,6 +94,17 @@
       Map.put(seg, :annotation, annotation)
     end
 
+    @spec add_metadata(
+      seg         :: t,
+      metadata :: map
+    ) :: t
+    def add_metadata(seg, metadata) do
+      metadata
+      |> Enum.reduce(seg, fn {key, value}, seg ->
+        Map.put(seg, :metadata, Map.put(seg.metadata, key, value))
+      end)
+    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/subsegment.ex

@@ -66,6 +66,14 @@
       put_in(seg.segment.annotation, annotation)
     end
 
+    @spec add_metadata(
+      seg         :: t,
+      metadata :: map
+    ) :: t
+    def add_metadata(seg, metadata) do
+      put_in(seg.segment, Segment.add_metadata(seg.segment, metadata))
+    end
+
     @spec set_aws(seg :: t, params :: map) :: t
     def set_aws(seg, params) do
       Map.put(seg, :aws, params)