oxia-client 0.1.0

Native Rust client SDK for Oxia, a distributed key-value store. Provides an idiomatic, asynchronous API built on Tokio.
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

use oxia::{Notification, OxiaClient};
use std::time::Duration;
use tracing::info;
use tracing::level_filters::LevelFilter;

#[tokio::main]
async fn main() {
    tracing_subscriber::fmt()
        .with_env_filter(
            tracing_subscriber::EnvFilter::builder()
                .with_default_directive(LevelFilter::INFO.into())
                .from_env_lossy(),
        )
        .with_target(false)
        .init();

    let client = OxiaClient::connect("localhost:6648").await.unwrap();

    // Subscribe to notifications
    let mut notifications = client.notifications().await.unwrap();
    info!("Subscribed to notifications, performing operations...");

    // Spawn a task to listen for notifications
    let listener = tokio::spawn(async move {
        let mut count = 0;
        while let Some(notification) = notifications.recv().await {
            match &notification {
                Notification::KeyCreated { key, version_id } => {
                    info!("Notification: KeyCreated key={key:?} version_id={version_id:?}");
                }
                Notification::KeyModified { key, version_id } => {
                    info!("Notification: KeyModified key={key:?} version_id={version_id:?}");
                }
                Notification::KeyDeleted { key } => {
                    info!("Notification: KeyDeleted key={key:?}");
                }
                Notification::KeyRangeDeleted {
                    key,
                    key_range_last,
                } => {
                    info!(
                        "Notification: KeyRangeDeleted key={key:?} key_range_last={key_range_last:?}"
                    );
                }
                other => info!("Notification: {other}"),
            }
            count += 1;
            if count >= 5 {
                break;
            }
        }
        info!("Received {} notifications total", count);
    });

    // Give the notification listener time to connect
    tokio::time::sleep(Duration::from_millis(500)).await;

    // Perform operations that will generate notifications
    client.put("notify/key1", "value1").await.unwrap();
    info!("Put notify/key1");

    client.put("notify/key1", "value2").await.unwrap();
    info!("Modified notify/key1");

    client.put("notify/key2", "value3").await.unwrap();
    info!("Put notify/key2");

    client.delete("notify/key1").await.unwrap();
    info!("Deleted notify/key1");

    client.delete_range("notify/", "notify/~").await.unwrap();
    info!("Delete range notify/*");

    // Wait for listener to process notifications (with timeout)
    let _ = tokio::time::timeout(Duration::from_secs(5), listener).await;

    client.close().await.unwrap();
}