feat(tvix/build): add GRPCBuildServiceWrapper
This produces a gRPC BuildService server for anything implementing our BuildService trait. Change-Id: I59c690a432b5e1f59209fd67e2718cb8c935adf2 Reviewed-on: https://cl.tvl.fyi/c/depot/+/10628 Autosubmit: flokli <flokli@flokli.de> Reviewed-by: raitobezarius <tvl@lahfa.xyz> Tested-by: BuildkiteCI
This commit is contained in:
parent
4fb4fc263d
commit
c01ec8ee38
2 changed files with 39 additions and 0 deletions
35
tvix/build/src/proto/grpc_buildservice_wrapper.rs
Normal file
35
tvix/build/src/proto/grpc_buildservice_wrapper.rs
Normal file
|
@ -0,0 +1,35 @@
|
||||||
|
use crate::buildservice::BuildService;
|
||||||
|
use std::ops::Deref;
|
||||||
|
use tonic::async_trait;
|
||||||
|
|
||||||
|
use super::{Build, BuildRequest};
|
||||||
|
|
||||||
|
/// Implements the gRPC server trait ([self::build_service_server::BuildService]
|
||||||
|
/// for anything implementing [BuildService].
|
||||||
|
pub struct GRPCBuildServiceWrapper<BUILD> {
|
||||||
|
inner: BUILD,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<BUILD> GRPCBuildServiceWrapper<BUILD> {
|
||||||
|
pub fn new(build_service: BUILD) -> Self {
|
||||||
|
Self {
|
||||||
|
inner: build_service,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[async_trait]
|
||||||
|
impl<BUILD> crate::proto::build_service_server::BuildService for GRPCBuildServiceWrapper<BUILD>
|
||||||
|
where
|
||||||
|
BUILD: Deref<Target = dyn BuildService> + Send + Sync + 'static,
|
||||||
|
{
|
||||||
|
async fn do_build(
|
||||||
|
&self,
|
||||||
|
request: tonic::Request<BuildRequest>,
|
||||||
|
) -> Result<tonic::Response<Build>, tonic::Status> {
|
||||||
|
match self.inner.do_build(request.into_inner()).await {
|
||||||
|
Ok(resp) => Ok(tonic::Response::new(resp)),
|
||||||
|
Err(e) => Err(tonic::Status::internal(e.to_string())),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
|
@ -3,6 +3,10 @@ use std::path::{Path, PathBuf};
|
||||||
use itertools::Itertools;
|
use itertools::Itertools;
|
||||||
use tvix_castore::proto::{NamedNode, ValidateNodeError};
|
use tvix_castore::proto::{NamedNode, ValidateNodeError};
|
||||||
|
|
||||||
|
mod grpc_buildservice_wrapper;
|
||||||
|
|
||||||
|
pub use grpc_buildservice_wrapper::GRPCBuildServiceWrapper;
|
||||||
|
|
||||||
tonic::include_proto!("tvix.build.v1");
|
tonic::include_proto!("tvix.build.v1");
|
||||||
|
|
||||||
#[cfg(feature = "tonic-reflection")]
|
#[cfg(feature = "tonic-reflection")]
|
||||||
|
|
Loading…
Reference in a new issue