pgmq 0.34.0-alpha.4

A distributed message queue for Rust applications, on Postgres.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62

use pgmq::Message;
use rand::RngExt;
use serde_derive::{Deserialize, Serialize};

#[derive(Serialize, Debug, Deserialize, Eq, PartialEq)]
struct MyMessage {
    foo: String,
    num: u64,
}

impl Default for MyMessage {
    fn default() -> Self {
        MyMessage {
            foo: "bar".to_owned(),
            num: rand::rng().random_range(0..100),
        }
    }
}

#[ignore]
#[cfg(feature = "install-sql-embedded")]
#[tokio::test]
async fn test_sql_lifecycle() {
    let test_num = rand::rng().random_range(0..100000);
    let test_queue = format!("test_sql_lifecycle_{}", test_num);
    let db_url = std::env::var("DATABASE_URL")
        .unwrap_or_else(|_| "postgres://postgres:postgres@localhost:5432/postgres".to_owned());
    let test_db_name = format!("pgmq_test_{}", test_num);
    let test_db_url = replace_db_string(&db_url, &format!("/{test_db_name}"));
    println!("test_db_url: {test_db_url}");
    let pool = sqlx::PgPool::connect(&db_url).await.unwrap();
    sqlx::query(sqlx::AssertSqlSafe(format!(
        "CREATE DATABASE {test_db_name};"
    )))
    .execute(&pool)
    .await
    .unwrap();

    let queue = pgmq::PGMQueueExt::new(test_db_url, 1).await.unwrap();
    #[cfg(feature = "install-sql-embedded")]
    queue.install_sql_from_embedded().await.unwrap();
    queue.create(&test_queue).await.unwrap();

    let sent_msg = MyMessage::default();
    let msg_id = queue.send(&test_queue, &sent_msg).await.unwrap();
    let read_msg: Message<MyMessage> = queue.read(&test_queue, 30).await.unwrap().unwrap();
    assert_eq!(msg_id, read_msg.msg_id);
    assert_eq!(sent_msg, read_msg.message);
    queue.archive(&test_queue, msg_id).await.unwrap();
    let read_none: Option<Message<MyMessage>> = queue.read(&test_queue, 30).await.unwrap();
    assert!(read_none.is_none());
}

fn replace_db_string(s: &str, replacement: &str) -> String {
    match s.rfind('/') {
        Some(pos) => {
            let prefix = &s[0..pos];
            format!("{prefix}{replacement}")
        }
        None => s.to_string(),
    }
}