使用golang在實現(xiàn)一個websocket服務(wù)端-創(chuàng)新互聯(lián)

本篇文章為大家展示了使用golang在實現(xiàn)一個websocket 服務(wù)端，內(nèi)容簡明扼要并且容易理解，絕對能使你眼前一亮，通過這篇文章的詳細介紹希望你能有所收獲。

在金州等地區(qū)，都構(gòu)建了全面的區(qū)域性戰(zhàn)略布局，加強發(fā)展的系統(tǒng)性、市場前瞻性、產(chǎn)品創(chuàng)新能力，以專注、極致的服務(wù)理念，為客戶提供網(wǎng)站設(shè)計制作、網(wǎng)站設(shè)計網(wǎng)站設(shè)計制作按需網(wǎng)站建設(shè),公司網(wǎng)站建設(shè),企業(yè)網(wǎng)站建設(shè),成都品牌網(wǎng)站建設(shè),全網(wǎng)營銷推廣,成都外貿(mào)網(wǎng)站制作,金州網(wǎng)站建設(shè)費用合理。

創(chuàng)建一個websocket的服務(wù)端

package smile

import (
  "errors"
  "log"
  "net/http"
  "sync"
  "time"

  "github.com/gorilla/websocket"
)

const (
  // 允許等待的寫入時間
  writeWait = 10 * time.Second

  // Time allowed to read the next pong message from the peer.
  pongWait = 60 * time.Second

  // Send pings to peer with this period. Must be less than pongWait.
  pingPeriod = (pongWait * 9) / 10

  // Maximum message size allowed from peer.
  maxMessageSize = 512
)

// 大的連接ID，每次連接都加1 處理
var maxConnId int64

// 客戶端讀寫消息
type wsMessage struct {
  // websocket.TextMessage 消息類型
  messageType int
  data    []byte
}

// ws 的所有連接
// 用于廣播
var wsConnAll map[int64]*wsConnection

var upgrader = websocket.Upgrader{
  ReadBufferSize: 1024,
  WriteBufferSize: 1024,
  // 允許所有的CORS 跨域請求，正式環(huán)境可以關(guān)閉
  CheckOrigin: func(r *http.Request) bool {
    return true
  },
}

// 客戶端連接
type wsConnection struct {
  wsSocket *websocket.Conn // 底層websocket
  inChan  chan *wsMessage // 讀隊列
  outChan chan *wsMessage // 寫隊列

  mutex   sync.Mutex // 避免重復(fù)關(guān)閉管道,加鎖處理
  isClosed bool
  closeChan chan byte // 關(guān)閉通知
  id    int64
}

func wsHandler(resp http.ResponseWriter, req *http.Request) {
  // 應(yīng)答客戶端告知升級連接為websocket
  wsSocket, err := upgrader.Upgrade(resp, req, nil)
  if err != nil {
    log.Println("升級為websocket失敗", err.Error())
    return
  }
  maxConnId++
  // TODO 如果要控制連接數(shù)可以計算，wsConnAll長度
  // 連接數(shù)保持一定數(shù)量，超過的部分不提供服務(wù)
  wsConn := &wsConnection{
    wsSocket: wsSocket,
    inChan:  make(chan *wsMessage, 1000),
    outChan:  make(chan *wsMessage, 1000),
    closeChan: make(chan byte),
    isClosed: false,
    id:    maxConnId,
  }
  wsConnAll[maxConnId] = wsConn
  log.Println("當(dāng)前在線人數(shù)", len(wsConnAll))

  // 處理器,發(fā)送定時信息，避免意外關(guān)閉
  go wsConn.processLoop()
  // 讀協(xié)程
  go wsConn.wsReadLoop()
  // 寫協(xié)程
  go wsConn.wsWriteLoop()
}

// 處理隊列中的消息
func (wsConn *wsConnection) processLoop() {
  // 處理消息隊列中的消息
  // 獲取到消息隊列中的消息，處理完成后，發(fā)送消息給客戶端
  for {
    msg, err := wsConn.wsRead()
    if err != nil {
      log.Println("獲取消息出現(xiàn)錯誤", err.Error())
      break
    }
    log.Println("接收到消息", string(msg.data))
    // 修改以下內(nèi)容把客戶端傳遞的消息傳遞給處理程序
    err = wsConn.wsWrite(msg.messageType, msg.data)
    if err != nil {
      log.Println("發(fā)送消息給客戶端出現(xiàn)錯誤", err.Error())
      break
    }
  }
}

// 處理消息隊列中的消息
func (wsConn *wsConnection) wsReadLoop() {
  // 設(shè)置消息的大長度
  wsConn.wsSocket.SetReadLimit(maxMessageSize)
  wsConn.wsSocket.SetReadDeadline(time.Now().Add(pongWait))
  for {
    // 讀一個message
    msgType, data, err := wsConn.wsSocket.ReadMessage()
    if err != nil {
      websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway, websocket.CloseAbnormalClosure)
      log.Println("消息讀取出現(xiàn)錯誤", err.Error())
      wsConn.close()
      return
    }
    req := &wsMessage{
      msgType,
      data,
    }
    // 放入請求隊列,消息入棧
    select {
    case wsConn.inChan <- req:
    case <-wsConn.closeChan:
      return
    }
  }
}

// 發(fā)送消息給客戶端
func (wsConn *wsConnection) wsWriteLoop() {
  ticker := time.NewTicker(pingPeriod)
  defer func() {
    ticker.Stop()
  }()
  for {
    select {
    // 取一個應(yīng)答
    case msg := <-wsConn.outChan:
      // 寫給websocket
      if err := wsConn.wsSocket.WriteMessage(msg.messageType, msg.data); err != nil {
        log.Println("發(fā)送消息給客戶端發(fā)生錯誤", err.Error())
        // 切斷服務(wù)
        wsConn.close()
        return
      }
    case <-wsConn.closeChan:
      // 獲取到關(guān)閉通知
      return
    case <-ticker.C:
      // 出現(xiàn)超時情況
      wsConn.wsSocket.SetWriteDeadline(time.Now().Add(writeWait))
      if err := wsConn.wsSocket.WriteMessage(websocket.PingMessage, nil); err != nil {
        return
      }
    }
  }
}

// 寫入消息到隊列中
func (wsConn *wsConnection) wsWrite(messageType int, data []byte) error {
  select {
  case wsConn.outChan <- &wsMessage{messageType, data}:
  case <-wsConn.closeChan:
    return errors.New("連接已經(jīng)關(guān)閉")
  }
  return nil
}

// 讀取消息隊列中的消息
func (wsConn *wsConnection) wsRead() (*wsMessage, error) {
  select {
  case msg := <-wsConn.inChan:
    // 獲取到消息隊列中的消息
    return msg, nil
  case <-wsConn.closeChan:

  }
  return nil, errors.New("連接已經(jīng)關(guān)閉")
}

// 關(guān)閉連接
func (wsConn *wsConnection) close() {
  log.Println("關(guān)閉連接被調(diào)用了")
  wsConn.wsSocket.Close()
  wsConn.mutex.Lock()
  defer wsConn.mutex.Unlock()
  if wsConn.isClosed == false {
    wsConn.isClosed = true
    // 刪除這個連接的變量
    delete(wsConnAll, wsConn.id)
    close(wsConn.closeChan)
  }
}

// 啟動程序
func StartWebsocket(addrPort string) {
  wsConnAll = make(map[int64]*wsConnection)
  http.HandleFunc("/ws", wsHandler)
  http.ListenAndServe(addrPort, nil)
}

上述內(nèi)容就是使用golang在實現(xiàn)一個websocket 服務(wù)端，你們學(xué)到知識或技能了嗎？如果還想學(xué)到更多技能或者豐富自己的知識儲備，歡迎關(guān)注創(chuàng)新互聯(lián)成都網(wǎng)站設(shè)計公司行業(yè)資訊頻道。

另外有需要云服務(wù)器可以了解下創(chuàng)新互聯(lián)scvps.cn，海內(nèi)外云服務(wù)器15元起步，三天無理由+7*72小時售后在線，公司持有idc許可證，提供“云服務(wù)器、裸金屬服務(wù)器、高防服務(wù)器、香港服務(wù)器、美國服務(wù)器、虛擬主機、免備案服務(wù)器”等云主機租用服務(wù)以及企業(yè)上云的綜合解決方案，具有“安全穩(wěn)定、簡單易用、服務(wù)可用性高、性價比高”等特點與優(yōu)勢，專為企業(yè)上云打造定制，能夠滿足用戶豐富、多元化的應(yīng)用場景需求。

名稱欄目：使用golang在實現(xiàn)一個websocket服務(wù)端-創(chuàng)新互聯(lián)
分享地址：http://muchs.cn/article0/dhcooo.html

成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián)，為您提供網(wǎng)站制作、ChatGPT、網(wǎng)頁設(shè)計公司、動態(tài)網(wǎng)站、虛擬主機、靜態(tài)網(wǎng)站

聲明：本網(wǎng)站發(fā)布的內(nèi)容（圖片、視頻和文字）以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主，如果涉及侵權(quán)請盡快告知，我們將會在第一時間刪除。文章觀點不代表本網(wǎng)站立場，如需處理請聯(lián)系客服。電話：028-86922220；郵箱：631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載，或轉(zhuǎn)載時需注明來源：創(chuàng)新互聯(lián)

猜你還喜歡下面的內(nèi)容