如何在 golang 中使用 gin-gonic 服务器编写流 API?试过 c.Stream 没用 [英] How to write a stream API using gin-gonic server in golang? Tried c.Stream didnt work
本文介绍了如何在 golang 中使用 gin-gonic 服务器编写流 API?试过 c.Stream 没用的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!
问题描述
我想在 golang 中使用 gin-gonic 服务器创建一个流 API.
I wanted to create one streaming API using gin-gonic server in golang.
func StreamData(c *gin.Context) {
chanStream := make(chan int, 10)
go func() {for i := 0; i < 5; i++ {
chanStream <- i
time.Sleep(time.Second * 1)
}}()
c.Stream(func(w io.Writer) bool {
c.SSEvent("message", <-chanStream)
return true
})
}
router.GET("/stream", controller.StreamData)
但是当我试图到达终点时,它只是卡住了,没有响应.是否有人使用了流功能,以便他/她可以指出我可能正在做的错误.谢谢!
But when I am trying to hit the endpoint, it just stucks and no response comes. Has someone used the stream function so that he/she can point the mistake which I might be doing. Thanks!
推荐答案
如果流结束,您应该返回 false.并关闭 chan.
You should return false if the stream ended. And close the chan.
package main
import (
"io"
"time"
"github.com/gin-gonic/contrib/static"
"github.com/gin-gonic/gin"
"github.com/mattn/go-colorable"
)
func main() {
gin.DefaultWriter = colorable.NewColorableStderr()
r := gin.Default()
r.GET("/stream", func(c *gin.Context) {
chanStream := make(chan int, 10)
go func() {
defer close(chanStream)
for i := 0; i < 5; i++ {
chanStream <- i
time.Sleep(time.Second * 1)
}
}()
c.Stream(func(w io.Writer) bool {
if msg, ok := <-chanStream; ok {
c.SSEvent("message", msg)
return true
}
return false
})
})
r.Use(static.Serve("/", static.LocalFile("./public", true)))
r.Run()
}
附加
客户端代码应该是:
<!DOCTYPE html>
<html>
<head>
<meta charset="UTF-8">
<title></title>
</head>
<body>
<script>
var stream = new EventSource("/stream");
stream.addEventListener("message", function(e){
console.log(e.data);
});
</script>
</body>
</html>
这篇关于如何在 golang 中使用 gin-gonic 服务器编写流 API?试过 c.Stream 没用的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持IT屋!
查看全文