Permalink
Fetching contributors…
Cannot retrieve contributors at this time
65 lines (54 sloc) 2.05 KB
%%%
%%% Copyright 2012, Basho Technologies, Inc. All Rights Reserved.
%%%
%%% Licensed under the Apache License, Version 2.0 (the "License");
%%% you may not use this file except in compliance with the License.
%%% You may obtain a copy of the License at
%%%
%%% http://www.apache.org/licenses/LICENSE-2.0
%%%
%%% Unless required by applicable law or agreed to in writing, software
%%% distributed under the License is distributed on an "AS IS" BASIS,
%%% WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
%%% See the License for the specific language governing permissions and
%%% limitations under the License.
%%%
%%%-------------------------------------------------------------------
%%% File: folsom_sample_slide.erl
%%% @author Russell Brown <russelldb@basho.com>
%%% @doc
%%% Sliding window sample. Last Window seconds readings are recorded.
%%% @end
%%%-----------------------------------------------------------------
-module(folsom_sample_slide).
-export([
new/1,
update/2,
get_values/1,
moment/0,
trim/2,
resize/2
]).
-include("folsom.hrl").
-define(WIDTH, 16). %% Keep this a power of two
new(Size) ->
Sample = #slide{window = Size},
Pid = folsom_sample_slide_sup:start_slide_server(?MODULE, Sample#slide.reservoir, Sample#slide.window),
Sample#slide{server=Pid}.
update(#slide{reservoir = Reservoir} = Sample, Value) ->
Moment = moment(),
X = erlang:system_info(scheduler_id),
Rnd = X band (?WIDTH-1),
ets:insert(Reservoir, {{Moment, Rnd}, Value}),
Sample.
resize(Sample, NewSize) ->
folsom_sample_slide_server:resize(Sample#slide.server, NewSize),
Sample#slide{window = NewSize}.
get_values(#slide{window = Window, reservoir = Reservoir}) ->
Oldest = moment() - Window,
ets:select(Reservoir, [{{{'$1','_'},'$2'},[{'>=', '$1', Oldest}],['$2']}]).
moment() ->
folsom_utils:now_epoch().
trim(Reservoir, Window) ->
Oldest = moment() - Window,
ets:select_delete(Reservoir, [{{{'$1','_'},'_'},[{'<', '$1', Oldest}],['true']}]).