wangsir0624 / websocket

用Go语言写的websocket服务器

websocket

安装

go get github.com/wangsir0624/websocket

用法

基本用法

package main

import (
	"fmt"
	"websocket"
)

func main() {
	//监听IP和端口
	server := websocket.Listen("127.0.0.1", 11111)
	
	//注册创建连接回调事件
	server.On("connection", func(conn *websocket.Conn) {
		fmt.Printf("accept the connection from %s\r\n", conn.RemoteAddr())
	})
	
	//注册错误回调函数
	server.On("error", func(conn *websocket.Conn) {
		fmt.Println(conn.GetErr())
	})
	
	//注册接受消息回调函数
	server.On("message", func(conn *websocket.Conn) {
		fmt.Printf("receive message from %s: %s\r\n", conn.RemoteAddr(), conn.GetData())

		conn.Send(conn.GetData())
	})
	
	//注册关闭连接回调函数
	server.On("close", func(conn *websocket.Conn) {
		fmt.Printf("the connection from %s was closed\r\n", conn.RemoteAddr())
	})
	
	//运行服务器
	server.Run()
}

广播

对所有连接广播消息

server.On("message", func(conn *websocket.Conn) {
		fmt.Printf("receive message from %s: %s\r\n", conn.RemoteAddr(), conn.GetData())

		conn.GetServer().Broadcast(conn.GetData())
})

对其他连接广播消息

server.On("message", func(conn *websocket.Conn) {
		fmt.Printf("receive message from %s: %s\r\n", conn.RemoteAddr(), conn.GetData())

		conn.GetServer().BroadcastToOthers(conn.GetData(), conn)

		//方法二
		conn.GetServer().BroadcastOnly(conn.GetData(), func(c *websocket.Conn) bool {
			return c.RemoteAddr().String() != conn.RemoteAddr().String()
		})

		//方法三
		conn.GetServer().BroadcastExcept(conn.GetData(), func(c *websocket.Conn) bool {
			return c.RemoteAddr().String() == conn.RemoteAddr().String()
		})
})

服务器状态监控

本服务器采用HTTP协议的方式来监控服务器状态，监听端口为websocket端口加1，比如上面的例子中websocket端口为11111，那么http端口即为11112，在浏览器中输入http://127.0.0.1:11112，就可以监听服务器运行状态了