我正在使用HTTP和Websockets构建一个单页应用程序。用户提交表单,然后我将响应发送给客户端。以下是摘要客户端。
var html = `<!DOCTYPE html>
<meta charset="utf-8">
<head>
</head>
<body>
<script>
var ws = new WebSocket("ws://localhost:8000/ws")
ws.onmessage = function(e) {
document.getElementById("output").innerHTML += e.data + "<br>"
}
function submitFunction() {
document.getElementById("output").innerHTML += ""
return false
}
</script>
<form
enctype="multipart/x-www-form-urlencoded"
action="http://localhost:8000/"
method="post"
>`
这是服务器。如果请求不是POST,我将编写/渲染html(parseAndExecute),以建立新的websocket连接。如果请求是POST(来自表单),则我开始处理并最终写入websocket。
func (c *Config) ServeHtml(w http.ResponseWriter,r *http.Request) {
if r.Method == http.MethodPost {
//process
channel <- data
}
c.parseAndExecute(w)
}
func (sh *SocketHandler) ServeHTTP(w http.ResponseWriter,r *http.Request) {
ws,err := upgrader.Upgrade(w,r,nil)
if err != nil {
w.Write([]byte(fmt.Sprintf("",err)))
return
}
//defer ws.Close()
// discard received messages
go func(c *websocket.Conn) {
for {
if _,_,err := c.NextReader(); err != nil {
c.Close()
break
}
}
}(ws)
data <- channel
只有在不刷新页面的情况下,一切都可以正常工作。如果我不刷新,则可以继续提交表单,并看到不同的输出逐行显示。需要澄清的是,它实际上仅在页面已经打开时才起作用,因此从不调用parseAndExecute
。该函数解析并执行html / template来创建新的websocket客户端。
任何页面刷新或最初浏览localhost:8000都会在服务器上导致websocket: close sent
。
我不确定该如何解决。服务器是否需要正常处理断开连接并允许重新连接?还是客户需要做点什么?看来服务器应该升级/ws
上的任何连接,所以建立多少个新的websocket客户端无关紧要,但显然我的理解是错误的。
我没有关闭服务器上的websocket连接,因为只要程序正在运行,它就应该保持连接状态。当用户停止该程序时,我认为它将自动关闭。
完整的SocketHandler代码:
func (sh *SocketHandler) ServeHTTP(w http.ResponseWriter,err)))
return
}
// discard received messages
go func(c *websocket.Conn) {
for {
if _,err := c.NextReader(); err != nil {
c.Close()
break
}
}
}(ws)
cmd := <-sh.cmdCh
log.Printf("Executing")
stdout,err := cmd.StdoutPipe()
if err != nil {
w.Write([]byte(err.Error()))
return
}
defer stdout.Close()
stderr,err := cmd.StderrPipe()
if err != nil {
w.Write([]byte(err.Error()))
return
}
defer stderr.Close()
if err := cmd.Start(); err != nil {
w.Write([]byte(err.Error()))
return
}
s := bufio.NewScanner(io.MultiReader(stdout,stderr))
for s.Scan() {
err := ws.WriteMessage(1,s.Bytes())
if err != nil {
log.Printf("Error writing to client: %v",err)
ws.Close()
}
}
if err := cmd.Wait(); err != nil {
w.Write([]byte(err.Error()))
return
}
log.Println("Done")
}