能用 REST API 发送 WebSocket 消息吗？

在 Go 中开发一个应用程序时，需要在 REST API 处理和 WebSocket 服务器之间进行交互。文章描述了使用 Gorilla WebSocket 库在 Go 中实现此目标的方法。代码包括用于处理 WebSocket 连接、发送和接收消息以及设置 REST 端点的函数。通过将 WebSocket 变量声明为全局变量并使用赋值运算符，可以在 REST 端点处理程序中向 WebSocket 写入消息。提供的代码演示了如何使用 REST API 触发 WebSocket 消息，为开发人员提供了在 Go 应用程序中集成 RESTful 服务和实时通信的指导。

我需要扩展本教程的代码以运行常规休息进程和 websocket 服务器作为 go 例程的一部分。

这是 go 部分：

package main

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

    "github.com/gorilla/websocket"
)

var upgrader = websocket.upgrader{
    readbuffersize:  1024,
    writebuffersize: 1024,
}

var ws = websocket.conn{}

func reader(conn *websocket.conn) {
    for {
        // read in a message
        messagetype, p, err := conn.readmessage()
        if err != nil {
            log.println(err)
            return
        }
        // print out that message for clarity
        fmt.println(string(p))

        if err := conn.writemessage(messagetype, p); err != nil {
            log.println(err)
            return
        }
    }
}

func homepage(w http.responsewriter, r *http.request) {
    fmt.fprintf(w, "home page")
}

func restresponsepage(w http.responsewriter, r *http.request) {
    text := []byte("rest response")
    if err := ws.writemessage(websocket.textmessage, text); err != nil {
        log.println(err)
    }
    fmt.fprintf(w, "rest response")
}

func wsendpoint(w http.responsewriter, r *http.request) {

    upgrader.checkorigin = func(r *http.request) bool { return true }

    // upgrade this connection to a websocket
    // connection
    ws, err := upgrader.upgrade(w, r, nil)
    if err != nil {
        log.println(err)
    }

    log.println("client connected")
    err = ws.writemessage(1, []byte("hi from server!"))
    if err != nil {
        log.println(err)
    }

    reader(ws)
}

func setuproutes() {
    http.handlefunc("/", homepage)
    http.handlefunc("/restrequest", restresponsepage)
}

func setupwsroute() {
    http.handlefunc("/ws", wsendpoint)
}

func main() {
    fmt.println("hello world")

    go func() {
        setupwsroute()
        log.fatal(http.listenandserve(":8081", nil))
    }()

    setuproutes()

    log.fatal(http.listenandserve(":8080", nil))
}

以及客户端index.html：

  
    
    
    
    
  
  
    
Hello World

    
  

当我点击restrequest端点时，在写入restresponsepage响应写入器w之前，我想向websocket写入一条消息，我将其声明为全局var ws，并在连接升级时在读取器方法内进行初始化。 p>

这在某种程度上不起作用，相反，有一个异常，我不知道如何解决它。

解决方案

感谢@thinkgoodly 的帮助，改变本地 ws 的声明方式并使用赋值运算符而不是短变量声明实现了这一点。

这是完整的代码，以防有人想使用它。 :)

package main

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

    "github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{
    ReadBufferSize:  1024,
    WriteBufferSize: 1024,
}

var WS *websocket.Conn

func reader(conn *websocket.Conn) {
    for {
        // read in a message
        messageType, p, err := conn.ReadMessage()
        if err != nil {
            log.Println(err)
            return
        }
        // print out that message for clarity
        fmt.Println(string(p))

        if err := conn.WriteMessage(messageType, p); err != nil {
            log.Println(err)
            return
        }
    }
}

func homePage(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintf(w, "Home Page")
}

func restResponsePage(w http.ResponseWriter, r *http.Request) {
    text := []byte("rest response")
    if err := WS.WriteMessage(websocket.TextMessage, text); err != nil {
        log.Println(err)
    }
    fmt.Fprintf(w, "rest response")
}

func wsEndpoint(w http.ResponseWriter, r *http.Request) {

    upgrader.CheckOrigin = func(r *http.Request) bool { return true }

    // upgrade this connection to a WebSocket
    // connection
    WS, _ = upgrader.Upgrade(w, r, nil)
    // if err != nil {
    //  log.Println(err)
    // }

    log.Println("Client Connected")
    err1 := WS.WriteMessage(1, []byte("Hi from server!"))
    if err1 != nil {
        log.Println(err1)
    }

    reader(WS)
}

func setupRoutes() {
    http.HandleFunc("/", homePage)
    http.HandleFunc("/restRequest", restResponsePage)
}

func setupWsRoute() {
    http.HandleFunc("/ws", wsEndpoint)
}

func main() {
    fmt.Println("Hello World")

    go func() {
        setupWsRoute()
        log.Fatal(http.ListenAndServe(":8081", nil))
    }()

    setupRoutes()

    log.Fatal(http.ListenAndServe(":8080", nil))
}

到这里，我们也就讲完了《能用 REST API 发送 WebSocket 消息吗？》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！