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

Question

我有一个带有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（）进行了尝试，但没有得到所需的输出。 有人可以帮我揭开神秘面纱吗？

Answer 1

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

当一个线程需要等待另一个线程中发生某种情况时，您要解决的是一个经典问题。 有很多方法可以实现此目的，只需检查如何使Java线程等待另一个线程的输出中提出的选项数量即可。 。

最简单的方法是使用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并设置一些超时时间即可 ：

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

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

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