From fb6e3dc066de70a4bce68e97c8af0eb90dab0795 Mon Sep 17 00:00:00 2001 From: Feng Lee Date: Thu, 5 Mar 2015 02:26:22 +0800 Subject: [PATCH] new -> init --- apps/emqtt/src/emqtt_client.erl | 32 +++++++++++++++---------------- apps/emqtt/src/emqtt_parser.erl | 6 +++--- apps/emqtt/src/emqtt_protocol.erl | 4 ++-- 3 files changed, 20 insertions(+), 22 deletions(-) diff --git a/apps/emqtt/src/emqtt_client.erl b/apps/emqtt/src/emqtt_client.erl index ba0c42f74..406ba9598 100644 --- a/apps/emqtt/src/emqtt_client.erl +++ b/apps/emqtt/src/emqtt_client.erl @@ -19,7 +19,6 @@ %% OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE %% SOFTWARE. %%------------------------------------------------------------------------------ - -module(emqtt_client). -author('feng@emqtt.io'). @@ -40,27 +39,26 @@ -include("emqtt_packet.hrl"). %%Client State... --record(state, { - transport, - socket, - peer_name, - conn_name, - await_recv, - conn_state, - conserve, - parse_state, - proto_state, - keepalive -}). +-record(state, {transport, + socket, + peer_name, + conn_name, + await_recv, + conn_state, + conserve, + parse_state, + proto_state, + keepalive}). start_link(SockArgs) -> {ok, proc_lib:spawn_link(?MODULE, init, [SockArgs])}. +%%TODO: rename? info(Pid) -> gen_server:call(Pid, info). init(SockArgs = {Transport, Sock, _SockFun}) -> - %%TODO: replace emqtt_net?? + %transform if ssl. {ok, NewSock} = esockd_connection:accept(SockArgs), {ok, Peername} = emqtt_net:peer_string(Sock), {ok, ConnStr} = emqtt_net:connection_string(Sock, inbound), @@ -72,8 +70,8 @@ init(SockArgs = {Transport, Sock, _SockFun}) -> await_recv = false, conn_state = running, conserve = false, - parse_state = emqtt_parser:new(), - proto_state = emqtt_protocol:initial_state(Transport, NewSock, Peername)}), + parse_state = emqtt_parser:init(), + proto_state = emqtt_protocol:init(Transport, NewSock, Peername)}), gen_server:enter_loop(?MODULE, [], State, 10000). handle_call(info, _From, State = #state{ @@ -174,7 +172,7 @@ process_received_bytes(Bytes, {ok, ProtoState1} -> process_received_bytes( Rest, - State#state{ parse_state = emqtt_parser:new(), + State#state{ parse_state = emqtt_parser:init(), proto_state = ProtoState1 }); {error, Error} -> lager:error("MQTT protocol error ~p for connection ~p~n", [Error, ConnStr]), diff --git a/apps/emqtt/src/emqtt_parser.erl b/apps/emqtt/src/emqtt_parser.erl index 2e4909688..c06cce08d 100644 --- a/apps/emqtt/src/emqtt_parser.erl +++ b/apps/emqtt/src/emqtt_parser.erl @@ -31,7 +31,7 @@ -include("emqtt_packet.hrl"). %% API --export([new/0, parse/2]). +-export([init/0, parse/2]). %%%----------------------------------------------------------------------------- @@ -40,8 +40,8 @@ %% %% @end %%%----------------------------------------------------------------------------- --spec new() -> none. -new() -> none. +-spec init() -> none. +init() -> none. %%%----------------------------------------------------------------------------- %% @doc diff --git a/apps/emqtt/src/emqtt_protocol.erl b/apps/emqtt/src/emqtt_protocol.erl index 17ef7adc4..aa7c72341 100644 --- a/apps/emqtt/src/emqtt_protocol.erl +++ b/apps/emqtt/src/emqtt_protocol.erl @@ -30,7 +30,7 @@ %% API Function Exports %% ------------------------------------------------------------------ --export([initial_state/3, client_id/1]). +-export([init/3, client_id/1]). -export([handle_packet/2, send_message/2, send_packet/2, redeliver/2, shutdown/2]). @@ -73,7 +73,7 @@ -define(PUBACK_PACKET(PacketId), #mqtt_packet_puback { packet_id = PacketId }). -initial_state(Transport, Socket, Peername) -> +init(Transport, Socket, Peername) -> #proto_state{ transport = Transport, socket = Socket,