Question

我正在运行单个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程序是否有效/惯用。

Answer 1

您的解决方案将导致生成的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读取len（ch）条消息吗？

1 个答案: