golang中的响应流回客户端的操作指南

大家好，今天本人给大家带来文章《golang中的响应流回客户端的操作指南》，文中内容主要涉及到，如果你对Golang方面的知识点感兴趣，那就请各位朋友继续看下去吧~希望能真正帮到你们，谢谢！

我在服务器端的方法中有以下代码，它将流响应发送回客户端。

var cfg = "..." // comes from request parameters
var clientId = "...." // comes from request parameters
var data = allCids.([]int64)
// can we run this in parallel with X number of workers?
for _, cid := range data {
  pd := repo.GetCustomerData(strconv.FormatInt(cid, 10))
  if !pd.IsCorrect {
    continue
  }
  resources := us.helperCom.GenerateResourceString(pd)
  val, _ := us.GenerateInfo(clientId, resources, cfg)
  if err := stream.Send(val); err != nil {
    log.Printf("send error %v", err)
  }
}

我的困惑是我们可以让它并发运行吗？或者流总是以单线程方式运行？

正确答案

您可以使工作并发，但不能使流上的发送并发。来自grpc-go docs：

使用流时，必须注意避免从不同的 goroutine 对同一个 stream 多次调用 sendmsg 或 recvmsg 。 [...]

因此，您可以在单独的 goroutine 中运行并发代码，并在公共通道上发送输出值。然后，主流处理程序遍历此通道并顺序调用 stream.send - 因此请记住，只有当网络响应花费的时间少于获取数据的时间时，所有这些都是值得的。

代码如下所示：

// buffered, so the workers can send and exit immediately
out := make(chan , len(data))

// you will have to close the out chan when all work is done
wg := &sync.WaitGroup{}
wg.Add(len(data))
go func() {
    wg.Wait()
    close(out)
}()

for _, cid := range data {
  // don't close around the loop variable
  go func (id int64) {
    defer wg.Done()
    val, err := // obtain output value somehow
    if err != nil {
        return
    }
    out <- val
  }(cid)
}

for val := range out {
  if err := stream.Send(val); err != nil {
    log.Printf("send error %v", err)
  }
}

goroutines 的数量是 data 中元素的数量。如果你想控制goroutine的数量，可以批处理data。如果这样做，请相应地调整通道缓冲区。

终于介绍完啦！小伙伴们，这篇关于《golang中的响应流回客户端的操作指南》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！