zhenyi
9a0c26e5f6
- Add voting mechanism with term tracking and vote persistence - Implement election triggering logic with majority vote counting - Add primary/replica role transition handling with state management - Integrate health check failure detection for automatic elections - Refactor actor messaging system for distributed coordination - Update repository registration to query cluster for existing primary - Add broadcast mechanism for role change notifications - Implement proper term comparison and duplicate request filtering - Upgrade dependency versions including tokio-util for async utilities - Optimize code formatting and line wrapping for improved readability - Remove redundant blank lines and improve code structure consistency - Enhance error logging and trace information for debugging purposes
86 lines
3.6 KiB
Rust
86 lines
3.6 KiB
Rust
use crate::pb::ref_service_server::RefService;
|
|
use crate::pb::*;
|
|
|
|
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))
|
|
}
|
|
}
|