Akka分布式Pub / Sub背压

时间:2017-12-11 01:17:46

标签: java akka publish-subscribe akka-stream akka-cluster

我正在使用Akka Distributed Pub / Sub并拥有一个发布者和一个订阅者。我的发布者比订阅者快。有没有办法在某一点之后放慢发布商的速度?

发布商代码:

public class Publisher extends AbstractActor {
    private ActorRef mediator;

    static public Props props() {
        return Props.create(Publisher.class, () -> new Publisher());
    }

    public Publisher () {
        this.mediator = DistributedPubSub.get(getContext().system()).mediator();
        this.self().tell(0, ActorRef.noSender());
    }

    @Override
    public Receive createReceive() {
        return receiveBuilder()
            .match(Integer.class, msg -> {
                // Sending message to Subscriber
                mediator.tell(
                    new DistributedPubSubMediator.Send(
                        "/user/" + Subscriber.class.getName(),
                        msg.toString(),
                        false),
                    getSelf());

                getSelf().tell(++msg, ActorRef.noSender());
            })
            .build();
    }
}

订阅者代码:

public class Subscriber extends AbstractActor {
    static public Props props() {
        return Props.create(Subscriber.class, () -> new Subscriber());
    }

    public Subscriber () {
        ActorRef mediator = DistributedPubSub.get(getContext().system()).mediator();
        mediator.tell(new DistributedPubSubMediator.Put(getSelf()), getSelf());
    }

    @Override
    public Receive createReceive() {
        return receiveBuilder()
            .match(String.class, msg -> {
                System.out.println("Subscriber message received: " + msg);
                Thread.sleep(10000);
            })
            .build();
    }
}

1 个答案:

答案 0 :(得分:1)

不幸的是,按照目前的设计,我不认为有办法提供"背压"原始发件人。由于您使用ActorRef.tell将消息发送到mediator,因此无法获得下游接收器正在备份的信号。这是因为您使用的方法tell会返回void

切换询问

如果您将tell切换为ask,则可以设置一个适当的Timeout值,该值至少会让您知道何时未在特定内容中收到回复持续时间。

切换到流

"Back-pressure" is a primary feature of akka streams。因此,通过切换到流实现,您将能够实现您期望的目标。

如果可以从原始数据创建流Source,那么您可以使用Sink.actorRefSink创建mediator并使用Flow.throttle控制流向调解员的流量。