velo 0.1.0

High-level facade for Velo distributed systems
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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129

// SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
// SPDX-License-Identifier: Apache-2.0

//! Basic messaging integration test: two Velo instances exchange unary ping-pong.

use std::sync::Arc;
use std::time::Duration;

use bytes::Bytes;
use velo::*;
use velo_transports::tcp::{TcpTransport, TcpTransportBuilder};

fn new_transport() -> Arc<TcpTransport> {
    let listener = std::net::TcpListener::bind("127.0.0.1:0").unwrap();
    Arc::new(
        TcpTransportBuilder::new()
            .from_listener(listener)
            .unwrap()
            .build()
            .unwrap(),
    )
}

#[tokio::test(flavor = "multi_thread", worker_threads = 4)]
async fn two_instance_ping_pong() {
    // Create server
    let server = Velo::builder()
        .add_transport(new_transport())
        .build()
        .await
        .unwrap();

    tokio::time::sleep(Duration::from_millis(100)).await;

    // Register ping handler on server — echoes payload back
    let ping_handler = Handler::unary_handler("ping", |ctx| Ok(Some(ctx.payload))).build();
    server.register_handler(ping_handler).unwrap();

    // Create client
    let client = Velo::builder()
        .add_transport(new_transport())
        .build()
        .await
        .unwrap();

    tokio::time::sleep(Duration::from_millis(100)).await;

    // Register peers with each other
    client.register_peer(server.peer_info()).unwrap();
    server.register_peer(client.peer_info()).unwrap();

    tokio::time::sleep(Duration::from_millis(200)).await;

    // Send ping from client to server
    let payload = Bytes::from_static(b"hello");
    let response: Bytes = client
        .unary("ping")
        .unwrap()
        .raw_payload(payload.clone())
        .instance(server.instance_id())
        .send()
        .await
        .unwrap();

    assert_eq!(response, payload, "Server should echo payload back");
}

#[tokio::test(flavor = "multi_thread", worker_threads = 4)]
async fn typed_unary_json() {
    use serde::{Deserialize, Serialize};

    #[derive(Serialize, Deserialize, Debug, PartialEq)]
    struct AddRequest {
        a: i32,
        b: i32,
    }

    #[derive(Serialize, Deserialize, Debug, PartialEq)]
    struct AddResponse {
        result: i32,
    }

    // Create server
    let server = Velo::builder()
        .add_transport(new_transport())
        .build()
        .await
        .unwrap();

    tokio::time::sleep(Duration::from_millis(100)).await;

    // Register typed add handler (Handler::typed_unary, field is `input`)
    let handler = Handler::typed_unary("add", |ctx: TypedContext<AddRequest>| {
        Ok(AddResponse {
            result: ctx.input.a + ctx.input.b,
        })
    })
    .build();
    server.register_handler(handler).unwrap();

    // Create client
    let client = Velo::builder()
        .add_transport(new_transport())
        .build()
        .await
        .unwrap();

    tokio::time::sleep(Duration::from_millis(100)).await;

    // Connect
    client.register_peer(server.peer_info()).unwrap();
    server.register_peer(client.peer_info()).unwrap();

    tokio::time::sleep(Duration::from_millis(200)).await;

    // Send typed unary request (payload method serializes, send returns TypedUnaryResult)
    let request = AddRequest { a: 17, b: 25 };
    let response: AddResponse = client
        .typed_unary("add")
        .unwrap()
        .payload(&request)
        .unwrap()
        .instance(server.instance_id())
        .send()
        .await
        .unwrap();

    assert_eq!(response.result, 42);
}