可以读取管道 ch 的消息长度吗？

在处理消息频道 `ch` 时，文章探索了使用 `len(ch)` 确定频道中消息数量的有效性。虽然这种方法可以防止生成 goroutine 在没有消息发送到频道时旋转，但它可能导致批量处理消息，这需要使用计时器超时。因此，文章建议使用选择语句与超时相结合来管理批量处理，从而确保 goroutine 仅在有消息需要处理时才执行。

我正在运行单个 goroutine 来处理与某些用户相关的消息通道。处理消息后，用户状态将由 goroutine 更新并存储在数据库中。当对数据库的请求正在进行时，可以将许多消息发送到通道。我想在向数据库发送另一个请求之前处理它们。

目前我正在使用 len(ch) 来检查通道中的消息数量并在 for 循环中读取它们。

func (c *consumer) handleUser(userID string, ch chan Message) {
  user := c.db.LoadUser(userID)
  for {
    var msgs []Message
    for n := len(ch); n > 0; n-- {
      msgs = append(msgs, <-ch)
    }
    apply.Messages(user, msgs)
    c.db.SaveUser(user)
}

ch := make(chan Message, 100)
go c.handleUser("user-1", ch) 

我在互联网上搜索这是否是一些常见的模式，但我找不到类似的解决方案，我想知道我的方法对于 go 程序是否有效/惯用。

解决方案

您的解决方案将导致生成的 goroutine 在通道上旋转，直到至少发送一条消息。换句话说，goroutine 根本不会阻塞。

在这里，您尝试批量处理多条消息。有不同的方法可以实现这一点。但要回答的主要问题是：你如何知道这批消息已完成？发送者 goroutine 可能知道这一点，并且它可以将所有消息打包在一个切片中。另一方面，您可能不知道批次何时准备就绪。在这些情况下，您需要使用超时，如下例所示。

func (c *consumer) handleUser(userID string, ch chan Message) {
    user := c.db.LoadUser(userID)
    for {
        var msgs []Message
        select {
        case msg := <-ch:
            //Append the message in the current batch slice
            msgs = append(msgs, msg)
        //Wait up to 5 seconds and then process the batch
        case <-time.After(time.Second * 5):
            //Timeout: process the batch of messages
            if len(msgs) > 0 {
                apply.Messages(user, msgs)
                c.db.SaveUser(user)
            }
        }
    }
}

请注意，执行此函数的可能 goroutine 仅在确实有事情要做时才运行。

本篇关于《可以读取管道 ch 的消息长度吗？》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！