golangにgin-gonicサーバーを使用してストリームAPIを書き込む方法は？試したc.Streamの作業が無かった

Question

golangでgin-gonicサーバーを使用して1つのストリーミングAPIを作成したかったのです。

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) 

しかし、私がエンドポイントにヒットしようとすると、ちょうど突っ込んでしまい、応答がありません。 誰かが私がやっている間違いを指摘できるように、ストリーム機能を使用しましたか？ありがとう！

Answer 1

ストリームが終了した場合は、falseを返す必要があります。そして、チャンを閉じる。

package main 

import (
    "io" 
    "time" 

    "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.Run() 
}