zhenyi
a40da90ef9
- Reformatted build script with proper indentation and line breaks - Added tonic-health dependency to Cargo.toml and updated lock file - Improved error handling in disk cache with concurrent deletion checks - Refactored conditional chains using && and let expressions - Reformatted struct initialization and function parameter lists - Added proper spacing and alignment in language stats processing - Improved assertion formatting in test cases - Reorganized import statements and code layout in multiple files - Updated metrics functions with better parameter handling and formatting
63 lines
2.8 KiB
Rust
63 lines
2.8 KiB
Rust
use crate::pb::remote_service_server::RemoteService;
|
|
use crate::pb::*;
|
|
use crate::remote::find_remote::{find_remote_repository, find_remote_root_ref};
|
|
|
|
use super::GitksService;
|
|
|
|
#[tonic::async_trait]
|
|
impl RemoteService for GitksService {
|
|
async fn find_remote_repository(
|
|
&self,
|
|
request: tonic::Request<FindRemoteRepositoryRequest>,
|
|
) -> Result<tonic::Response<FindRemoteRepositoryResponse>, tonic::Status> {
|
|
let m = crate::metrics::RequestMetrics::new("gitks.RemoteService/FindRemoteRepository");
|
|
let inner = request.into_inner();
|
|
let span =
|
|
tracing::info_span!("remote.find_remote_repository", remote_url = %inner.remote_url);
|
|
let _enter = span.enter();
|
|
tracing::info!(remote_url = %inner.remote_url, "find_remote_repository");
|
|
let resp = find_remote_repository(inner).map_err(super::into_status)?;
|
|
tracing::info!(
|
|
refs_count = resp.refs.len(),
|
|
exists = resp.exists,
|
|
"find_remote_repository done"
|
|
);
|
|
m.record("ok");
|
|
Ok(tonic::Response::new(resp))
|
|
}
|
|
|
|
async fn find_remote_root_ref(
|
|
&self,
|
|
request: tonic::Request<FindRemoteRootRefRequest>,
|
|
) -> Result<tonic::Response<FindRemoteRootRefResponse>, tonic::Status> {
|
|
let m = crate::metrics::RequestMetrics::new("gitks.RemoteService/FindRemoteRootRef");
|
|
let inner = request.into_inner();
|
|
let span =
|
|
tracing::info_span!("remote.find_remote_root_ref", remote_url = %inner.remote_url);
|
|
let _enter = span.enter();
|
|
tracing::info!(remote_url = %inner.remote_url, "find_remote_root_ref");
|
|
let resp = find_remote_root_ref(inner).map_err(super::into_status)?;
|
|
tracing::info!(ref_name = %resp.ref_name, target_oid = %resp.target_oid, "find_remote_root_ref done");
|
|
m.record("ok");
|
|
Ok(tonic::Response::new(resp))
|
|
}
|
|
|
|
async fn update_remote_mirror(
|
|
&self,
|
|
request: tonic::Request<UpdateRemoteMirrorRequest>,
|
|
) -> Result<tonic::Response<UpdateRemoteMirrorResponse>, tonic::Status> {
|
|
let m = crate::metrics::RequestMetrics::new("gitks.RemoteService/UpdateRemoteMirror");
|
|
let inner = request.into_inner();
|
|
let repo = self.repo_label(inner.repository.as_ref());
|
|
let span = tracing::info_span!("remote.update_remote_mirror", %repo);
|
|
let _enter = span.enter();
|
|
tracing::info!(repo = %repo, remote_url = %inner.remote_url, force = inner.force, prune = inner.prune, "update_remote_mirror");
|
|
let _rate = self.acquire_rate_limit(inner.repository.as_ref()).await?;
|
|
let gb = self.resolve(inner.repository.as_ref())?;
|
|
let resp = gb.update_remote_mirror(inner).map_err(super::into_status)?;
|
|
tracing::info!(ok = resp.ok, "update_remote_mirror done");
|
|
m.record("ok");
|
|
Ok(tonic::Response::new(resp))
|
|
}
|
|
}
|