websocket
This commit is contained in:
Feng Lee
parent
153f8c2dba
commit
9343a7c419
|
|
@ -0,0 +1,59 @@
|
||||||
|
<!doctype html>
|
||||||
|
<html>
|
||||||
|
<head>
|
||||||
|
<title>MQTT Over Mochiweb websocket</title>
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<h1>MQTT Over Mochiweb websocket</h1>
|
||||||
|
|
||||||
|
<div id="connect">
|
||||||
|
<button id="btnConn">Connect</button>
|
||||||
|
State: <span id="connstate" style="font-weight:bold;"></span>
|
||||||
|
</div>
|
||||||
|
<br/><i>Protip: open your javascript error console, just in case..</i><br/>
|
||||||
|
<hr/>
|
||||||
|
<div id="connected">
|
||||||
|
<form id="sendForm">
|
||||||
|
<input id="phrase" type="text"/>
|
||||||
|
<input id="btnSend" class="button" type="submit" name="connect"
|
||||||
|
value="Send"/>
|
||||||
|
</form>
|
||||||
|
</div>
|
||||||
|
<hr/>
|
||||||
|
<div id="msgs"></div>
|
||||||
|
|
||||||
|
<script type="text/javascript">
|
||||||
|
var ws;
|
||||||
|
if (!window.WebSocket) {
|
||||||
|
alert("WebSocket not supported by this browser");
|
||||||
|
}
|
||||||
|
function $(id) {
|
||||||
|
return document.getElementById(id);
|
||||||
|
}
|
||||||
|
function go() {
|
||||||
|
ws = new WebSocket("ws://" + location.host + "/mqtt/wsocket");
|
||||||
|
ws.onopen = function () {
|
||||||
|
$('connstate').innerHTML = 'CONNECTED';
|
||||||
|
}
|
||||||
|
ws.onclose = function () {
|
||||||
|
$('connstate').innerHTML = 'CLOSED';
|
||||||
|
}
|
||||||
|
ws.onmessage = function (e) {
|
||||||
|
var p = document.createElement('pre');
|
||||||
|
p.appendChild(document.createTextNode(e.data));
|
||||||
|
$('msgs').appendChild(p);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
$('sendForm').onsubmit = function (event) {
|
||||||
|
var p = $('phrase');
|
||||||
|
ws.send(p.value);
|
||||||
|
p.value='';
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
$('btnConn').onclick = function(event) {
|
||||||
|
go(); return false;
|
||||||
|
};
|
||||||
|
</script>
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
|
|
||||||
|
|
@ -20,7 +20,7 @@
|
||||||
%%% SOFTWARE.
|
%%% SOFTWARE.
|
||||||
%%%-----------------------------------------------------------------------------
|
%%%-----------------------------------------------------------------------------
|
||||||
%%% @doc
|
%%% @doc
|
||||||
%%% emqttd http handler.
|
%%% emqttd http publish API and websocket client.
|
||||||
%%%
|
%%%
|
||||||
%%% @end
|
%%% @end
|
||||||
%%%-----------------------------------------------------------------------------
|
%%%-----------------------------------------------------------------------------
|
||||||
|
|
@ -37,35 +37,47 @@
|
||||||
-export([handle/1]).
|
-export([handle/1]).
|
||||||
|
|
||||||
handle(Req) ->
|
handle(Req) ->
|
||||||
case authorized(Req) of
|
handle(Req:get(method), Req:get(path), Req).
|
||||||
true ->
|
|
||||||
Path = Req:get(path),
|
|
||||||
Method = Req:get(method),
|
|
||||||
handle(Method, Path, Req);
|
|
||||||
false ->
|
|
||||||
Req:respond({401, [], <<"Fobbiden">>})
|
|
||||||
end.
|
|
||||||
|
|
||||||
handle('POST', "/mqtt/publish", Req) ->
|
handle('POST', "/mqtt/publish", Req) ->
|
||||||
Params = mochiweb_request:parse_post(Req),
|
Params = mochiweb_request:parse_post(Req),
|
||||||
lager:info("HTTP Publish: ~p~n", [Params]),
|
lager:info("HTTP Publish: ~p~n", [Params]),
|
||||||
Qos = int(get_value("qos", Params, "0")),
|
case authorized(Req) of
|
||||||
Retain = bool(get_value("retain", Params, "0")),
|
true ->
|
||||||
Topic = list_to_binary(get_value("topic", Params)),
|
Qos = int(get_value("qos", Params, "0")),
|
||||||
Message = list_to_binary(get_value("message", Params)),
|
Retain = bool(get_value("retain", Params, "0")),
|
||||||
case {validate(qos, Qos), validate(topic, Topic)} of
|
Topic = list_to_binary(get_value("topic", Params)),
|
||||||
{true, true} ->
|
Message = list_to_binary(get_value("message", Params)),
|
||||||
emqttd_pubsub:publish(http, #mqtt_message{qos = Qos,
|
case {validate(qos, Qos), validate(topic, Topic)} of
|
||||||
retain = Retain,
|
{true, true} ->
|
||||||
topic = Topic,
|
emqttd_pubsub:publish(http, #mqtt_message{qos = Qos,
|
||||||
payload = Message}),
|
retain = Retain,
|
||||||
Req:ok({"text/plan", <<"ok\n">>});
|
topic = Topic,
|
||||||
{false, _} ->
|
payload = Message}),
|
||||||
Req:respond({400, [], <<"Bad QoS">>});
|
Req:ok({"text/plan", <<"ok\n">>});
|
||||||
{_, false} ->
|
{false, _} ->
|
||||||
Req:respond({400, [], <<"Bad Topic">>})
|
Req:respond({400, [], <<"Bad QoS">>});
|
||||||
|
{_, false} ->
|
||||||
|
Req:respond({400, [], <<"Bad Topic">>})
|
||||||
|
end;
|
||||||
|
false ->
|
||||||
|
Req:respond({401, [], <<"Fobbiden">>})
|
||||||
|
end;
|
||||||
|
|
||||||
|
|
||||||
|
handle(_Method, "/mqtt/wsocket", Req) ->
|
||||||
|
lager:info("Websocket Headers: ~p~n", [Req:get(headers)]),
|
||||||
|
Up = Req:get_header_value("Upgrade"),
|
||||||
|
case Up =/= undefined andalso string:to_lower(Up) =:= "websocket" of
|
||||||
|
true ->
|
||||||
|
emqttd_websocket:init(Req);
|
||||||
|
false ->
|
||||||
|
Req:respond({400, [], <<"Bad Request">>})
|
||||||
end;
|
end;
|
||||||
|
|
||||||
|
handle('GET', "/mqtt/" ++ File, Req) ->
|
||||||
|
mochiweb_request:serve_file(File, docroot(), Req);
|
||||||
|
|
||||||
handle(_Method, _Path, Req) ->
|
handle(_Method, _Path, Req) ->
|
||||||
Req:not_found().
|
Req:not_found().
|
||||||
|
|
||||||
|
|
@ -101,4 +113,8 @@ int(S) -> list_to_integer(S).
|
||||||
bool("0") -> false;
|
bool("0") -> false;
|
||||||
bool("1") -> true.
|
bool("1") -> true.
|
||||||
|
|
||||||
|
docroot() ->
|
||||||
|
{file, Here} = code:is_loaded(?MODULE),
|
||||||
|
Dir = filename:dirname(filename:dirname(Here)),
|
||||||
|
filename:join([Dir, "priv", "www"]).
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,42 @@
|
||||||
|
%%%-----------------------------------------------------------------------------
|
||||||
|
%%% @Copyright (C) 2012-2015, Feng Lee <feng@emqtt.io>
|
||||||
|
%%%
|
||||||
|
%%% Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
%%% of this software and associated documentation files (the "Software"), to deal
|
||||||
|
%%% in the Software without restriction, including without limitation the rights
|
||||||
|
%%% to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
%%% copies of the Software, and to permit persons to whom the Software is
|
||||||
|
%%% furnished to do so, subject to the following conditions:
|
||||||
|
%%%
|
||||||
|
%%% The above copyright notice and this permission notice shall be included in all
|
||||||
|
%%% copies or substantial portions of the Software.
|
||||||
|
%%%
|
||||||
|
%%% THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
%%% IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
%%% FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
%%% AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
%%% LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
%%% OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||||
|
%%% SOFTWARE.
|
||||||
|
%%%-----------------------------------------------------------------------------
|
||||||
|
%%% @doc
|
||||||
|
%%% emqttd websocket client.
|
||||||
|
%%%
|
||||||
|
%%% @end
|
||||||
|
%%%-----------------------------------------------------------------------------
|
||||||
|
-module(emqttd_websocket).
|
||||||
|
|
||||||
|
-export([init/1, loop/3]).
|
||||||
|
|
||||||
|
-record(state, {}).
|
||||||
|
|
||||||
|
init(Req) ->
|
||||||
|
{ReentryWs, _ReplyChannel} = mochiweb_websocket:upgrade_connection(Req, fun ?MODULE:loop/3),
|
||||||
|
ReentryWs(#state{}).
|
||||||
|
|
||||||
|
loop(Payload, State, ReplyChannel) ->
|
||||||
|
io:format("Received data: ~p~n", [Payload]),
|
||||||
|
ReplyChannel(Payload),
|
||||||
|
State.
|
||||||
|
|
||||||
|
|
||||||
Loading…
Reference in New Issue