86 lines
2.2 KiB
Erlang
86 lines
2.2 KiB
Erlang
%%--------------------------------------------------------------------
|
|
%% Copyright (c) 2019 EMQ Technologies Co., Ltd. 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.
|
|
%%--------------------------------------------------------------------
|
|
|
|
-module(test_client).
|
|
|
|
-behaviour(ecpool_worker).
|
|
|
|
-behaviour(gen_server).
|
|
|
|
-define(SERVER, ?MODULE).
|
|
|
|
-export([ connect/1
|
|
, plus/3
|
|
, callback/2
|
|
, stop/2
|
|
]).
|
|
|
|
-export([init/1,
|
|
handle_call/3,
|
|
handle_cast/2,
|
|
handle_info/2,
|
|
terminate/2,
|
|
code_change/3
|
|
]).
|
|
|
|
connect(Opts) ->
|
|
case proplists:get_value(multiprocess, Opts, false) of
|
|
true ->
|
|
{ok, Pid1} = gen_server:start_link(?MODULE, [Opts], []),
|
|
{ok, Pid2} = gen_server:start_link(?MODULE, [Opts], []),
|
|
{ok, {Pid1, Pid2}, #{supervisees => [Pid1, Pid2]}};
|
|
false ->
|
|
gen_server:start_link(?MODULE, [Opts], [])
|
|
end.
|
|
|
|
plus(Pid, L, R) ->
|
|
gen_server:call(Pid, {plus, L, R}).
|
|
|
|
callback(Result, SendTo) ->
|
|
SendTo ! {result, Result}.
|
|
|
|
stop(Pid, Reason) ->
|
|
gen_server:call(Pid, {stop, Reason}).
|
|
|
|
%%-----------------------------------------------------------------------------
|
|
%% gen_server Function Definitions
|
|
%%-----------------------------------------------------------------------------
|
|
|
|
init(Args) ->
|
|
{ok, Args}.
|
|
|
|
handle_call({stop, Reason}, _From, State) ->
|
|
{stop, Reason, ok, State};
|
|
|
|
handle_call({plus, L, R}, _From, State) ->
|
|
{reply, L + R, State};
|
|
|
|
handle_call(_Req, _From, State) ->
|
|
{reply, ok, State}.
|
|
|
|
handle_cast(_Msg, State) ->
|
|
{noreply, State}.
|
|
|
|
handle_info(_Info, State) ->
|
|
{noreply, State}.
|
|
|
|
terminate(_Reason, _State) ->
|
|
ok.
|
|
|
|
code_change(_OldVsn, State, _Extra) ->
|
|
{ok, State}.
|
|
|