fizzbee-mbt 0.1.1

FizzBee Model Based Testing (MBT) Rust binding
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

use crate::plugin_service::FizzBeeServiceImpl;
use crate::pb::fizz_bee_mbt_plugin_service_server::FizzBeeMbtPluginServiceServer;
use crate::error::MbtError;
use crate::traits::{Model, DispatchModel};
use tokio::runtime::Builder;
use tonic::transport::Server;

// Imports for UDS
use tokio::net::UnixListener;
use tokio_stream::wrappers::UnixListenerStream;
use std::path::Path;
use std::fs;

// The function signature is now generic and accepts the model and dispatcher
pub fn start_grpc_server<D>(
    dispatcher: D,
) -> Result<(), MbtError>
where
    D: Model + DispatchModel + Send + Sync + 'static,
{
    let rt = Builder::new_multi_thread()
        .enable_all()
        .build()
        .map_err(MbtError::from_err)?;

    rt.block_on(async {
        // --- UDS Configuration ---
        let path = Path::new("/tmp/fizzbee_mbt.sock");

        // IMPORTANT: Clean up the old socket file if it exists, otherwise bind() will fail.
        if path.exists() {
            fs::remove_file(path).map_err(MbtError::from_err)?;
        }

        // Bind to the Unix socket
        let uds = UnixListener::bind(path).map_err(MbtError::from_err)?;

        // Wrap the listener in a stream to use with tonic
        let incoming = UnixListenerStream::new(uds);

        println!("gRPC server listening on UDS socket: {}", path.display());
        // --- End UDS Configuration ---

        let service = FizzBeeServiceImpl::new(dispatcher);

        Server::builder()
            .add_service(FizzBeeMbtPluginServiceServer::new(service))
            // Use serve_with_incoming for Unix Domain Sockets
            .serve_with_incoming(incoming)
            .await
            .map_err(MbtError::from_err)
    })
}