zhenyi
66afd932ed
- Add FindCommit, ListCommitsByOid, CommitIsAncestor RPCs to CommitService - Add CheckObjectsExist, CommitsByMessage, GetCommitStats RPCs to CommitService - Add LastCommitForPath, CountCommits, CountDivergingCommits RPCs to CommitService - Add RawDiff, RawPatch, FindChangedPaths RPCs to DiffService - Add FindMergeBase, WriteRef, SearchFilesByContent RPCs to RepositoryService - Add SearchFilesByName, ObjectsSize, RepositorySize RPCs to RepositoryService - Add FindLicense, OptimizeRepository, GetRawChanges RPCs to RepositoryService - Add FetchRemote, CreateRepositoryFromURL RPCs to RepositoryService - Implement server handlers for all new RPC methods - Add new modules for commit counting, finding, and querying features - Add new modules for diff changed paths and raw operations - Add new modules for refs and remote operations - Remove unnecessary comments from various source files - Update proto definitions with new message types and service methods
86 lines
3.6 KiB
Rust
86 lines
3.6 KiB
Rust
use crate::pb::*;
|
|
use crate::pb::ref_service_server::RefService;
|
|
|
|
use super::GitksService;
|
|
|
|
#[tonic::async_trait]
|
|
impl RefService for GitksService {
|
|
async fn find_default_branch_name(
|
|
&self,
|
|
request: tonic::Request<FindDefaultBranchNameRequest>,
|
|
) -> Result<tonic::Response<FindDefaultBranchNameResponse>, tonic::Status> {
|
|
let m = crate::metrics::RequestMetrics::new("gitks.RefService/FindDefaultBranchName");
|
|
let inner = request.into_inner();
|
|
let _rate = self.acquire_rate_limit(inner.repository.as_ref()).await?;
|
|
let gb = self.resolve(inner.repository.as_ref())?;
|
|
let resp = gb.find_default_branch_name().map_err(super::into_status)?;
|
|
m.record("ok");
|
|
Ok(tonic::Response::new(resp))
|
|
}
|
|
|
|
async fn ref_exists(
|
|
&self,
|
|
request: tonic::Request<RefExistsRequest>,
|
|
) -> Result<tonic::Response<RefExistsResponse>, tonic::Status> {
|
|
let m = crate::metrics::RequestMetrics::new("gitks.RefService/RefExists");
|
|
let inner = request.into_inner();
|
|
let _rate = self.acquire_rate_limit(inner.repository.as_ref()).await?;
|
|
let gb = self.resolve(inner.repository.as_ref())?;
|
|
let resp = gb.ref_exists(inner).map_err(super::into_status)?;
|
|
m.record("ok");
|
|
Ok(tonic::Response::new(resp))
|
|
}
|
|
|
|
async fn update_references(
|
|
&self,
|
|
request: tonic::Request<UpdateReferencesRequest>,
|
|
) -> Result<tonic::Response<UpdateReferencesResponse>, tonic::Status> {
|
|
let m = crate::metrics::RequestMetrics::new("gitks.RefService/UpdateReferences");
|
|
let inner = request.into_inner();
|
|
let _rate = self.acquire_rate_limit(inner.repository.as_ref()).await?;
|
|
let gb = self.resolve(inner.repository.as_ref())?;
|
|
let resp = gb.update_references(inner).map_err(super::into_status)?;
|
|
m.record("ok");
|
|
Ok(tonic::Response::new(resp))
|
|
}
|
|
|
|
async fn delete_refs(
|
|
&self,
|
|
request: tonic::Request<DeleteRefsRequest>,
|
|
) -> Result<tonic::Response<DeleteRefsResponse>, tonic::Status> {
|
|
let m = crate::metrics::RequestMetrics::new("gitks.RefService/DeleteRefs");
|
|
let inner = request.into_inner();
|
|
let _rate = self.acquire_rate_limit(inner.repository.as_ref()).await?;
|
|
let gb = self.resolve(inner.repository.as_ref())?;
|
|
let resp = gb.delete_refs(inner).map_err(super::into_status)?;
|
|
m.record("ok");
|
|
Ok(tonic::Response::new(resp))
|
|
}
|
|
|
|
async fn find_refs_by_oid(
|
|
&self,
|
|
request: tonic::Request<FindRefsByOidRequest>,
|
|
) -> Result<tonic::Response<FindRefsByOidResponse>, tonic::Status> {
|
|
let m = crate::metrics::RequestMetrics::new("gitks.RefService/FindRefsByOID");
|
|
let inner = request.into_inner();
|
|
let _rate = self.acquire_rate_limit(inner.repository.as_ref()).await?;
|
|
let gb = self.resolve(inner.repository.as_ref())?;
|
|
let resp = gb.find_refs_by_oid(inner).map_err(super::into_status)?;
|
|
m.record("ok");
|
|
Ok(tonic::Response::new(resp))
|
|
}
|
|
|
|
async fn list_refs(
|
|
&self,
|
|
request: tonic::Request<ListRefsRequest>,
|
|
) -> Result<tonic::Response<ListRefsResponse>, tonic::Status> {
|
|
let m = crate::metrics::RequestMetrics::new("gitks.RefService/ListRefs");
|
|
let inner = request.into_inner();
|
|
let _rate = self.acquire_rate_limit(inner.repository.as_ref()).await?;
|
|
let gb = self.resolve(inner.repository.as_ref())?;
|
|
let resp = gb.list_all_refs(inner).map_err(super::into_status)?;
|
|
m.record("ok");
|
|
Ok(tonic::Response::new(resp))
|
|
}
|
|
}
|