rs-zero 0.2.6

Rust-first microservice framework inspired by go-zero engineering practices
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110

#![cfg(feature = "cache-redis")]

use std::time::Duration;

use rs_zero::{
    cache::{CacheKey, CacheStore},
    cache_redis::{
        RedisCacheConfig, RedisCacheStore, RedisClusterRedirect, RedisClusterRouter,
        RedisUnavailablePolicy,
    },
};

#[tokio::test]
#[ignore = "requires RS_ZERO_TEST_REDIS_URL pointing to a running Redis instance"]
async fn redis_single_node_cache_roundtrip_and_degradation_config() {
    let url = std::env::var("RS_ZERO_TEST_REDIS_URL")
        .unwrap_or_else(|_| "redis://127.0.0.1:6379".to_string());
    let store = RedisCacheStore::new(RedisCacheConfig {
        url,
        namespace: format!("rs-zero-redis-hardening-{}", unique_suffix()),
        unavailable_policy: RedisUnavailablePolicy::fail_open_for_cache(),
        ..RedisCacheConfig::default()
    })
    .expect("store");

    store.health_check().await.expect("health");
    let key = CacheKey::new(store.namespace(), ["cluster", "roundtrip"]);
    store
        .set_raw(&key, b"ok".to_vec(), Some(Duration::from_secs(5)))
        .await
        .expect("set");
    assert_eq!(
        store.get_raw(&key).await.expect("get"),
        Some(b"ok".to_vec())
    );
    store.delete(&key).await.expect("delete");
}

#[tokio::test]
#[ignore = "requires RS_ZERO_TEST_REDIS_CLUSTER_URL pointing to a Redis Cluster node"]
async fn redis_cluster_env_is_available_for_manual_moved_ask_validation() {
    let url = std::env::var("RS_ZERO_TEST_REDIS_CLUSTER_URL")
        .expect("RS_ZERO_TEST_REDIS_CLUSTER_URL is required");
    let store = RedisCacheStore::new(RedisCacheConfig {
        url,
        namespace: format!("rs-zero-redis-cluster-{}", unique_suffix()),
        cluster: rs_zero::cache_redis::RedisClusterConfig {
            enabled: true,
            max_redirects: 8,
            read_from_replicas: false,
        },
        ..RedisCacheConfig::default()
    })
    .expect("cluster store");

    store.health_check().await.expect("cluster health");
    let first = CacheKey::new(store.namespace(), ["{manual-cluster-a}", "value"]);
    let second = CacheKey::new(store.namespace(), ["{manual-cluster-b}", "value"]);
    store
        .set_raw(&first, b"first".to_vec(), Some(Duration::from_secs(5)))
        .await
        .expect("set first");
    store
        .set_raw(&second, b"second".to_vec(), Some(Duration::from_secs(5)))
        .await
        .expect("set second");
    assert_eq!(
        store.get_raw(&first).await.expect("get first"),
        Some(b"first".to_vec())
    );
    assert_eq!(
        store.get_raw(&second).await.expect("get second"),
        Some(b"second".to_vec())
    );
    store
        .delete_many(&[first.clone(), second.clone()])
        .await
        .expect("delete many");
    assert!(
        store
            .get_raw(&first)
            .await
            .expect("get deleted first")
            .is_none()
    );
    assert!(
        store
            .get_raw(&second)
            .await
            .expect("get deleted second")
            .is_none()
    );

    let mut router = RedisClusterRouter::new(8);
    let redirect = RedisClusterRedirect::parse(&format!(
        "MOVED {} 127.0.0.1:6379",
        RedisClusterRouter::slot_for_key(&first)
    ))
    .expect("redirect parse");
    router.apply_redirect(&redirect);
    assert!(router.endpoint_for_key(&first).is_some());
}

fn unique_suffix() -> String {
    let nanos = std::time::SystemTime::now()
        .duration_since(std::time::UNIX_EPOCH)
        .unwrap_or_default()
        .as_nanos();
    format!("{nanos}-{}", std::process::id())
}