websocket/examples/chat/client.go at main · gorilla/websocket

link管理

链接快照平台

输入网页链接，自动生成快照
标签化管理网页链接

相关文章推荐

飘逸的荔枝 · WebSockets support in ...· 1 月前 ·

乐观的香菇 · 酒店爱沙尼亚: 通过 ...· 3 月前 ·

文质彬彬的香槟 · 基于无扰动增量容量和差分电压分析方法的锂离子 ...· 3 月前 ·

爱看球的路灯 · 【傲娇零字幕组】传颂之物 ...· 5 月前 ·

近视的鸡蛋 · 首页 - ...· 5 月前 ·

成熟的水龙头 · 我用Qt和Python开发了一个视频合并工具 ...· 5 月前 ·

package main

import (

"bytes"

"log"

"net/http"

"time"

"github.com/gorilla/websocket"

)

const (

// Time allowed to write a message to the peer.

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

)

var (

newline = [] byte { '\n' }

space = [] byte { ' ' }

)

var upgrader = websocket. Upgrader {

ReadBufferSize : 1024 ,

WriteBufferSize : 1024 ,

}

// Client is a middleman between the websocket connection and the hub.

type Client struct {

hub * Hub

// The websocket connection.

conn * websocket. Conn

// Buffered channel of outbound messages.

send chan [] byte

}

// readPump pumps messages from the websocket connection to the hub.

// The application runs readPump in a per-connection goroutine. The application

// ensures that there is at most one reader on a connection by executing all

// reads from this goroutine.

func ( c * Client ) readPump () {

defer func () {

c . hub . unregister <- c

c . conn . Close ()

}()

c . conn . SetReadLimit ( maxMessageSize )

c . conn . SetReadDeadline ( time . Now (). Add ( pongWait ))

c . conn . SetPongHandler ( func ( string ) error { c . conn . SetReadDeadline ( time . Now (). Add ( pongWait )); return nil })

for {

_ , message , err := c . conn . ReadMessage ()

if err != nil {

if websocket . IsUnexpectedCloseError ( err , websocket . CloseGoingAway , websocket . CloseAbnormalClosure ) {

log . Printf ( "error: %v" , err )

}

break

}

message = bytes . TrimSpace ( bytes . Replace ( message , newline , space , - 1 ))

c . hub . broadcast <- message

}

// writePump pumps messages from the hub to the websocket connection.

// A goroutine running writePump is started for each connection. The

// application ensures that there is at most one writer to a connection by

// executing all writes from this goroutine.

func ( c * Client ) writePump () {

ticker := time . NewTicker ( pingPeriod )

defer func () {

ticker . Stop ()

c . conn . Close ()

}()

for {

select {

case message , ok := <- c . send :

c . conn . SetWriteDeadline ( time . Now (). Add ( writeWait ))

if ! ok {

// The hub closed the channel.

c . conn . WriteMessage ( websocket . CloseMessage , [] byte {})

return

}

w , err := c . conn . NextWriter ( websocket . TextMessage )

if err != nil {

return

}

w . Write ( message )

// Add queued chat messages to the current websocket message.

n := len ( c . send )

for i := 0 ; i < n ; i ++ {

w . Write ( newline )

w . Write ( <- c . send )

}

if err := w . Close (); err != nil {

return

}

case <- ticker . C :

c . conn . SetWriteDeadline ( time . Now (). Add ( writeWait ))

if err := c . conn . WriteMessage ( websocket . PingMessage , nil ); err != nil {

return

}

// serveWs handles websocket requests from the peer.

func serveWs ( hub * Hub , w http. ResponseWriter , r * http. Request ) {

conn , err := upgrader . Upgrade ( w , r , nil )

if err != nil {

log . Println ( err )

return

}

client := & Client { hub : hub , conn : conn , send : make ( chan [] byte , 256 )}

client . hub . register <- client

// Allow collection of memory referenced by the caller by doing all work in

// new goroutines.

go client . writePump ()

go client . readPump ()

}