如何停止Java方法,直到它从MQTT代理收到消息为止

时间:2018-08-11 16:41:41

标签: java mqtt paho

我有一个带有2个方法的类,每个方法分别发布和预订一个公共MQTT代理。

public class operations{
    public void fetch(){
    // do something

     mqttConn.mqttSubscriberFetch(MQTTtopic);

    // do something
   }

    public void store(){
   // do something

    mqttConn.mqttSubscriberStore(MQTTtopic);

   // do something
   }

}

用于方法获取的MQTT方法如下:

public class mqttConnectionFetch implements  MqttCallback{

    MqttClient client;

    String topic;

    public mqttConnectionFetch() {
    }


    public void mqttSubscriberFetch(String topic) {

        final String MQTT_BROKER = "tcp://localhost:1883" ;
        final String CLIENT_ID = "fetch";
        try {
            client = new MqttClient(MQTT_BROKER, CLIENT_ID);
            client.connect();
            client.setCallback(this);
            client.subscribe(topic);
            MqttMessage message = new MqttMessage();

        } catch (MqttException e) {
            e.printStackTrace();
        }
    }



    @Override
    public void connectionLost(Throwable cause) {
        // TODO Auto-generated method stub

    }

    @Override
    public void messageArrived(String topic, MqttMessage message)
            throws Exception {

        System.out.println("the message received is "+message);

        if(message.toString().equals("Processed")) {
            MqttPublisherFetch("Processed", "operations/fetch");

        }

    }

    @Override
    public void deliveryComplete(IMqttDeliveryToken token) {
        // TODO Auto-generated method stub

    }


    public void MqttPublisherFetch(String message, String topic) {
        final String MQTT_BROKER = "tcp://localhost:1883" ;
        final String CLIENT_ID = "store";
        try {
            client = new MqttClient(MQTT_BROKER, CLIENT_ID);
            client.connect();
            createMqttMessage(message,topic);
            client.disconnect();
        } catch (MqttException e) {
            e.printStackTrace();
        }
    }

    private void createMqttMessage(String message, String topic) throws MqttException {
        MqttMessage publishMessage = new MqttMessage();
        publishMessage.setPayload(message.getBytes());
        client.publish(topic, publishMessage);

    }


}

现在,我正在尝试具有这样的功能:只要我的获取方法订阅了某个主题,如果来自代理的消息是“正在处理”,它就应该进入等待状态。虽然存储方法应该可以正常工作。并且当收到的消息为“已处理”时,提取应再次开始工作。

我使用普通的java wait()和start()尝试了此操作,但是我没有得到所需的输出。 有人可以帮我揭开这个谜吗?

1 个答案:

答案 0 :(得分:0)

据我了解,存储方法包括几个步骤。步骤之一是通过MQTT发送消息并等待响应。从store方法的客户端角度来看,它是同步的,即客户端只有在整个处理过程(包括通过MQTT进行的异步发送/接收)完成后才能接收响应。

当一个线程需要等待另一个线程中发生某种情况时,您要解决的是一个经典问题。有多种方法可以实现此目的,只需检查How to make a Java thread wait for another thread's output?中建议的选项数量即可。

最简单的方法是使用CountDownLatch。我将使用fetch方法,因为这是您为其提供代码的方法。

首先,您将修改mqttSubscriberFetch以创建内部CountDownLatch对象:

private CountDownLatch processingFinishedLatch;

public void mqttSubscriberFetch(String topic) {

    final String MQTT_BROKER = "tcp://localhost:1883" ;
    final String CLIENT_ID = "fetch";
    try {
        client = new MqttClient(MQTT_BROKER, CLIENT_ID);
        client.connect();
        client.setCallback(this);
        client.subscribe(topic);
        MqttMessage message = new MqttMessage();
        processingFinishedLatch = new CountDownLatch(1);

    } catch (MqttException e) {
        e.printStackTrace();
    }
}

然后您需要在消息接收回调中触发此信号:

@Override
public void messageArrived(String topic, MqttMessage message)
        throws Exception {

    System.out.println("the message received is "+message);

    if(message.toString().equals("Processed")) {
        MqttPublisherFetch("Processed", "operations/fetch");
        this.processingFinishedLatch.countDown();
    }

}

此外,您还需要为mqttConnectionFetch的客户端提供一种等待闩锁变为零的方法:

void waitForProcessingToFinish() throws InterruptedException {
     this.processingFinishedLatch.await();
}

fetch方法应像这样使用它:

public void fetch(){
// do something

mqttConn.mqttSubscriberFetch(MQTTtopic);

// do everything that is needed to initiate processing

mqttConn.waitForProcessingToFinish()

// do something

}

正在执行fetch的线程将在waitForProcessingToFinish中等待,直到闩锁达到零为止,然后在出现适当的消息时才会发生这种情况。

可以轻松修改此方法以解决消息永不出现时的问题。只需使用await with some timeout

boolean waitForProcessingToFinish(long timeout,
        TimeUnit unit) throws InterruptedException {
     return this.processingFinishedLatch.await();
}

fetch应该检查返回的值,如果发生超时,可能会向调用者返回错误。

请注意,此解决方案的不利之处在于正在处理fetch的线程将一直忙碌。如果这是线程池中用于处理传入HTTP请求的线程,则这可能会限制您的服务器可以同时处理的请求数。 有一些方法可以缓解这种情况,例如async servletsRxJavaSpring WebFlux。该问题的实际解决方案在很大程度上取决于您用于REST API实现的技术。