3. Aeron发布与订阅

433 阅读2分钟

回顾

上一篇文章主要带大家一起阅读了MediaDriver.launch以及Aeron.connect相关源码,今天就带大家一起看下sub和pub相关的源码设计

Demo

public static void main(String[] args)
{
    final String channel = "aeron:ipc";
    final String message = "my message";
    final IdleStrategy idle = new SleepingIdleStrategy();
    final UnsafeBuffer unsafeBuffer = new UnsafeBuffer(ByteBuffer.allocate(256));
    try (MediaDriver driver = MediaDriver.launch();
        Aeron aeron = Aeron.connect();
        Subscription sub = aeron.addSubscription(channel, 10);
        Publication pub = aeron.addPublication(channel, 10))
    {
        while (!pub.isConnected())
        {
            idle.idle(); // 简单理解为阻塞即可
        }
        unsafeBuffer.putStringAscii(0, message);
        System.out.println("sending:" + message);
        while (pub.offer(unsafeBuffer) < 0)
        {
            idle.idle();
        }
        FragmentHandler handler = (buffer, offset, length, header) ->
            System.out.println("received:" + buffer.getStringAscii(offset));
        while (sub.poll(handler, 1) <= 0)
        {
            idle.idle();
        }
    }
}

Sub流程

image.png 通过上述流程图可以看到,addSub方法其实会把对应的指令写入到cncBytBuffer里,关于cncFile前一篇文章也介绍了,有不懂的可以再去回顾下。

这里把数据写入到cncFile里后,对应的命令又是谁来处理的呢?答案是:DriverConductor。 DriverConductor通过clientCommandAdapter来处理具体的命令,对应的流程如图:

image.png 这里sub有三种类型,ipcSub,spySub,netSub,分别对应进程间sub,本地mock,以及网络sub。 这里可以看到,sub通过完成后,会发送“ON_SUBSCRIPTION_READY”消息,表示当前订阅动作已经完成,ready消息在哪里接受的呢?答案是在awaitRespons方法里,会阻塞等待对应的ready消息 image.png awaitRespons处理流程如下:

image.png 最终是在DriverEventsAdapter的onMessage方法里处理ready消息。

这其实就是aeron源码读起来比较困难的地方,并不是传统的a调用b,b调用c,很多地方都是异步的,因此,大家在阅读相关源码的时候,一定不要着急。

Pub流程

addPub方法最终也是把对应的pub指令写入到对应的cncFile里,同样,也是DriverConductor来处理对应的pub指令,对应的流程图如下: image.png 这里pub有两种类型,ipcPub,netPub,分别对应进程间pub和网络pub,对应的ready消息处理和sub一样,也是在DriverEventsAdapter里来处理

小结

本篇文章依然是从最开始的Demo入口,分别分析了sub和pub相关流程,下一篇将会带大家一起看下aeron是如何发送以及接受数据的。