ZeroMQ的介绍之Poller详解

漫天飞舞的雪花

于 2024-09-25 14:31:08 发布

阅读量734

点赞数 16

文章标签：网络服务器后端

本文链接：https://blog.csdn.net/lvdepeng123/article/details/142522513

版权

zmq.Poller 是 ZeroMQ 提供的一种机制，用于监控多个 socket 的事件，如读（读入）和写（写出）事件。这种机制可以实现高效的 I/O 多路复用，允许在一个线程内处理多个 socket 的 I/O 操作。

事件类型

zmq.POLLIN：表示 socket 上有数据可读。即，当一个 socket 的 POLLIN 事件被触发时，意味着这个 socket 上有一个或多个消息待处理，可以调用 recv 方法来读取消息。
zmq.POLLOUT：表示 socket 上可以写数据。即，当一个 socket 的 POLLOUT 事件被触发时，意味着这个 socket 可以发送数据，可以调用 send 方法来发送消息。
zmq.POLLERR：表示 socket 上发生了错误。这个事件通常用来检测异常情况。

作用

监控多个 socket 上的事件：
- 这个机制允许你在一个单独的线程或事件循环中监听多个 socket，从而避免使用多线程或多进程来处理并发 I/O 操作。
进行 I/O 多路复用：
- 通过 poller，你可以同时处理来自多个 socket 的消息，并根据事件的到达时间顺序进行处理。这种机制在实现高性能的网络通信应用时非常有用。

一般用在什么时候

代理服务器：
- 在代理模式（如 ROUTER-DEALER 或 PUB-SUB）下，经常使用 poller 来同时监听前端和后端的连接，并根据接收到的消息进行转发。
负载均衡器：
- 在负载均衡器场景下，你可能需要同时监听多个客户端和工作节点的连接，并根据请求和响应进行调度和分配工作。
高并发服务器：
- 在构建高并发服务器时，可以使用 poller 同时监听多个客户端连接，以便在有数据可读或可写时进行处理，从而提高 I/O 处理效率。
消息路由：
- 在需要灵活的消息路由策略时，可以使用 poller 来监听多个 socket，从而动态地将消息从一个 socket 路由到另一个 socket。

代码示例

我们将进一步扩展一个代理服务器的示例，介绍如何监听和处理 POLLIN 和 POLLOUT 事件。

package main

import (
    "fmt"
    "log"
    "github.com/pebbe/zmq4"
)

func main() {
    context, _ := zmq4.NewContext()
    defer context.Term()

    // 创建 frontend 和 backend socket
    frontend, _ := zmq4.NewSocket(zmq4.ROUTER)
    defer frontend.Close()
    frontend.Bind("tcp://*:5559")

    backend, _ := zmq4.NewSocket(zmq4.DEALER)
    defer backend.Close()
    backend.Bind("tcp://*:5560")

    // 创建 poller 并添加 socket
    items := zmq4.NewPoller()
    items.Add(frontend, zmq4.POLLIN)
    items.Add(backend, zmq4.POLLIN)

    for {
        // 等待事件发生
        sockets, err := items.Poll(-1) // -1 表示无限等待
        if err != nil {
            log.Fatal(err)
        }

        for _, socket := range sockets {
            switch socket.Socket {
            case frontend:
                // 处理来自前端的消息，并转发到后端
                msg, _ := frontend.RecvMessage(0)
                fmt.Println("Message from frontend to backend:", msg)
                
                // 将消息发送到 backend，如果 backend 可写（POLLOUT），那么可以立即发送，否则等待 POLLOUT 事件
                if backendEvents, _ := zmq4.Poller.New(); backendEvents.PollAll(backend, zmq4.POLLOUT); len(backendEvents) > 0 {
                    backend.SendMessage(msg)
                } else {
                    items.Add(backend, zmq4.POLLOUT)
                }
            case backend:
                // 处理来自后端的消息，并转发到前端
                msg, _ := backend.RecvMessage(0)
                fmt.Println("Message from backend to frontend:", msg)
                
                // 将消息发送到 frontend，如果 frontend 可写（POLLOUT），那么可以立即发送，否则等待 POLLOUT 事件
	            if frontendEvents, _ := zmq4.Poller.New(); frontendEvents.PollAll(frontend, zmq4.POLLOUT); len(frontendEvents) > 0 {
                    frontend.SendMessage(msg)
                } else {
                    items.Add(frontend, zmq4.POLLOUT)
                }
            }
        }
    }
}

`zmq.PollAll`

zmq.PollAll 方法可以指定多个事件掩码一起监听。它类似于 Poll 方法，但返回的是所有事件的集合，并且可以指定多个事件掩码一起监听。

进一步详细说明

初始化 Context 和 Sockets：
- context：ZeroMQ 上下文，用于创建 socket。
- frontend：ROUTER socket，用于前端（客户端）连接。
- backend：DEALER socket，用于后端（工作者）连接。
创建 Poller 并添加 Socket：
- items := zmq.NewPoller()：创建一个新的 Poller 实例。
- items.Add(frontend, zmq.POLLIN) 和 items.Add(backend, zmq.POLLIN)：分别将前端和后端 socket 添加到 Poller 中，并指定监听 POLLIN 事件。
等候事件发生：
- sockets, err := items.Poll(-1)：等待任意一个添加到 Poller 的 socket 上发生事件。
- 如果 Poller 检测到 frontend 有数据可读，则从 frontend 读取消息，并尝试将消息发送到 backend。如果 backend 不可写，则添加 POLLIN 事件监听。
- 类似地，如果 Poller 检测到 backend 有数据可读，则从 backend 读取消息，并尝试将消息发送到 frontend。如果 frontend 不可写，则添加 POLLIN 事件监听。