finga
1100ba1b13
Add support for timeout and deadline parameters when messages are sent to a message queue. To keep code duplication small, the `utils.rs` file contains a helper function to parse the `Duration` from the `String` received from the timeout parameter. Add conflict rules for the timeout and deadline parameters.
69 lines
2 KiB
Rust
69 lines
2 KiB
Rust
use crate::utils;
|
|
use anyhow::Result;
|
|
use chrono::DateTime;
|
|
use clap::Clap;
|
|
|
|
/// Send a message to a message queue
|
|
#[derive(Clap, Debug)]
|
|
pub struct Send {
|
|
/// Use priority PRIO, PRIO >= 0
|
|
#[clap(short, long, default_value = "0")]
|
|
pub priority: u32,
|
|
/// Do not block
|
|
#[clap(short, long)]
|
|
pub non_blocking: bool,
|
|
/// Timeout in "<timeout>[s]" (default) or "<timeout>ms"
|
|
#[clap(short = 'o', long, conflicts_with = "deadline")]
|
|
pub timeout: Option<String>,
|
|
/// Deadline until messages are received (format: "%Y-%m-%d %H:%M:%S")
|
|
#[clap(short, long, conflicts_with = "timeout")]
|
|
pub deadline: Option<String>,
|
|
/// Name of the queue
|
|
#[clap(value_name = "QUEUE")]
|
|
pub queue: String,
|
|
/// Message to be sent to the queue
|
|
#[clap(value_name = "MESSAGE")]
|
|
pub msg: String,
|
|
}
|
|
|
|
impl Send {
|
|
pub fn run(&self, verbose: bool) -> Result<()> {
|
|
let mq = &mut posixmq::OpenOptions::writeonly();
|
|
|
|
if self.non_blocking {
|
|
mq.nonblocking();
|
|
}
|
|
|
|
if let Some(timeout) = &self.timeout {
|
|
mq.open(&self.queue)?.send_timeout(
|
|
self.priority,
|
|
&self.msg.as_bytes(),
|
|
utils::parse_duration(timeout)?,
|
|
)?;
|
|
|
|
if verbose {
|
|
println!("Sent message: \"{}\" to queue: {}", &self.msg, &self.queue);
|
|
}
|
|
} else if let Some(deadline) = &self.deadline {
|
|
mq.open(&self.queue)?.send_deadline(
|
|
self.priority,
|
|
&self.msg.as_bytes(),
|
|
DateTime::parse_from_str(deadline, "%Y-%m-%d %H:%M:%S")?.into(),
|
|
)?;
|
|
|
|
if verbose {
|
|
println!("Sent message: \"{}\" to queue: {}", &self.msg, &self.queue);
|
|
}
|
|
} else {
|
|
mq.open(&self.queue)?
|
|
.send(self.priority, &self.msg.as_bytes())?;
|
|
|
|
if verbose {
|
|
println!("Sent message: \"{}\" to queue: {}", &self.msg, &self.queue);
|
|
}
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
}
|