libhv/examples/mqtt/mqtt_client_test.cpp

/*
 * mqtt client
 *
 * @build   make examples
 *
 * @test    bin/mqtt_client_test 127.0.0.1 1883 topic payload
 *
 */

#include "mqtt_client.h"
using namespace hv;

/*
 * @test    MQTTS
 * #define  TEST_SSL 1
 *
 * @build   ./configure --with-mqtt --with-openssl && make clean && make
 *
 */
#define TEST_SSL        0
#define TEST_AUTH       0
#define TEST_RECONNECT  1
#define TEST_QOS        0

int main(int argc, char** argv) {
    if (argc < 5) {
        printf("Usage: %s host port topic payload\n", argv[0]);
        return -10;
    }
    const char* host = argv[1];
    int port = atoi(argv[2]);
    const char* topic = argv[3];
    const char* payload = argv[4];

    MqttClient cli;

    cli.onConnect = [topic, payload](MqttClient* cli) {
        printf("connected!\n");
#if TEST_QOS
        cli->subscribe(topic, 1, [topic, payload](MqttClient* cli) {
            printf("subscribe OK!\n");
            cli->publish(topic, payload, 1, 0, [](MqttClient* cli) {
                printf("publish OK!\n");
            });
        });
#else
        cli->subscribe(topic);
        cli->publish(topic, payload);
#endif
    };

    cli.onMessage = [](MqttClient* cli, mqtt_message_t* msg) {
        printf("topic: %.*s\n", msg->topic_len, msg->topic);
        printf("payload: %.*s\n", msg->payload_len, msg->payload);
        cli->disconnect();
        cli->stop();
    };

    cli.onClose = [](MqttClient* cli) {
        printf("disconnected!\n");
    };

#if TEST_AUTH
    cli.setAuth("test", "123456");
#endif

#if TEST_RECONNECT
    reconn_setting_t reconn;
    reconn_setting_init(&reconn);
    reconn.min_delay = 1000;
    reconn.max_delay = 10000;
    reconn.delay_policy = 2;
    cli.setReconnect(&reconn);
#endif

    int ssl = 0;
#if TEST_SSL
    ssl = 1;
#endif
    cli.connect(host, port, ssl);
    cli.run();
    return 0;
}