GOLANG实现的一个websock后台服务

最新推荐文章于 2024-10-30 16:46:30 发布

原创

最新推荐文章于 2024-10-30 16:46:30 发布 · 613 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#GOLANG #websocket

本文介绍如何使用GOLANG实现一个WebSocket后台服务，该服务每秒向客户端发送心跳，并能响应客户端的任何消息。服务通过channel和互斥锁处理并发的客户端通信，确保高并发场景下的稳定性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

用GOLANG和实现了一个websock后台服务，每隔1s，后台给客户端发送一个heartbeat，客户端给它发什么，它就回复什么。使用channel和互斥锁封装了对客户端的接收和发送动作，让服务器可以支持海量并发。

package main

import (
	"errors"
	_ "fmt"
	"github.com/gorilla/websocket"
	"net/http"
	"sync"
	"time"
)

type Connection struct {
	wsConn    *websocket.Conn
	inChan    chan []byte
	outChan   chan []byte
	closeChan chan byte

	mutex    sync.Mutex
	isClosed bool
}

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),
	}

	go conn.readLoop()
	go conn.writeLoop()

	return
}

func (conn *Connection) ReadMessage() (data []byte, err error) {
	select {
	case data = <-conn.inChan:
	case <-conn.closeChan:
		err = errors.New("connection is closed")

	}

	return
}

func (conn *Connection) WriteMessage(data []byte) (err error) {
	select {
	case conn.outChan <- data:
	case <-