initial commit

third_party/libhv/examples/mqtt/mqtt_pub.c

@@ -0,0 +1,134 @@
+/*
+ * mqtt publish
+ *
+ * @build   make examples
+ * @sub     bin/mqtt_sub 127.0.0.1 1883 topic
+ * @pub     bin/mqtt_pub 127.0.0.1 1883 topic payload
+ *
+ */
+
+#include "hv.h"
+#include "mqtt_client.h"
+
+/*
+ * @test    MQTTS
+ * #define  TEST_SSL 1
+ *
+ * @build   ./configure --with-mqtt --with-openssl && make clean && make
+ *
+ */
+#define TEST_SSL        0
+#define TEST_AUTH       0
+
+/*
+ * workflow:
+ * mqtt_client_new -> mqtt_client_xxx -> mqtt_client_run
+ *
+ * mqtt_client_set_xxx ->
+ * mqtt_client_connect ->
+ * on_connack -> mqtt_client_publish ->
+ * on_puback -> mqtt_client_disconnect ->
+ * on_disconnect -> mqtt_client_stop
+ *
+ */
+
+static void on_mqtt(mqtt_client_t* cli, int type) {
+    printf("on_mqtt type=%d\n", type);
+    switch(type) {
+    case MQTT_TYPE_CONNECT:
+        printf("mqtt connected!\n");
+        break;
+    case MQTT_TYPE_DISCONNECT:
+        printf("mqtt disconnected!\n");
+    {
+        mqtt_message_t* msg = (mqtt_message_t*)mqtt_client_get_userdata(cli);
+        HV_FREE(msg);
+        mqtt_client_set_userdata(cli, NULL);
+        mqtt_client_stop(cli);
+    }
+        break;
+    case MQTT_TYPE_CONNACK:
+        printf("mqtt connack!\n");
+    {
+        mqtt_message_t* msg = (mqtt_message_t*)mqtt_client_get_userdata(cli);
+        if (msg == NULL) return;
+        int mid = mqtt_client_publish(cli, msg);
+        printf("mqtt publish mid=%d\n", mid);
+        if (msg->qos == 0) {
+            mqtt_client_disconnect(cli);
+        } else if (msg->qos == 1) {
+            // wait MQTT_TYPE_PUBACK
+        } else if (msg->qos == 2) {
+            // wait MQTT_TYPE_PUBREC
+        }
+    }
+        break;
+    case MQTT_TYPE_PUBACK: /* qos = 1 */
+        printf("mqtt puback mid=%d\n", cli->mid);
+        mqtt_client_disconnect(cli);
+        break;
+    case MQTT_TYPE_PUBREC: /* qos = 2 */
+        printf("mqtt pubrec mid=%d\n", cli->mid);
+        // wait MQTT_TYPE_PUBCOMP
+        break;
+    case MQTT_TYPE_PUBCOMP: /* qos = 2 */
+        printf("mqtt pubcomp mid=%d\n", cli->mid);
+        mqtt_client_disconnect(cli);
+        break;
+    default:
+        break;
+    }
+}
+
+static int mqtt_publish(const char* host, int port, const char* topic, const char* payload) {
+    mqtt_client_t* cli = mqtt_client_new(NULL);
+    if (cli == NULL) return -1;
+    cli->keepalive = 10;
+
+    // client_id
+    char client_id[64];
+    snprintf(client_id, sizeof(client_id), "mqtt_pub_%ld", hv_getpid());
+    printf("client_id: %s\n", client_id);
+    mqtt_client_set_id(cli, client_id);
+    // will
+    mqtt_message_t will;
+    memset(&will, 0, sizeof(will));
+    will.topic = "will";
+    will.payload = "This is a will.";
+    mqtt_client_set_will(cli, &will);
+#if TEST_AUTH
+    mqtt_client_set_auth(cli, "test", "123456");
+#endif
+
+    mqtt_message_t* msg = NULL;
+    HV_ALLOC_SIZEOF(msg);
+    msg->topic = topic;
+    msg->topic_len = strlen(topic);
+    msg->payload = payload;
+    msg->payload_len = strlen(payload);
+    msg->qos = 1;
+    mqtt_client_set_userdata(cli, msg);
+    mqtt_client_set_callback(cli, on_mqtt);
+
+    int ssl = 0;
+#if TEST_SSL
+    ssl = 1;
+#endif
+    mqtt_client_connect(cli, host, port, ssl);
+    mqtt_client_run(cli);
+    mqtt_client_free(cli);
+    return 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];
+
+    return mqtt_publish(host, port, topic, payload);
+}