有兩個可用庫,下面是兩個簡單的例子
web socket 似乎只是相當於在普通的http 協議中加入了一個升級協議的部分(協議升級後依然使用第一次通信的端口和 tcp 連接似乎)
gorilla/websocket
中顯式的體現了這一步x/net/websocket
中則把這一步封裝成 websocket.Handler(handler)
github.com/gorilla/websocket
package main
import (
"fmt"
"net/http"
"log"
"github.com/gorilla/websocket"
)
func IndexHandler(w http.ResponseWriter, r *http.Request) {
fmt.Fprintln(w, "hello world")
}
var upgrader = websocket.Upgrader{} // use default options
func webSocketHandler(w http.ResponseWriter, r *http.Request) {
c, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Print("upgrade:", err)
return
}
defer c.Close()
for {
mt, message, err := c.ReadMessage()
if err != nil {
log.Println("read:", err)
break
}
log.Printf("recv: %s", message)
err = c.WriteMessage(mt, message)
if err != nil {
log.Println("write:", err)
break
}
}
}
func main() {
http.HandleFunc("/", IndexHandler)
http.HandleFunc("/ws", webSocketHandler)
http.ListenAndServe(":8000", nil)
}
golang.org/x/net/websocket
package main
import (
"fmt"
"net/http"
"log"
"golang.org/x/net/websocket"
)
func main() {
http.Handle("/ws",websocket.Handler(Echo))
if err:=http.ListenAndServe(":8000",nil);err!=nil{
log.Fatal(err)
}
}
func Echo(w *websocket.Conn) {
for {
var msg string
if err:= websocket.Message.Receive(w,&msg);err!=nil{
fmt.Println("read ",err)
break
}
fmt.Println("reciv: "+msg)
if err := websocket.Message.Send(w, msg); err != nil {
fmt.Println("write:",err)
break
}
}
}
前端
<html>
<head>
<title>好好學習</title>
</head>
<body>
<script type="text/javascript">
var sock = null;
// var wsuri = "wss://127.0.0.1:8080"; //本地的地址 是可以改變的哦
var wsuri = "ws://localhost:8080/shiming"; //本地的地址 是可以改變的哦
window.onload = function() {
//可以看到客戶端JS,很容易的就通過WebSocket函數建立了一個與服務器的連接sock,當握手成功後,會觸發WebScoket對象的onopen事件,告訴客戶端連接已經成功建立。客戶端一共綁定了四個事件。
console.log("開始了 onload");
sock = new WebSocket(wsuri);
//建立連接後觸發
sock.onopen = function() {
console.log(" 建立連接後觸發 connected to " + wsuri);
}
// 關閉連接時候觸發
sock.onclose = function(e) {
console.log("關閉連接時候觸發 connection closed (" + e.code + ")");
}
// 收到消息後觸發
sock.onmessage = function(e) {
console.log("收到消息後觸發 message received: " + e.data);
}
//發生錯誤的時候觸發
sock.onerror=function (e) {
console.log("發生錯誤時候觸發"+wsuri)
}
};
//如果sock被關閉掉了 這裏 也會報錯的啊
function send() {
var msg = document.getElementById('message').value;
sock.send(msg);
};
</script>
<h1>GoWebSocketDemo</h1>
<form>
<p>
Message: <input id="message" type="text" value="你好啊 shiming 小哥哥 嘿嘿 ">
</p>
</form>
<button onclick="send();">給服務器發送消息</button>
</body>
</html>
參考:
https://github.com/gorilla/websocket/blob/master/examples/echo/server.go
https://www.jianshu.com/p/a5b187f7e669