feat: Add send command

Adds a command that can be used to send simple messages.

In the future this may also accept arguments from stdin to send
non-text messages.
This commit is contained in:
Vincent Ambo 2017-10-20 16:36:01 +02:00 committed by Karl Erik Asbjørnsen
parent a023e05474
commit d742c6526b

View file

@ -2,7 +2,7 @@ extern crate clap;
extern crate posix_mq; extern crate posix_mq;
use clap::{App, SubCommand, Arg, ArgMatches, AppSettings}; use clap::{App, SubCommand, Arg, ArgMatches, AppSettings};
use posix_mq::{Name, Queue}; use posix_mq::{Name, Queue, Message};
use std::fs::{read_dir, File}; use std::fs::{read_dir, File};
use std::io::{self, Read, Write}; use std::io::{self, Read, Write};
use std::process::exit; use std::process::exit;
@ -78,6 +78,24 @@ fn run_receive(queue_name: &str) {
}; };
} }
fn run_send(queue_name: &str, content: &str) {
let name = Name::new(queue_name).expect("Invalid queue name");
let queue = Queue::open(name).expect("Could not open queue");
let message = Message {
data: content.as_bytes().to_vec(),
priority: 0,
};
match queue.send(&message) {
Ok(_) => (),
Err(e) => {
writeln!(io::stderr(), "Could not send message: {}", e).ok();
exit(1);
}
}
}
fn main() { fn main() {
let ls = SubCommand::with_name("ls").about("list message queues"); let ls = SubCommand::with_name("ls").about("list message queues");
@ -105,6 +123,13 @@ fn main() {
.about("Receive a message from a queue") .about("Receive a message from a queue")
.arg(&queue_arg); .arg(&queue_arg);
let send = SubCommand::with_name("send")
.about("Send a message to a queue")
.arg(&queue_arg)
.arg(Arg::with_name("message")
.help("the message to send")
.required(true));
let matches = App::new("mq") let matches = App::new("mq")
.setting(AppSettings::SubcommandRequiredElseHelp) .setting(AppSettings::SubcommandRequiredElseHelp)
@ -114,6 +139,7 @@ fn main() {
.subcommand(inspect) .subcommand(inspect)
.subcommand(create) .subcommand(create)
.subcommand(receive) .subcommand(receive)
.subcommand(send)
.get_matches(); .get_matches();
match matches.subcommand() { match matches.subcommand() {
@ -121,6 +147,10 @@ fn main() {
("inspect", Some(cmd)) => run_inspect(cmd.value_of("queue").unwrap()), ("inspect", Some(cmd)) => run_inspect(cmd.value_of("queue").unwrap()),
("create", Some(cmd)) => run_create(cmd), ("create", Some(cmd)) => run_create(cmd),
("receive", Some(cmd)) => run_receive(cmd.value_of("queue").unwrap()), ("receive", Some(cmd)) => run_receive(cmd.value_of("queue").unwrap()),
("send", Some(cmd)) => run_send(
cmd.value_of("queue").unwrap(),
cmd.value_of("message").unwrap()
),
_ => unimplemented!(), _ => unimplemented!(),
} }
} }