async-rdma 0.2.0

A rust async wrapper for RDMA ibvers lib
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

//! This demo shows how to establish a connection between server and client
//! and the usage of rdma `read`, `write` and `send&recv` APIs.
//!
//! You can try this example by running:
//!
//!     cargo run --example server
//!
//! And then start client in another terminal by running:
//!
//!     cargo run --example client

use async_rdma::{LocalMrWriteAccess, Rdma};
use std::alloc::Layout;
use tracing::debug;

async fn send_lmr_to_server(rdma: &Rdma) {
    let mut lmr = rdma.alloc_local_mr(Layout::new::<char>()).unwrap();
    unsafe { *(lmr.as_mut_ptr() as *mut char) = 'h' };
    rdma.send_local_mr(lmr).await.unwrap();
}

async fn request_then_write(rdma: &Rdma) {
    let mut rmr = rdma.request_remote_mr(Layout::new::<char>()).await.unwrap();
    let mut lmr = rdma.alloc_local_mr(Layout::new::<char>()).unwrap();
    unsafe { *(lmr.as_mut_ptr() as *mut char) = 'e' };
    rdma.write(&lmr, &mut rmr).await.unwrap();
    rdma.send_remote_mr(rmr).await.unwrap();
}

async fn send_data_to_server(rdma: &Rdma) {
    let mut lmr = rdma.alloc_local_mr(Layout::new::<char>()).unwrap();
    unsafe { *(lmr.as_mut_ptr() as *mut char) = 'y' };
    rdma.send(&lmr).await.unwrap();
}

#[tokio::main]
async fn main() {
    tracing_subscriber::fmt::init();
    debug!("client start");
    let rdma = Rdma::connect("127.0.0.1:5555", 1, 1, 512).await.unwrap();
    send_lmr_to_server(&rdma).await;
    request_then_write(&rdma).await;
    send_data_to_server(&rdma).await;
    debug!("client done");
}