Add benchmark options to tedge mqtt pub

Signed-off-by: Didier Wenzek <didier.wenzek@free.fr>

Author: didier-wenzek

crates/core/tedge/src/cli/mqtt/cli.rs

@@ -30,6 +30,12 @@ pub enum TEdgeMqttCli {
         /// Retain flag
         #[clap(short, long = "retain")]
         retain: bool,
+        /// Repeat the message
+        #[clap(long)]
+        repeat: Option<u32>,
+        /// Pause between repeated messages (e.g., 60s, 1h)
+        #[clap(long, default_value = "1s")]
+        sleep: SecondsOrHumanTime,
     },
 
     /// Subscribe a MQTT topic.
@@ -69,6 +75,8 @@ impl BuildCommand for TEdgeMqttCli {
                     message,
                     qos,
                     retain,
+                    repeat,
+                    sleep,
                 } => MqttPublishCommand {
                     host: config.mqtt.client.host.clone(),
                     port: config.mqtt.client.port.into(),
@@ -80,6 +88,8 @@ impl BuildCommand for TEdgeMqttCli {
                     ca_file: auth_config.ca_file.clone(),
                     ca_dir: auth_config.ca_dir,
                     client_auth_config: auth_config.client,
+                    count: repeat.unwrap_or(1),
+                    sleep: sleep.duration(),
                 }
                 .into_boxed(),
                 TEdgeMqttCli::Sub {

crates/core/tedge/src/cli/mqtt/publish.rs

@@ -22,6 +22,8 @@ pub struct MqttPublishCommand {
     pub ca_file: Option<Utf8PathBuf>,
     pub ca_dir: Option<Utf8PathBuf>,
     pub client_auth_config: Option<MqttAuthClientConfig>,
+    pub count: u32,
+    pub sleep: std::time::Duration,
 }
 
 #[async_trait::async_trait]
@@ -36,7 +38,15 @@ impl Command for MqttPublishCommand {
     }
 
     async fn execute(&self, _: TEdgeConfig) -> Result<(), MaybeFancy<anyhow::Error>> {
-        Ok(publish(self).await?)
+        let mut i = 0;
+        loop {
+            publish(self).await?;
+            i += 1;
+            if i == self.count {
+                return Ok(());
+            }
+            tokio::time::sleep(self.sleep).await;
+        }
     }
 }