1.websocket 簡介
以往瀏覽器要獲取服務(wù)端數(shù)據(jù)娜庇，都是通過發(fā)送 HTTP 請求蛮粮，然后等待服務(wù)端回應(yīng)的嚎莉。也就是說瀏覽器端一直是整個請求的發(fā)起者腿椎，只有它主動桌硫，才能獲取到數(shù)據(jù)夭咬。而要讓瀏覽器一側(cè)能夠獲取到服務(wù)端的實時數(shù)據(jù)啃炸，就需要不停地向服務(wù)端發(fā)起請求。雖然大多數(shù)情況下并沒有獲取到實際數(shù)據(jù)卓舵，但這大大增加了網(wǎng)絡(luò)壓力南用，對于服務(wù)端來說壓力也直線上升。
image.png
由于 WebSocket 是全雙工通信初婆，所以當(dāng)建立了 WebSocket 連接之后虫碉，接下來的通信就類似于傳統(tǒng)的 TCP 通信了绪钥±鹤疲客戶端和服務(wù)端可以相互發(fā)送數(shù)據(jù)充择，不再有實時性的問題羔砾。
image.png
image.png
2.開發(fā)包的選擇
在 Go 官方的 SDK 中表窘，并不包含對 WebSocket 的支持票腰，所以必須使用第三方庫匣屡。
要使用 Golang 開發(fā) WebSocket封救，選擇基本就在 x/net/websocket 和 gorilla/websocket 之間〉纷鳎《Go Web 編程》一書中的例子使用了x/net/websocket作為開發(fā)包誉结，而且貌似它也更加官方且正式。而實際根據(jù)我在網(wǎng)上查詢得到的反饋看來券躁，并非如此惩坑。x/net/websocket貌似 Bug 較多，且較為不穩(wěn)定也拜，問題解決也并不及時以舒。相比之下，gorilla/websocket則更加優(yōu)秀慢哈。
下載鏈接及手冊
3.推送服務(wù)實現(xiàn)
server 啟動以后會注冊兩個 Handler稀轨。
websocketHandler 用于提供瀏覽器端發(fā)送 Upgrade 請求并升級為 WebSocket 連接。 pushHandler 用于提供外部推送端發(fā)送推送數(shù)據(jù)的請求岸军。
3-1.go代碼的簡單實現(xiàn)
package main
import (
"net/http"
"log"
"github.com/gorilla/websocket"
)
var (
upgrader = websocket.Upgrader {
// 讀取存儲空間大小
ReadBufferSize:1024,
// 寫入存儲空間大小
WriteBufferSize:1024,
// 允許跨域
CheckOrigin: func(r *http.Request) bool {
return true
},
}
)
func wsHandler(w http.ResponseWriter, r *http.Request) {
var (
wbsCon *websocket.Conn
err error
data []byte
)
// 完成http應(yīng)答奋刽，在httpheader中放下如下參數(shù)
if wbsCon, err = upgrader.Upgrade(w, r, nil);err != nil {
return // 獲取連接失敗直接返回
}
for {
// 只能發(fā)送Text, Binary 類型的數(shù)據(jù),下劃線意思是忽略這個變量.
if _, data, err = wbsCon.ReadMessage();err != nil {
goto ERR // 跳轉(zhuǎn)到關(guān)閉連接
}
if err = wbsCon.WriteMessage(websocket.TextMessage, data); err != nil {
goto ERR // 發(fā)送消息失敗，關(guān)閉連接
}
}
ERR:
// 關(guān)閉連接
wbsCon.Close()
}
func main() {
// 當(dāng)有請求訪問ws時艰赞，執(zhí)行此回調(diào)方法
http.HandleFunc("/ws",wsHandler)
// 監(jiān)聽127.0.0.1:7777
err := http.ListenAndServe("0.0.0.0:7777", nil)
if err != nil {
log.Fatal("ListenAndServe", err.Error())
}
}
3-2.前端代碼
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="utf-8">
<style>
*{
margin:0px;
padding:0px;
}
</style>
</head>
<body>
<div style="margin-left:400px">
<h3>群聊天室</h3>
<div style="border:1px solid;width: 600px;height: 300px;">
<div id="msgArea" style="width:100%;height: 100%;text-align:start;resize: none;font-family: 微軟雅黑;font-size: 20px;overflow-y: scroll"></div>
</div>
<div style="border:1px solid;width: 600px;height: 100px;">
<div style="width:100%;height: 100%;">
<textarea id="userMsg" style="width:100%;height: 100%;text-align:start;resize: none;font-family: 微軟雅黑;font-size: 20px;"></textarea>
</div>
</div>
<div style="border:1px solid;width: 600px;height: 25px;">
<button style="float: right;" onclick="sendMsg()">發(fā)送</button>
</div>
</div>
</body>
</html>
<script src="http://libs.baidu.com/jquery/2.1.4/jquery.min.js"></script>
<script>
var ws;
$(function(){
link();
})
function link () {
ws = new WebSocket("ws://127.0.0.1:7777/ws");//連接服務(wù)器
ws.onopen = function(event){
console.log(event);
alert('連接了');
};
ws.onmessage = function (event) {
var date = new Date();
var msg = "<p>"+date.toLocaleString()+"</p>"+"<p>"+event.data+"</p>";
$("#msgArea").append(msg);
}
ws.onclose = function(event){alert("已經(jīng)與服務(wù)器斷開連接\r\n當(dāng)前連接狀態(tài)："+this.readyState);};
ws.onerror = function(event){alert("WebSocket異常佣谐！");};
}
function sendMsg(){
var msg = $("#userMsg").val();
ws.send(msg);
}
</script>
3-3. 結(jié)果分析
運行的結(jié)果
image.png
交互的過程
1.request請求時，要求將協(xié)議升級為websocket 參數(shù)
2.response返回時方妖，如果成功則返回websocket 參數(shù)
image.png
數(shù)據(jù)傳送形式是把數(shù)據(jù)拆成多個frame進行發(fā)送和接收狭魂，這個不用我們關(guān)心。
image.png
image.png
4.封裝Websocket
image.png
image.png
image.png
image.png
1.目錄結(jié)構(gòu)
image.png
2.server.go
package main
import (
"net/http"
"log"
"github.com/gorilla/websocket"
"github.com/go-websocket/impl"
"time"
)
var (
upgrader = websocket.Upgrader {
// 讀取存儲空間大小
ReadBufferSize:1024,
// 寫入存儲空間大小
WriteBufferSize:1024,
// 允許跨域
CheckOrigin: func(r *http.Request) bool {
return true
},
}
)
func wsHandler(w http.ResponseWriter, r *http.Request) {
var (
wsConn *websocket.Conn
err error
// data []byte
conn *impl.Connection
data []byte
)
// 完成http應(yīng)答党觅，在httpheader中放下如下參數(shù)
if wsConn, err = upgrader.Upgrade(w, r, nil);err != nil {
return // 獲取連接失敗直接返回
}
if conn, err = impl.InitConnection(wsConn); err != nil {
goto ERR
}
go func() {
var (
err error
)
for {
// 每隔一秒發(fā)送一次心跳
if err = conn.WriteMessage([]byte("heartbeat")); err != nil {
return
}
time.Sleep(1 * time.Second)
}
}()
for {
if data, err = conn.ReadMessage(); err != nil {
goto ERR
}
if err = conn.WriteMessage(data); err != nil {
goto ERR
}
}
ERR:
// 關(guān)閉當(dāng)前連接
}
func main() {
// 當(dāng)有請求訪問ws時雌澄，執(zhí)行此回調(diào)方法
http.HandleFunc("/ws",wsHandler)
// 監(jiān)聽127.0.0.1:7777
err := http.ListenAndServe("0.0.0.0:7777", nil)
if err != nil {
log.Fatal("ListenAndServe", err.Error())
}
}
3.connection.go
package impl
import (
"github.com/gorilla/websocket"
"sync"
"errors"
)
type Connection struct {
// 存放websocket連接
wsConn *websocket.Conn
// 用于存放數(shù)據(jù)
inChan chan []byte
// 用于讀取數(shù)據(jù)
outChan chan []byte
closeChan chan byte
mutex sync.Mutex
// chan是否被關(guān)閉
isClosed bool
}
// 讀取Api
func (conn *Connection) ReadMessage() (data []byte, err error) {
//select是Go中的一個控制結(jié)構(gòu)，類似于用于通信的switch語句杯瞻。每個case必須是一個通信操作镐牺，要么是發(fā)送要么是接收。 select隨機執(zhí)行一個可運行的case魁莉。如果沒有case可運行睬涧，它將阻塞募胃，直到有case可運行。一個默認的子句應(yīng)該總是可運行的畦浓。
select {
case data = <- conn.inChan:
case <- conn.closeChan:
err = errors.New("connection is closed")
}
return
}
// 發(fā)送Api
func (conn *Connection) WriteMessage(data []byte) (err error) {
select {
case conn.outChan <- data:
case <- conn.closeChan:
err = errors.New("connection is closed")
}
return
}
// 關(guān)閉連接的Api
func (conn *Connection) Close() {
// 線程安全的Close痹束，可以并發(fā)多次調(diào)用也叫做可重入的Close
conn.wsConn.Close()
conn.mutex.Lock()
if !conn.isClosed {
// 關(guān)閉chan,但是chan只能關(guān)閉一次
close(conn.closeChan)
conn.isClosed = true
}
conn.mutex.Unlock()
}
// 初始化長連接
func InitConnection(wsConn *websocket.Conn) (conn *Connection, err error) {
conn = &Connection{
wsConn: wsConn,
inChan: make(chan []byte, 1000),
outChan: make(chan []byte, 1000),
closeChan: make(chan byte, 1),
}
// 啟動讀協(xié)程
go conn.readLoop()
// 啟動寫協(xié)程
go conn.writeLoop()
return
}
// 內(nèi)部實現(xiàn)
func (conn *Connection) readLoop() {
var (
data []byte
err error
)
for {
if _, data, err = conn.wsConn.ReadMessage(); err != nil {
goto ERR
}
// 容易阻塞到這里，等待inChan有空閑的位置
select {
case conn.inChan <- data:
case <- conn.closeChan: // closeChan關(guān)閉的時候執(zhí)行
goto ERR
}
}
ERR:
conn.Close()
}
func (conn *Connection) writeLoop() {
var (
data []byte
err error
)
for {
select {
case data = <- conn.outChan:
case <- conn.closeChan:
goto ERR
}
data = <- conn.outChan
if err = conn.wsConn.WriteMessage(websocket.TextMessage, data); err != nil {
goto ERR
}
}
ERR:
conn.Close()
}
5.第三方資源
https://www.imooc.com/video/17599
https://www.cnblogs.com/snowInPluto/p/8688453.html
