riak-old / src / riak_connect.erl

%% 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.    

%% @doc This tiny server provides a gossip bridge between riak nodes.

-module(riak_connect).

-behaviour(gen_server).
-export([start_link/0]).
-export([init/1, handle_call/3, handle_cast/2, handle_info/2,
	 terminate/2, code_change/3]).
-export([cast/2, stop/0]).
-record(state, {me}).

-include_lib("eunit/include/eunit.hrl").

%% @private
start_link() -> gen_server:start_link({local, ?MODULE}, ?MODULE, [], []).

%% @private
init([]) -> {ok, #state{me=node()}}.

stop() -> gen_server:cast(?MODULE, stop).

%% @private
cast(RemoteNode, Msg) -> gen_server:cast({riak_connect, RemoteNode}, Msg).

%% @private
handle_cast({gossip_ring, Ring}, State) ->
    riak_ring_gossiper ! {gossip_ring, Ring},
    {noreply, State};
handle_cast({set_ring, Ring}, State) ->
    riak_ring_gossiper ! {set_ring, Ring},
    {noreply, State};
handle_cast({get_ring, RemoteNode}, State) ->
    riak_ring_gossiper ! {get_ring, RemoteNode},
    {noreply, State};
handle_cast(stop, State) -> {stop, normal, State}.

%% @private
handle_info(_Info, State) -> {noreply, State}.

%% @private
terminate(_Reason, _State) -> ok.

%% @private
code_change(_OldVsn, State, _Extra) ->  {ok, State}.

%% @private
handle_call(_, _From, State) -> {noreply, State}.

connect_test() ->
    {ok, _Pid}  = riak_connect:start_link(),
    F = fun(Pid) ->
                register(riak_ring_gossiper, self()),
                Loop = fun([], _) -> 
                               Pid ! ok;
                          (WaitingFor, Loop) ->
                               receive
                                   {gossip_ring, _} ->
                                       Loop(WaitingFor -- [gossip_ring], Loop);
                                   {set_ring, _} ->
                                       Loop(WaitingFor -- [set_ring], Loop);
                                   {get_ring, _} ->
                                       Loop(WaitingFor -- [get_ring], Loop)
                               end
                       end,
                Loop([gossip_ring, set_ring, get_ring], Loop)
        end,
    Self = self(),
    spawn(fun() -> F(Self) end),
    gen_server:cast(?MODULE, {gossip_ring, test}),
    gen_server:cast(?MODULE, {set_ring, test}),
    gen_server:cast(?MODULE, {get_ring, test}),
    R = receive
            ok -> ok
        end,
    riak_connect:stop(),
    ?assertEqual(R, ok).
Tip: Filter by directory path e.g. /media app.js to search for public/media/app.js.
Tip: Use camelCasing e.g. ProjME to search for ProjectModifiedEvent.java.
Tip: Filter by extension type e.g. /repo .js to search for all .js files in the /repo directory.
Tip: Separate your search with spaces e.g. /ssh pom.xml to search for src/ssh/pom.xml.
Tip: Use ↑ and ↓ arrow keys to navigate and return to view the file.
Tip: You can also navigate files with Ctrl+j (next) and Ctrl+k (previous) and view the file with Ctrl+o.
Tip: You can also navigate files with Alt+j (next) and Alt+k (previous) and view the file with Alt+o.