实现一个简单的websocket 的连接

[yaogengzhu]

第一步：路由注册

package ws

import (
	"ginchat/service"

	"github.com/gin-gonic/gin"
)

func WsRouter() *gin.Engine {
	r := gin.New()
	v1 := r.Group("/v1/ws")

	v1.GET("/connect", service.HandlerWs)
	return r
}

第二步实现路由方法

package service

import (
	"fmt"
	"log"
	"net/http"

	"github.com/gin-gonic/gin"
	"github.com/gorilla/websocket"
)

var upgrade = websocket.Upgrader{
	CheckOrigin: func(r *http.Request) bool {
		return true
	},
}

func HandlerWs(c *gin.Context) {
	// 升级成websocket协议
	ws, err := upgrade.Upgrade(c.Writer, c.Request, nil)
	if err != nil {
		log.Fatalln(err, "err")
	}
	fmt.Println("ws connect success")
	// 完成时关闭连接释放资源
	defer ws.Close()

	go func() {
		// 监听连接“完成”事件，其实也可以说丢失事件
		<-c.Done()
		fmt.Println("ws lost connection")
	}()

	for {
		// 读取客户端发送过来的消息，如果没发就会一直阻塞住
		mt, message, err := ws.ReadMessage()

		if err != nil {
			fmt.Println("read error")
			fmt.Println(err)
			break
		}
		if string(message) == "ping" {
			message = []byte("恭喜您连接成功，您可以发送消息了")
		}
		// 将读取到的消息写回客户端，这里是原封不动的写回去
		fmt.Print("receive message: ", mt, "\n")
		err = ws.WriteMessage(mt, message)
		if err != nil {
			fmt.Println(err)
			break
		}

	}
}

参考文章： https://juejin.cn/post/7103737973782511646

[yaogengzhu]

ngnix 部署时处理

  location /v1/ws/connect {
      proxy_pass http://xxxx:8888/v1/ws/connect;
      proxy_http_version 1.1;
      proxy_set_header Upgrade $http_upgrade;
      proxy_set_header Connection "Upgrade";
      proxy_set_header Host $host;
      proxy_set_header X-Real-IP $remote_addr;
      proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
      proxy_set_header X-Forwarded-Proto $scheme;
  }