去 websocket eof

Question

我正在尝试制作一个简单的命令转发器，将我的家用计算机连接到我拥有的服务器，以便我可以将命令推送到我的服务器，而我的家用电脑可以获取它。这些命令对于我的下载器来说是简单的暂停/恢复。我的设计是，在服务器上，我运行一个集线器实例，它创建一个用于传递命令的窗口和一个用于后端将这些命令传递到我的电脑的窗口。我用一个通道限制这两个“窗口”，它们运行一个服务器。当客户端连接并向集线器发送消息时，它会通过通道传输到后端窗口，然后传输到真正的后端（在我的家用电脑上）。当后端响应集线器上的后端窗口时，集线器将结果打印回客户端。

通过这种方法，只有第一条消息可以通过并与我的下载器一起使用。每次收到消息才能正常工作时，我都必须将家用电脑的后端重新连接到集线器。我认为这不是使用 websocket 的正确方法，所以我在这里。一次成功请求后（当后端完成工作并回复结果时），它会永远循环并出现 EOF 错误。

代码的重要部分是：

• 主要可执行文件
• 集线器处理程序
• 后端连接器

如果您将源代码放入 GOPATH 中（i正在为 go 的提示版本开发它以支持现代 websockets），编译它： go build gosab/cmd，运行它：

• ./cmd -mode="hub" hub
• ./cmd -mode= "backend" --address="localhost:8082" backend

要将消息传递到集线器，请使用以下 javascript：

var s = new WebSocket("ws://localhost:8082")
s.send("1 5")

那么我该如何处理它呢？通道是两个不同请求之间进行通信的好方法吗？

Answer 1

我很惊讶你还没有收到这个问题的答复。

您需要做的是类似于下面的代码。当您收到传入的 websocket 连接时，会为该连接生成一个新的 goroutine。如果你让该 goroutine 结束，它会断开 websocket 客户端的连接。

我假设您不一定要在同一台计算机上运行客户端和服务器。如果您总是这样，那么最好通过通道等进行内部通信，而不是使用 websockets 或网络端口。我只是提到这一点，因为我不完全确定你用它来做什么。我只是希望我回答了你问题的正确部分。

package main

import (
    "code.google.com/p/go.net/websocket"
    "flag"
    "fmt"
    "net/http"
    "os"
    "time"
)

type Message struct {
    RequestID      int
    Command        string
    SomeOtherThing string
    Success        bool
}

var mode *string = flag.String("mode", "<nil>", "Mode: server or client")
var address *string = flag.String("address", "localhost:8080", "Bind address:port")

func main() {
    flag.Parse()

    switch *mode {
    case "server":
        RunServer()
    case "client":
        RunClient()
    default:
        flag.Usage()
    }
}

func RunServer() {
    http.Handle("/", http.FileServer(http.Dir("www")))
    http.Handle("/server", websocket.Handler(WSHandler))
    fmt.Println("Starting Server")
    err := http.ListenAndServe(*address, nil)
    if err != nil {
        fmt.Printf("HTTP failed: %s\n", err.Error())
        os.Exit(1)
    }
}

func WSHandler(ws *websocket.Conn) {
    defer ws.Close()
    fmt.Println("Client Connected")
    for {
        var message Message
        err := websocket.JSON.Receive(ws, &message)
        if err != nil {
            fmt.Printf("Error: %s\n", err.Error())
            return
        }
        fmt.Println(message)

        // do something useful here...

        response := new(Message)
        response.RequestID = message.RequestID
        response.Success = true
        response.SomeOtherThing = "The hot dog left the castle as requested."
        err = websocket.JSON.Send(ws, response)
        if err != nil {
            fmt.Printf("Send failed: %s\n", err.Error())
            os.Exit(1)
        }
    }
}

func RunClient() {
    fmt.Println("Starting Client")
    ws, err := websocket.Dial(fmt.Sprintf("ws://%s/server", *address), "", fmt.Sprintf("http://%s/", *address))
    if err != nil {
        fmt.Printf("Dial failed: %s\n", err.Error())
        os.Exit(1)
    }
    incomingMessages := make(chan Message)
    go readClientMessages(ws, incomingMessages)
    i := 0
    for {
        select {
        case <-time.After(time.Duration(2e9)):
            i++
            response := new(Message)
            response.RequestID = i
            response.Command = "Eject the hot dog."
            err = websocket.JSON.Send(ws, response)
            if err != nil {
                fmt.Printf("Send failed: %s\n", err.Error())
                os.Exit(1)
            }
        case message := <-incomingMessages:
            fmt.Println(message)
        }
    }
}

func readClientMessages(ws *websocket.Conn, incomingMessages chan Message) {
    for {
        var message Message
        err := websocket.JSON.Receive(ws, &message)
        if err != nil {
            fmt.Printf("Error: %s\n", err.Error())
            return
        }
        incomingMessages <- message
    }
}