add webtekcos behaviour

Author: HpyHacking

examples/server.erl

@@ -9,6 +9,8 @@ main(_) ->
   application:start(sasl),
   application:start(webtekcos),
 
+  webtekcos_event_logger:add_handler(),
+
   io:format("Open client.html test websocket server.~n"),
   io:format("It's listen on localhost:3008.~n"),
   io:format("Press Ctrl+C to shutdown server!!!~n"),

src/webtekcos.erl

@@ -1,6 +1,7 @@
 -module(webtekcos).
 -export([start_link/0, start_link/2, start_link/3, stop/1, loop/1]).
 -export([send_data/1, send_data/2, close/0]).
+-export([behaviour_info/1]).
 
 start_link() ->
   start_link(?MODULE, "127.0.0.1", 3008).
@@ -15,6 +16,15 @@ start_link(Name, Host, Port) when is_list(Host), is_integer(Port) ->
 stop(Name) ->
   mochiweb_socket_server:stop(Name).
 
+behaviour_info(callbacks) ->
+  [{connect, 0},
+   {disconnect, 1},
+   {handle_message, 2},
+   {handle_data, 2}];
+
+behaviour_info(_Other) ->
+  undefined.
+
 %%%
 %%% websocket protocol
 %%%

src/webtekcos_sample_handler.erl

@@ -1,4 +1,5 @@
 -module(webtekcos_sample_handler).
+-behaviour(webtekcos).
 -export([connect/0, disconnect/1, handle_data/2, handle_message/2]).
 
 connect() -> 
@@ -22,5 +23,4 @@ handle_data(Data, {counter, Counter}) ->
 
 handle_message({timeout, _, ?MODULE}, LoopData) ->
   webtekcos:send_data(list_to_binary("Hello Timer")),
-  io:format("<<<<< ~p~n", ["Hello Timer"]),
   LoopData.