opencensus

-module(oc_stat_aggregation_sum).

-export([init/3,
         type/0,
         add_sample/4,
         export/2,
         clear_rows/2]).

-behavior(oc_stat_aggregation).

init(_Name, _Keys, Options) ->
    Options.

type() ->
    sum.

-spec add_sample(oc_stat_view:name(), oc_tags:tags(), number(), any()) -> ok.
add_sample(Name, Tags, Value, Options) ->
    case counters_sum:observe(Name, Tags, Value) of
        unknown ->
            case counters_sum:new(Name, Tags, 1, Value) of
                ok -> ok;
                false ->
                    add_sample(Name, Tags, Value, Options)
            end;
        _ ->
            ok
    end.

export(Name, _Options) ->
    Rows = maps:values(maps:map(fun(Tags, {Count, Sum}) ->
                                        Mean = case Count of
                                                   0 -> 0;
                                                   _ -> Sum / Count
                                               end,
                                        #{tags => Tags,
                                          value => #{count => Count,
                                                     sum => Sum,
                                                     mean => Mean}}
                                end,
                                counters_sum:value(Name))),
    #{type => type(),
      rows => Rows}.

clear_rows(Name, _Options) ->
    counters_sum:remove(Name),
    ok.