分享每一天-golang实现websocket统计实时在线人数

golang实现websocket统计实时在线人数

GoLang 1188 0 创建时间：2023-02-26 14:46:02

#### golang实现websocket统计实时在线人数
 利用golang工具包github.com/gorilla/websocket 实现websocket服务。定义一个全局变量，存储所有在线的客户端，创建一个map来保存所有在线客户端的结构体。当客户端链接成功就保存到到全局变量，将每个新连接的客户端添加到map中，并更新在线用户数。监听广播通道，并向所有客户端发送消息的函数。上代码：
 ```go
package main

import (
	"fmt"
	"github.com/gorilla/websocket"
	"log"
	"net"
	"net/http"
	"strings"
	"time"
)

// 定义一个全局变量，存储所有在线的客户端
//var clients = make(map[*websocket.Conn]bool)

// 定义一个结构体来存储每个客户端的连接和其他信息
type Client struct {
	Conn *websocket.Conn // 连接对象
	Name string          // 客户端名称
}

// 创建一个map来保存所有在线客户端的结构体
var clients = make(map[string]*Client)

// 定义一个全局变量，用于广播在线人数
var broadcast = make(chan int)

// 定义一个升级器，将HTTP连接升级为websocket连接
var upgrader = websocket.Upgrader{
	CheckOrigin: func(r *http.Request) bool {
		return true // 允许跨域请求
	},
}

func main() {
	// 创建一个静态文件服务器，用于展示网页
	fs := http.FileServer(http.Dir("public"))
	http.Handle("/", fs)

// 注册一个处理器，用于处理websocket连接请求
	http.HandleFunc("/ws", handleConnections)

// 启动一个goroutine，用于监听广播通道，并向所有客户端发送消息
	go handleMessages()

// 启动web服务器，监听8080端口
	log.Println("http server started on :8089")
	err := http.ListenAndServe(":8089", nil)
	if err != nil {
		log.Fatal("ListenAndServe: ", err)
	}
}

// 处理websocket连接请求的函数
func handleConnections(w http.ResponseWriter, r *http.Request) {
	var name string
	xForwardedFor := r.Header.Get("X-Forwarded-For")
	ip := strings.TrimSpace(strings.Split(xForwardedFor, ",")[0])
	if ip != "" {
		name = ip
	} else {
		ip = strings.TrimSpace(r.Header.Get("X-Real-Ip"))
		if ip != "" {
			name = ip
		} else {
			if ip, _, err := net.SplitHostPort(strings.TrimSpace(r.RemoteAddr)); err == nil {
				name = ip
			}
		}
	}
	fmt.Println(name, time.Now().Format("2006-01-02 15:04:05"))
	// 将HTTP连接升级为websocket连接
	ws, err := upgrader.Upgrade(w, r, nil)
	if err != nil {
		log.Fatal(err)
	}
	defer ws.Close() // 延迟关闭连接

// 将当前客户端添加到全局变量中，并设置为true表示在线
	//clients[ws] = true
	// 将每个新连接的客户端添加到map中，并更新在线用户数
	clients[name] = &Client{Conn: ws, Name: name}

for {
		var msg int // 定义一个整型变量，用于接收客户端发送的消息（这里只接收数字）
		fmt.Println("Received from", name, ":", msg)
		// 读取客户端发送的消息，并存储到msg中
		// 如果出错，则表示客户端断开了连接，从全局变量中删除该客户端，并减少在线人数，并广播给其他客户端
		err := ws.ReadJSON(&msg)
		if err != nil {
			log.Printf("error: %v", err)
			//delete(clients, ws)       // 删除该客户端
			delete(clients, name)
			broadcast <- len(clients) // 广播当前在线人数
			break                     // 跳出循环
		}

// 如果没有出错，则表示客户端正常发送了消息（这里只处理数字1），增加在线人数，并广播给其他客户端
		if msg == 1 {
			broadcast <- len(clients) // 广播当前在线人数
		}
	}
}

// 监听广播通道，并向所有客户端发送消息的函数
func handleMessages() {
	for {
		msg := <-broadcast          // 从广播通道中接收消息
		fmt.Println("online:", msg) // 打印当前在线人数

// 遍历所有在线的客户端，并向他们发送消息
		for client := range clients {

err := clients[client].Conn.WriteJSON(msg) // 发送消息（这里只发送数字）
			if err != nil {
				log.Printf("error: %v", err)
				clients[client].Conn.Close() // 关闭连接
				delete(clients, client)      // 删除该客户端
			}
		}
	}
}

```

golang实现websocket统计实时在线人数

商品推荐

广告推荐

推荐文章