mirror of
https://github.com/valitydev/riak_test.git
synced 2024-11-06 16:45:29 +00:00
8f8fad10df
this test is not meant to be part of the main riak_test giddyup suites but it is useful for measuring certain properties of cluster metadata.
124 lines
4.4 KiB
Erlang
124 lines
4.4 KiB
Erlang
%% -------------------------------------------------------------------
|
|
%%
|
|
%% Copyright (c) 2013 Basho Technologies, Inc.
|
|
%%
|
|
%% This file is provided to you 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.
|
|
%%
|
|
%% -------------------------------------------------------------------
|
|
-module(cluster_meta_proxy_server).
|
|
|
|
-compile(export_all).
|
|
|
|
-behaviour(gen_server).
|
|
|
|
%% API
|
|
-export([start_link/0, history/0, is_empty/1]).
|
|
|
|
%% gen_server callbacks
|
|
-export([init/1, handle_call/3, handle_cast/2, handle_info/2,
|
|
terminate/2, code_change/3]).
|
|
|
|
-record(state, {history :: [any()],
|
|
last_msg_ts :: non_neg_integer()}).
|
|
|
|
-define(SERVER, ?MODULE).
|
|
|
|
start_link() ->
|
|
gen_server:start_link({global, ?SERVER}, ?MODULE, [], []).
|
|
|
|
history() ->
|
|
gen_server:call({global, ?SERVER}, history, infinity).
|
|
|
|
is_empty(ThresholdSecs) ->
|
|
gen_server:call({global, ?SERVER}, {is_empty, ThresholdSecs}, infinity).
|
|
|
|
init([]) ->
|
|
{ok, #state{history = []}}.
|
|
|
|
handle_call({is_empty, ThresholdSecs}, _From, State=#state{last_msg_ts=LastMsgTs}) ->
|
|
Now = moment(),
|
|
Reply = case process_info(self(), message_queue_len) of
|
|
{message_queue_len, 0} when (Now - LastMsgTs) > ThresholdSecs ->
|
|
true;
|
|
{message_queue_len, _} ->
|
|
false
|
|
end,
|
|
{reply, Reply, State};
|
|
handle_call(history, _From, State=#state{history=History}) ->
|
|
{reply, lists:reverse(History), State}.
|
|
|
|
handle_cast({From, Server, To, Msg}, State) ->
|
|
% {Keep, State1} = keep_msg(node_name(From), node_name(To), State),
|
|
State1 = add_to_history({From, To, Msg}, State),
|
|
gen_server:cast({Server, To}, Msg),
|
|
% true -> event_logger:event({dropped, node_name(From), node_name(To), Msg}) end,
|
|
{noreply, State1};
|
|
handle_cast(_Msg, State) ->
|
|
{noreply, State}.
|
|
|
|
%%--------------------------------------------------------------------
|
|
%% Function: handle_info(Info, State) -> {noreply, State} |
|
|
%% {noreply, State, Timeout} |
|
|
%% {stop, Reason, State}
|
|
%% Description: Handling all non call/cast messages
|
|
%%--------------------------------------------------------------------
|
|
handle_info(_Info, State) ->
|
|
{noreply, State}.
|
|
|
|
%%--------------------------------------------------------------------
|
|
%% Function: terminate(Reason, State) -> void()
|
|
%% Description: This function is called by a gen_server when it is about to
|
|
%% terminate. It should be the opposite of Module:init/1 and do any necessary
|
|
%% cleaning up. When it returns, the gen_server terminates with Reason.
|
|
%% The return value is ignored.
|
|
%%--------------------------------------------------------------------
|
|
terminate(_Reason, _State) ->
|
|
ok.
|
|
|
|
%%--------------------------------------------------------------------
|
|
%% Func: code_change(OldVsn, State, Extra) -> {ok, NewState}
|
|
%% Description: Convert process state when code is changed
|
|
%%--------------------------------------------------------------------
|
|
code_change(_OldVsn, State, _Extra) ->
|
|
{ok, State}.
|
|
|
|
%%--------------------------------------------------------------------
|
|
%%% Internal functions
|
|
%%--------------------------------------------------------------------
|
|
|
|
add_to_history(Entry, State=#state{history=History}) ->
|
|
State#state{history=[Entry | History], last_msg_ts = moment()}.
|
|
|
|
moment() ->
|
|
calendar:datetime_to_gregorian_seconds(calendar:universal_time()).
|
|
|
|
|
|
%% keep_msg(From, To, State) ->
|
|
%% Key = {From, To},
|
|
%% Cfg = State#state.config,
|
|
%% {Keep, Cfg1} =
|
|
%% case proplists:get_value(Key, Cfg, []) of
|
|
%% [] -> {true, Cfg};
|
|
%% [{keep, N}|Xs] -> {true, store(Key, [{keep, N - 1} || N > 1] ++ Xs, Cfg)};
|
|
%% [{drop, N}|Xs] -> {false, store(Key, [{drop, N - 1} || N > 1] ++ Xs, Cfg)}
|
|
%% end,
|
|
%% {Keep, State#state{ config = Cfg1 }}.
|
|
|
|
%% store(Key, [], Cfg) -> lists:keydelete(Key, 1, Cfg);
|
|
%% store(Key, Xs, Cfg) -> lists:keystore(Key, 1, Cfg, {Key, Xs}).
|
|
|
|
%% node_name(Node) ->
|
|
%% list_to_atom(hd(string:tokens(atom_to_list(Node),"@"))).
|