12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758 |
- package controllers
- import (
- logger "eta/eta_mini_ht_api/common/component/log"
- "fmt"
- beego "github.com/beego/beego/v2/server/web"
- "github.com/gorilla/websocket"
- "net/http"
- "time"
- )
- var upgrader = websocket.Upgrader{
- CheckOrigin: func(r *http.Request) bool {
- fmt.Println(r.Header.Get("Authorization"))
- return true
- },
- }
- type WebSocketController struct {
- beego.Controller
- }
- func (c *WebSocketController) Connect() {
- ws, err := upgrader.Upgrade(c.Ctx.ResponseWriter, c.Ctx.Request, nil)
- test := c.Ctx.Request.Header.Get("Authorization")
- fmt.Println("发大水范德萨" + test)
- if err != nil {
-
- return
- }
- defer ws.Close()
- for {
- var msg []byte
- _, msg, err = ws.ReadMessage()
- if err != nil {
- logger.Error("", err)
- break
- }
-
- fmt.Printf("Received: %s\n", msg)
-
- err = ws.WriteMessage(websocket.TextMessage, []byte("Server: "+string(msg)))
- if err != nil {
-
- break
- }
- time.Sleep(time.Second)
- }
- }
|