GoFrame缓冲输出到客户端Flush()

原创于 2022-08-08 13:38:12 发布 · 659 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#golang

golang 专栏收录该内容

5 篇文章

订阅专栏

本文探讨了在GHTTP框架中如何实现响应的实时输出，通过修改`ghttp_response_writer.go`文件的`Flush`方法来达到类似`http.ResponseWriter`的flush效果。示例代码展示了在标准库`net/http`中使用`Flusher`接口进行实时响应的方法，以及在GHTTP中进行相同操作的差异。文章强调了在处理长运行请求时，实时反馈数据的重要性。

版本：v2.x

官方文档中介绍任何时候可以通过OutputBuffer方法输出缓冲区数据到客户端，并清空缓冲区数据。但使用r.Response.Flush()发现并不能实时输出。

修改下net\ghttp\ghttp_response_writer.go文件的Flush方法

    if w.buffer.Len() > 0 {
		w.writer.Write(w.buffer.Bytes())
		w.writer.(http.Flusher).Flush() //添加此行
		w.buffer.Reset()
	}

以下为使用net/http原生写法

package main

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

func handler(res http.ResponseWriter, r *http.Request) {
	fmt.Fprintln(res, "<body>") //不加body没有效果
	for i := 0; i < 10; i++ {
		//fmt.Fprint(res, "<script>document.body.innerHTML = ''</script>")
		fmt.Fprintf(res, "%d<br>", i)
		if f, ok := res.(http.Flusher); ok {
			f.Flush()
		} else {
			log.Println("Damn, no flush")
		}
		time.Sleep(1000 * time.Millisecond)
	}

	fmt.Fprintln(res, "</body>")
}

func main() {
	http.HandleFunc("/", handler)
	log.Fatal(http.ListenAndServe(":8080", nil))
}