Interface Overview

// Talk about the interface 
func Handler(c *gin.Context) {
    
	fmt.Println(" Enter function ")
	// Get sender uid And the sender uid
	token := module.WsGetTokenFromHeader(c.Request.Header)
	fmt.Println(token)
	stuNum := dao.Redis.Get(token).Val()
	toStuNum := c.Query("toStuNum")
	if stuNum == toStuNum {
    
		c.JSON(http.StatusOK, gin.H{
    
			"status": 203,
			"msg":    " I can't communicate with myself ",
		})
		return
	}
	// upgrade websocket agreement 
	conn, err := (&websocket.Upgrader{
    
		CheckOrigin: func(r *http.Request) bool {
    
			return true
		},
		// Put the obtained parameters into this array 
		Subprotocols: []string{
    token},
	}).Upgrade(c.Writer, c.Request, nil) // upgrade ws agreement 
	if err != nil {
    
		http.NotFound(c.Writer, c.Request)
		return
	}
	// Create a user instance 
	client := &models.Client{
    
		ID:           createID(stuNum, toStuNum), //1->2
		SendID:       createID(toStuNum, stuNum), //2->1
		FromUsername: stuNum,
		ToUsername:   toStuNum,
		Socket:       conn,
		Send:         make(chan []byte),
	}
	// User registration to user management 
	mux.Lock()
	models.Manager.Register <- client
	mux.Unlock()
	go client.Read()
	go client.Write()
	go models.BatchRead(toStuNum, stuNum)
}

Interface slice

If the front end uses Sec-WebSocket-Protocol Field passing token It will cause websocket The problem of disconnection as soon as it is connected
At that time, it needs to be in Upgrader add to

		// Put the obtained parameters into this array 
		Subprotocols: []string{
    token},

To establish a connection

websocket upgrade

	conn, err := (&websocket.Upgrader{
    
		CheckOrigin: func(r *http.Request) bool {
    
			return true
		},
		// Put the obtained parameters into this array 
		Subprotocols: []string{
    token},
	}).Upgrade(c.Writer, c.Request, nil) // upgrade ws agreement 
	if err != nil {
    
		http.NotFound(c.Writer, c.Request)
		return
	}