use env_logger;
use lapin_async as lapin;
use std::{
sync::{
Arc,
atomic::{AtomicBool, Ordering},
},
thread,
time,
};
use crate::lapin::{
BasicProperties, Connection, ConnectionProperties, ConsumerSubscriber,
message::Delivery,
options::*,
types::FieldTable,
};
#[derive(Debug)]
struct Subscriber {
hello_world: Arc<AtomicBool>,
}
impl ConsumerSubscriber for Subscriber {
fn new_delivery(&self, delivery: Delivery) {
println!("received message: {:?}", delivery);
println!("data: {}", std::str::from_utf8(&delivery.data).unwrap());
assert_eq!(delivery.data, b"Hello world!");
self.hello_world.store(true, Ordering::SeqCst);
}
fn drop_prefetched_messages(&self) {}
fn cancel(&self) {}
}
#[test]
fn connection() {
let _ = env_logger::try_init();
let addr = std::env::var("AMQP_ADDR").unwrap_or_else(|_| "amqp://127.0.0.1:5672/%2f".into());
let conn = Connection::connect(&addr, ConnectionProperties::default()).wait().expect("connection error");
println!("CONNECTED with configuration: {:?}", conn.configuration());
let channel_a = conn.create_channel().wait().expect("create_channel");
let channel_b = conn.create_channel().wait().expect("create_channel");
let queue = channel_a.queue_declare("hello-async", QueueDeclareOptions::default(), FieldTable::default()).wait().expect("queue_declare");
println!("[{}] state: {:?}", line!(), conn.status().state());
println!("[{}] declared queue: {:?}", line!(), queue);
channel_a.queue_purge("hello-async", QueuePurgeOptions::default()).wait().expect("queue_purge");
println!("[{}] state: {:?}", line!(), conn.status().state());
let queue = channel_b.queue_declare("hello-async", QueueDeclareOptions::default(), FieldTable::default()).wait().expect("queue_declare");
println!("[{}] state: {:?}", line!(), conn.status().state());
println!("will consume");
let hello_world = Arc::new(AtomicBool::new(false));
let subscriber = Subscriber { hello_world: hello_world.clone(), };
channel_b.basic_consume(&queue, "my_consumer", BasicConsumeOptions::default(), FieldTable::default(), Box::new(subscriber)).wait().expect("basic_consume");
println!("[{}] state: {:?}", line!(), conn.status().state());
println!("will publish");
let payload = b"Hello world!";
channel_a.basic_publish("", "hello-async", BasicPublishOptions::default(), payload.to_vec(), BasicProperties::default()).wait().expect("basic_publish");
println!("[{}] state: {:?}", line!(), conn.status().state());
thread::sleep(time::Duration::from_millis(100));
assert!(hello_world.load(Ordering::SeqCst));
}