golang + vue + websocket 实现的简单在线消息转发例子

实现的功能是简单的websocket消息在线转发

应该算是学习产物吧，现在的版本应该没bug了（之前版本会挂）
之前的bug原因，close管道之后继续发送会panic
另外，在后期建立大量连接测试的时候，发送会产生broken pipe的错误，并导致发送服务阻塞
解决办法：1、如果其他goroutine中有对关闭管道的写操作，尽量不要用close
2、原因是关闭信号没有正确传达，设置超时时间，规定时间未接收到数据，关闭；发送超时，关闭

• 服务端只做数据转发，格式由客户端定义和解析

go代码

gin 启动一个web服务，在对应路由处理函数里升级为 websocket。要在https的站点使用的话，TLS相关的配置是必须的（主要是下载证书文件，之后只能以 wss://域名/* 的方式使用）。如果实在没条件，把对应的中间件注释掉,RunTLS改为Run,但只能在http站点下使用，wss://变为ws://

package main

import (
	"github.com/gin-gonic/gin"
	"github.com/gorilla/websocket"
	"github.com/unrolled/secure"
	"log"
	"net/http"
	"sync"
	"time"
)

// 自定义客户端
type Client struct {
   
   
	// 引入锁
	sync.Mutex
	// websocket连接
	conn *websocket.Conn
	// 读到的信息在这里
	readChan chan []byte
	// 需要发送的信息在这里
	writeChan chan []byte
	// 协程之间通信，连接是否已关闭
	closeChan chan struct{
   
   }
	// close closeChan 使用，避免panic
	closed bool
}

// 读取一条信息
// false 代表连接关闭失效/等待超时
func (c *Client) ReadMessage() ([]byte, bool) {
   
   
	select {
   
   
	// 超时，关闭连接
	// todo 心跳信息没有处理，在外部处理(自定义)
	case <-time.After(time.Minute*5 + time.Second*10):
		c.Close()
		return nil, false
	case data := <-c.readChan:
		return data, true
	case <-c.closeChan:
		return nil, false
	}
}

// 写入一条信息
// 不能判断成功/失败
func (c *Client) WriteMessage(data []byte) {
   
   
	select {
   
   
	case c.writeChan <- data:
	case <-c.closeChan:
	case <-time.After(time.Second * 3):
		// 为了防止阻塞，如果处理过慢，这里可以增加服务的稳定，不会导致一直阻塞影响其他接收者接收
		c.Close()
	}
}

// 关闭连接
func (c *Client) Close() {
   
   
	c.Lock()
	if !c.closed {
   
   
		close(c.closeChan)
		c.closed = true
		// ===
		// conn.Close 可以不加锁,多次使用,线程安全
		c.conn.Close()
		// ===
	}
	c.Unlock()
}

func (c *Client) readLoop() {
   
   
	for {
   
   
		_, msg, err := c.conn.ReadMessage()
		if err != nil {
   
   
			c.Close()
			return
		}
		select {
   
   
		case c.readChan <- msg:
		case <-c.closeChan:
			return
		}
	}
}

func (c *Client) writeLoop() {
   
   
	for {
   
   
		select {
   
   
		case msg := <-c.writeChan:
			// 因为缓冲区的存在所以，不会立即发送，加上异常关闭的连接也被视为存在
			// 可能会出现 broken pipe
			err := c.conn.WriteMessage(websocket.TextMessage, msg)
			if err != nil {
   
   
				c.Close()
				return
			}
		case <-c.closeChan:
			return
		}
	}
}

// 升级成websocket，开启读（生产者）、写（消费者）的协程，返回自定义*Client
func NewClient(c *gin.Context) (*Client, error) {
   
   
	//升级get请求为webSocket协议
	ws, err := upGrader