publisher_example.rs

/********************************************************************************
 * Copyright (c) 2023 Contributors to the Eclipse Foundation
 *
 * See the NOTICE file(s) distributed with this work for additional
 * information regarding copyright ownership.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Apache License Version 2.0 which is available at
 * https://www.apache.org/licenses/LICENSE-2.0
 *
 * SPDX-License-Identifier: Apache-2.0
 ********************************************************************************/

use std::{str::FromStr, time::SystemTime};

use clap::{command, Parser};
use log::{error, info};
use up_rust::{UMessageBuilder, UPayloadFormat, UStatus, UTransport, UUri};
use up_transport_mqtt5::{Mqtt5Transport, MqttClientOptions, TransportMode};

/// Publishes messages to a given topic using the MQTT 5 transport.
#[derive(Parser)]
#[command(version, about, long_about = None)]
struct Command {
    /// The uProtocol topic to publish messages to.
    #[arg(value_name = "URI", env = "TOPIC", default_value = "up://Vehicle_B/A8000/2/8A50", value_parser = UUri::from_str)]
    topic: UUri,

    #[command(flatten)]
    mqtt_client_options: MqttClientOptions,
}

#[tokio::main]
async fn main() -> Result<(), UStatus> {
    env_logger::init();

    let command = Command::parse();
    let authority = command.topic.authority_name.clone();

    let client = Mqtt5Transport::new(
        TransportMode::InVehicle,
        command.mqtt_client_options,
        authority,
    )
    .await?;

    loop {
        tokio::time::sleep(std::time::Duration::from_secs(1)).await;
        let current_time = SystemTime::now()
            .duration_since(SystemTime::UNIX_EPOCH)
            .unwrap()
            .as_secs();
        let message = UMessageBuilder::publish(command.topic.clone())
            .with_ttl(3600)
            .build_with_payload(
                current_time.to_string(),
                UPayloadFormat::UPAYLOAD_FORMAT_TEXT,
            )
            .expect("Failed to build message");

        if let Err(e) = client.send(message).await {
            error!("Failed to publish message: {e}");
        } else {
            info!(
                "Published message [topic: {}]: {}",
                command.topic.to_uri(true),
                current_time,
            );
        }
    }
}