tvix/build/src/buildservice/grpc.rs


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

use tonic::{async_trait, transport::Channel};

use crate::buildservice::BuildRequest;
use crate::proto::{self, build_service_client::BuildServiceClient};

use super::BuildService;

pub struct GRPCBuildService {
    client: BuildServiceClient<Channel>,
}

impl GRPCBuildService {
    #[allow(dead_code)]
    pub fn from_client(client: BuildServiceClient<Channel>) -> Self {
        Self { client }
    }
}

#[async_trait]
impl BuildService for GRPCBuildService {
    async fn do_build(&self, request: BuildRequest) -> std::io::Result<proto::Build> {
        let mut client = self.client.clone();
        client
            .do_build(Into::<proto::BuildRequest>::into(request))
            .await
            .map(|resp| resp.into_inner())
            .map_err(std::io::Error::other)
    }
}