Golang http.HandleFunc如何实现运行时输出实时展示到浏览器

你遇到的输出无法实时刷新的问题，由以下几个原因导致，对应修改方案如下：

1. 提前配置正确的响应头 #

浏览器默认会缓冲小体积响应内容，要触发逐段渲染，需要在handler函数最开头就添加响应头配置：

// 如果输出HTML内容可改为 text/html; charset=utf-8
w.Header().Set("Content-Type", "text/plain; charset=utf-8")
// 禁用缓存
w.Header().Set("Cache-Control", "no-cache")
// 避免浏览器自动识别内容类型导致缓冲
w.Header().Set("X-Content-Type-Options", "nosniff")

2. 修复格式化输出语法错误 #

你当前的fmt.Fprintf(w, "Connecting to host", host)写法有误，格式化字符串中没有占位符，host参数不会被拼接进输出内容，需要修改为：

// 添加%s占位符，末尾加换行避免浏览器缓冲单行内容
fmt.Fprintf(w, "Connecting to host %s\n", host)

你当前的Flush调用写法是正确的，但需要确认你没有使用会缓冲响应的中间件（比如部分gzip压缩中间件默认会攒够一定体积才输出内容），如果有这类中间件需要先关闭。另外如果类型断言w.(http.Flusher)报错，说明你使用的框架/中间件包装了响应对象，需要换成框架提供的响应刷新方法。

3. 修复代码逻辑隐患 #

你在for循环中写的defer scpClient.Close()不会在每轮循环结束执行，只会在整个handler函数退出时批量执行，会导致大量SSH连接泄漏，建议改为每轮循环结束主动调用Close，或者把每轮循环的处理逻辑封装为独立子函数，在子函数中使用defer。

另外你当前代码多处使用panic，一旦单台主机处理失败就会直接中断整个流程，已经写入的响应内容也可能被截断，建议改为输出错误信息后继续处理下一台主机：

client, session, err := connectToHost(user, host, key_path)
if err != nil {
    fmt.Fprintf(w, "Connect to host %s failed: %v\n", host, err)
    w.(http.Flusher).Flush()
    continue
}

完成以上修改后即可实现每处理一台主机，就实时向浏览器输出一行内容的效果。

内容的提问来源于stack exchange，提问作者Manu