Merge pull request #21 from erszcz/add-to-map

Add tr:to_map/1 to convert trace records to maps

Author: chrzaszcz

src/tr.erl

@@ -40,7 +40,8 @@
          app_modules/1,
          mfarity/1,
          mfargs/2,
-         ts/1]).
+         ts/1,
+         to_map/1]).
 
 %% gen_server callbacks
 -export([init/1,
@@ -696,6 +697,15 @@ mfargs({M, F, Arity}, Args) when length(Args) =:= Arity -> {M, F, Args}.
 -spec ts(tr()) -> string().
 ts(#tr{ts = TS}) -> calendar:system_time_to_rfc3339(TS, [{unit, microsecond}]).
 
+%% @doc Convert trace records to maps.
+-spec to_map(tr()) -> map().
+to_map(#tr{} = Tr) ->
+    IndexedFields = enumerate([record | record_info(fields, tr)]),
+    maps:from_list(lists:map(fun ({Index, Field}) -> {Field, element(Index, Tr)} end, IndexedFields)).
+
+enumerate(List) ->
+    lists:zip(lists:seq(1, length(List)), List).
+
 %% gen_server callbacks
 
 %% @private

test/tr_SUITE.erl

@@ -66,7 +66,8 @@ groups() ->
              match,
              do,
              next,
-             prev]},
+             prev,
+             to_map]},
      {call_stat, [simple_total,
                   tree_total,
                   simple_total_with_messages,
@@ -424,6 +425,25 @@ prev(_Config) ->
     ?assertEqual(T1, tr:prev(4, #{pred => Pred})),
     ?assertError(not_found, tr:prev(T1)).
 
+to_map(_Config) ->
+    Pid = erlang:list_to_pid("<0.115.0>"),
+    Trace = #tr{index = 5,
+                pid = Pid,
+                event = call,
+                mfa = {lists,keyfind,3},
+                data = [syntax_colors,1,[]],
+                ts = 1766078114348924,
+                info = no_info},
+    ?assertEqual(#{record => tr,
+                   index => 5,
+                   pid => Pid,
+                   event => call,
+                   mfa => {lists,keyfind,3},
+                   data => [syntax_colors,1,[]],
+                   ts => 1766078114348924,
+                   info => no_info},
+                 tr:to_map(Trace)).
+
 single_tb(_Config) ->
     tr:trace([{?MODULE, fib, 1}]),
     ?MODULE:fib(4),