go-iris-websocket 簡單聊天通訊
阿新 • • 發佈：2022-03-06
基於go的websocket大多使用gorilla/websocket
iris也提供了websoket的封裝，github.com/kataras/iris/v12/websocket
不過iris官方給的示例基本上都是依賴官方的js庫實現的neffos.js
Neffos.js對websocket進行了封裝，主要是房間進入和離開等事件的繫結，
對於訊息的傳遞也使用了自己定義的格式，不同的欄位使用分號進行分割
例如：;;;;0;0;helloworld，檢視iris中原始碼看到Message的定義：
type Message struct {
wait string
// The Namespace that this message sent to/received from.
Namespace string
// The Room that this message sent to/received from.
Room string
// The Event that this message sent to/received from.
Event string
// The actual body of the incoming/outcoming data.
Body []byte
// The Err contains any message's error, if any. 注意到有個IsNative的控制引數，使用OnNativeMessage事件和IsNative欄位就可以使用自己的格式通過websoket傳遞資料：
// True when user define it for writing, only its body is written as raw native websocket message, namespace, event and all other fields are empty.
// The receiver should accept it on the `OnNativeMessage` event.
// This field is not filled on sending/receiving.
IsNative bool
msg:=websocket.Message{
Body:[]byte("helloworld"),
IsNative:true,
}
接下來使用iris/websocket實現簡單的聊天通訊，go主要程式碼如下：
ws := websocket.New(websocket.DefaultGorillaUpgrader, websocket.Events{
websocket.OnNativeMessage: func(nsConn *websocket.NSConn, msg websocket.Message) error {
log.Printf("Server got: %s from [%s]", msg.Body, nsConn.Conn.ID())
ping := string(msg.Body)
pong := strings.Replace(ping,"？", "！", len(ping))
pong = strings.Replace(pong, "麼","",len(pong))
mg := websocket.Message{
Body:[]byte(pong),
IsNative:true,
}
nsConn.Conn.Write(mg)
return nil
},
})
完整go程式碼：
package main
import (
"log"
"strings"
"github.com/kataras/iris/v12"
"github.com/kataras/iris/v12/websocket"
)
func main() {
ws := websocket.New(websocket.DefaultGorillaUpgrader, websocket.Events{
websocket.OnNativeMessage: func(nsConn *websocket.NSConn, msg websocket.Message) error {
log.Printf("Server got: %s from [%s]", msg.Body, nsConn.Conn.ID())
ping := string(msg.Body)
pong := strings.Replace(ping,"？", "！", len(ping))
pong = strings.Replace(pong, "麼","",len(pong))
mg := websocket.Message{
Body:[]byte(pong),
IsNative:true,
}
nsConn.Conn.Write(mg)
return nil
},
})
ws.OnConnect = func(c *websocket.Conn) error {
log.Printf("[%s] Connected to server!", c.ID())
return nil
}
ws.OnDisconnect = func(c *websocket.Conn) {
log.Printf("[%s] Disconnected from server", c.ID())
}
ws.OnUpgradeError = func(err error) {
log.Printf("Upgrade Error: %v", err)
}
app := iris.New()
app.HandleDir("/","./html")
app.Get("/msg", websocket.Handler(ws))
app.Run(iris.Addr(":8080"))
}
html 程式碼：
<html>
<head>
<title>Client Page</title>
</head>
<body style="padding:10px;">
<input type="text" id="messageTxt" />
<button type="button" id="sendBtn">Send</button>
<div id="messages" style="width: 375px;margin:10 0 0 0px;border-top: 1px solid black;">
</div>
<script type="text/javascript">
var HOST = "localhost:8080"
var messageTxt = document.getElementById("messageTxt");
var messages = document.getElementById("messages");
var sendBtn = document.getElementById("sendBtn")
w = new WebSocket("ws://" + HOST + "/msg");
w.onopen = function () {
console.log("Websocket connection enstablished");
};
w.onclose = function () {
appendMessage("<div><center><h3>Disconnected</h3></center></div>");
};
w.onmessage = function (message) {
appendMessage("<div> srv: " + message.data + "</div>");
};
sendBtn.onclick = function () {
myText = messageTxt.value;
messageTxt.value = "";
appendMessage("<div style='color: red'> me: " + myText + "</div>");
w.send(myText);
};
messageTxt.addEventListener("keyup", function (e) {
if (e.keyCode === 13) {
e.preventDefault();
sendBtn.click();
}
});
function appendMessage(messageDivHTML) {
messages.insertAdjacentHTML('afterbegin', messageDivHTML);
}
</script>
</body>
</html>
效果：