mirror of
https://github.com/balzack/databag.git
synced 2025-02-12 03:29:16 +00:00
implementing websocket relay connection
This commit is contained in:
parent
c6be994b58
commit
7affec0006
@ -1,10 +1,60 @@
|
|||||||
package databag
|
package databag
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"github.com/gorilla/websocket"
|
||||||
|
"encoding/json"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
"errors"
|
||||||
)
|
)
|
||||||
|
|
||||||
//Signal handler for websocket connection with call events
|
var bridgeRelay BridgeRelay;
|
||||||
|
var relayUpgrader = websocket.Upgrader{}
|
||||||
|
|
||||||
|
//Status handler for websocket connection
|
||||||
func Signal(w http.ResponseWriter, r *http.Request) {
|
func Signal(w http.ResponseWriter, r *http.Request) {
|
||||||
|
|
||||||
|
// accept websocket connection
|
||||||
|
conn, err := relayUpgrader.Upgrade(w, r, nil)
|
||||||
|
if err != nil {
|
||||||
|
ErrMsg(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer conn.Close()
|
||||||
|
conn.SetReadLimit(APPBodyLimit)
|
||||||
|
|
||||||
|
// receive announce
|
||||||
|
t, m, res := conn.ReadMessage()
|
||||||
|
if res != nil {
|
||||||
|
ErrMsg(res)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if t != websocket.TextMessage {
|
||||||
|
ErrMsg(errors.New("invalid websocket message type"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
var a Announce
|
||||||
|
if err := json.Unmarshal(m, &a); err != nil {
|
||||||
|
ErrMsg(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// bind connection to bridge
|
||||||
|
bridgeRelay.SetConnection(conn, a.AppToken);
|
||||||
|
|
||||||
|
for true {
|
||||||
|
t, m, res := conn.ReadMessage()
|
||||||
|
if res != nil {
|
||||||
|
ErrMsg(res)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
if t != websocket.TextMessage {
|
||||||
|
ErrMsg(errors.New("invalid websocket message type"))
|
||||||
|
break
|
||||||
|
}
|
||||||
|
bridgeRelay.RelayMessage(conn, m);
|
||||||
|
}
|
||||||
|
|
||||||
|
// release connection from bridge
|
||||||
|
bridgeRelay.ClearConnection(conn);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user