mqtt api

opencv-motion-detect/mqtt_helper.hpp

@@ -73,10 +73,17 @@ class MqttHelper {
     private:
     string mqtt_addr;
     map<string, on_msg_fun_ptr_t> topics;
-    MQTTAsync client;
     protected:
     public:
-    int state = 0; // 0: initial ; 1: ready; 2: disconnected; 3: destroyed
+    MQTTAsync client;
+    enum class State{
+        None,
+        Ready,
+        Disconnected,
+        Destroyed
+    };
+
+    State state = State::None; // 0: initial ; 1: ready; 2: disconnected; 3: destroyed
     MqttHelper(string mqtt_addr, string id, int kai = 20, int cs = 1): mqtt_addr(mqtt_addr){
         // make connection, throw excpetions
         MQTTAsync_connectOptions conn_opts = MQTTAsync_connectOptions_initializer;
@@ -107,34 +114,36 @@ class MqttHelper {
 
     // can be called only state
     int subscribe(string topic, on_msg_fun_ptr_t on_msg/*, on_res_fun_ptr_t on_success = nullptr, on_res_fun_ptr_t on_failure = nullptr*/){
-        assert(state);
+        assert(state != State::None);
         MQTTAsync_responseOptions opts = MQTTAsync_responseOptions_initializer;
 	    int rc;
         // if(topics.contains(topic)){
         //     MQTTAsync_unsubscribe(topic);
         // }
         if(topics.count(topic) > 0){
+            spdlog::warn("already subscribed to {}", topic);
             return 0;
         }
-
         topics[topic] = on_msg;
         opts.onSuccess = on_subscribed;
         opts.onFailure = on_subscribed_fail;
-        opts.context = client;
+        opts.context = this;
+        
         if ((rc = MQTTAsync_subscribe(client, topic.c_str(), 1, &opts)) != MQTTASYNC_SUCCESS)
         {
-            printf("Failed to start subscribe, return code %d\n", rc);
-            exit(EXIT_FAILURE);
+            spdlog::error("failed to subscribe {}: {}",topic, MQTTAsync_strerror(rc));
+        }else{
+            spdlog::info("subscribe {} success", topic);
         }
     }
 
-    MqttPub operator [](string topic){
+    MqttPub operator[](string topic){
         MqttPub pub(client, topic);
         return pub;
     }
 
     int operator()(string topic, const void* const data, int len) {
-        if(topics.count(topic) >0) {
+        if(topics.count(topic) >0 && topics[topic] != nullptr) {
             topics[topic](std::move(MqttPub(client, topic)), data, len);
         }else{
             spdlog::warn("no handler installed for topic: {}", topic);
@@ -144,12 +153,13 @@ class MqttHelper {
 
     ~MqttHelper(){
         int rc;
-        MQTTAsync_disconnectOptions disc_opts = MQTTAsync_disconnectOptions_initializer;
+        MQTTAsync_disconnectOptions opts = MQTTAsync_disconnectOptions_initializer;
+        opts.onSuccess = on_disconn;
+        opts.context = this;
 
-        if ((rc = MQTTAsync_disconnect(client, &disc_opts)) != MQTTASYNC_SUCCESS)
+        if ((rc = MQTTAsync_disconnect(client, &opts)) != MQTTASYNC_SUCCESS)
         {
-            printf("Failed to start disconnect, return code %d\n", rc);
-            exit(EXIT_FAILURE);
+            spdlog::info("Failed to start disconnect: {}", MQTTAsync_strerror(rc));
         }
 
         MQTTAsync_destroy(&client);
@@ -159,7 +169,7 @@ class MqttHelper {
 
 void on_connlost(void *context, char *cause)
 {
-	MQTTAsync client = (MQTTAsync)context;
+	MqttHelper *self = (MqttHelper *) context;
 	MQTTAsync_connectOptions conn_opts = MQTTAsync_connectOptions_initializer;
 	int rc;
 
@@ -170,7 +180,7 @@ void on_connlost(void *context, char *cause)
 	printf("Reconnecting\n");
 	conn_opts.keepAliveInterval = 20;
 	conn_opts.cleansession = 1;
-	if ((rc = MQTTAsync_connect(client, &conn_opts)) != MQTTASYNC_SUCCESS)
+	if ((rc = MQTTAsync_connect(self->client, &conn_opts)) != MQTTASYNC_SUCCESS)
 	{
 		printf("Failed to start connect, return code %d\n", rc);
 	}
@@ -202,66 +212,81 @@ int on_msgarrvd(void *context, char *topic, int topicLen, MQTTAsync_message *mes
 
 void on_disconn(void* context, MQTTAsync_successData* response)
 {
-	printf("Successful disconnection\n");
     MqttHelper *self = (MqttHelper *) context;
-    self->state = 2;
-
+    self->state = MqttHelper::State::Disconnected;
+    printf("Successful disconnection\n");
 }
 
 
 void on_subscribed(void* context, MQTTAsync_successData* response)
 {
+    MqttHelper *self = (MqttHelper *) context;
 	printf("Subscribe succeeded\n");
 }
 
 void on_subscribed_fail(void* context, MQTTAsync_failureData* response)
 {
-	printf("Subscribe failed, rc %d\n", response ? response->code : 0);
+    MqttHelper *self = (MqttHelper *) context;
+	spdlog::error("Subscribe failed: {}", MQTTAsync_strerror(response ? response->code : 0));
 }
 
 
 void on_conn_fail(void* context, MQTTAsync_failureData* response)
 {
+    MqttHelper *self = (MqttHelper *) context;
 	printf("Connect failed, rc %d\n", response ? response->code : 0);
 }
 
 void on_sent(void* context, MQTTAsync_successData* response)
 {
-	MQTTAsync client = (MQTTAsync)context;
-	MQTTAsync_disconnectOptions opts = MQTTAsync_disconnectOptions_initializer;
-	int rc;
-
+	MqttHelper *self = (MqttHelper *) context;
 	printf("Message with token value %d delivery confirmed\n", response->token);
-
-	opts.onSuccess = on_disconn;
-	opts.context = client;
-
-	if ((rc = MQTTAsync_disconnect(client, &opts)) != MQTTASYNC_SUCCESS)
-	{
-		printf("Failed to start sendMessage, return code %d\n", rc);
-		exit(EXIT_FAILURE);
-	}
 }
 
 void on_connected(void* context, MQTTAsync_successData* response)
 {
     MqttHelper *self = (MqttHelper *) context;
-    self->state = 1;
+    self->state = MqttHelper::State::Ready;
     spdlog::info("connected");
+
+    MQTTAsync_responseOptions opts = MQTTAsync_responseOptions_initializer;
+    int rc;
+    // if(topics.contains(topic)){
+    //     MQTTAsync_unsubscribe(topic);
+    // }
+
+    string topic = "test.direct.topic";
+
+    opts.onSuccess = on_subscribed;
+    opts.onFailure = on_subscribed_fail;
+    opts.context = self;
+    
+    if ((rc = MQTTAsync_subscribe(self->client, topic.c_str(), 1, &opts)) != MQTTASYNC_SUCCESS)
+    {
+        printf("Failed to start subscribe, return code %d\n", rc);
+        exit(EXIT_FAILURE);
+    }else{
+        spdlog::info("successfully subscribed the test direct topic: {}", topic);
+    }
 }
 
 #define DEBUG
 #ifdef DEBUG
 
 int main(){
-    MqttHelper hlp("tcp://evcloud.ilabservice.cloud:1883", "testaabbc");
-    while(!hlp.state){
-        spdlog::info("not state");
-        this_thread::sleep_for(chrono::seconds(3));
+    {
+        MqttHelper hlp("tcp://evcloud.ilabservice.cloud:1883", "testaabbc");
+        while(hlp.state == MqttHelper::State::None){
+            spdlog::info("not state");
+            this_thread::sleep_for(chrono::milliseconds(200));
+        }
+        hlp["hello"].pub("hello", 6, 1);
+        hlp.subscribe("bbbb", NULL);
+        this_thread::sleep_for(chrono::seconds(5));
     }
-    hlp["hello"].pub("hello", 6, 1);
-    hlp.subscribe("bbbb", NULL);
-    this_thread::sleep_for(chrono::seconds(10));
+
+    this_thread::sleep_for(chrono::seconds(5));
+    
     return 0;
 }