zhenyi
5f4e9bdfa7
- Change tokio-stream dependency to include net feature for TcpListenerStream - Move TCP listener binding before etcd registry initialization in main function - Pass pre-bound TcpListener to gRPC server instead of just SocketAddr - Update gRPC server to use serve_with_incoming with TcpListenerStream - Prevent peers from attempting connections before gRPC server is ready - Ensure proper error handling for TCP binding failures during startup
69 lines
3.2 KiB
Rust
69 lines
3.2 KiB
Rust
pub mod auth;
|
|
pub mod channel;
|
|
pub mod channel_settings;
|
|
pub mod member;
|
|
pub mod permission;
|
|
|
|
use std::net::SocketAddr;
|
|
|
|
use crate::pb::core::token_service_server::TokenServiceServer;
|
|
use crate::pb::im::channel_audit_service_server::ChannelAuditServiceServer;
|
|
use crate::pb::im::channel_invitation_service_server::ChannelInvitationServiceServer;
|
|
use crate::pb::im::channel_repo_link_service_server::ChannelRepoLinkServiceServer;
|
|
use crate::pb::im::channel_role_service_server::ChannelRoleServiceServer;
|
|
use crate::pb::im::channel_service_server::ChannelServiceServer;
|
|
use crate::pb::im::channel_slash_command_service_server::ChannelSlashCommandServiceServer;
|
|
use crate::pb::im::channel_webhook_service_server::ChannelWebhookServiceServer;
|
|
use crate::pb::im::custom_emoji_service_server::CustomEmojiServiceServer;
|
|
use crate::pb::im::forum_tag_service_server::ForumTagServiceServer;
|
|
use crate::pb::im::im_integration_service_server::ImIntegrationServiceServer;
|
|
use crate::pb::im::member_service_server::MemberServiceServer;
|
|
use crate::pb::im::permission_service_server::PermissionServiceServer;
|
|
use crate::pb::im::stage_service_server::StageServiceServer;
|
|
use crate::pb::im::voice_service_server::VoiceServiceServer;
|
|
use tonic_health::ServingStatus;
|
|
|
|
use crate::service::AppService;
|
|
|
|
pub async fn start_grpc_server(
|
|
addr: SocketAddr,
|
|
listener: tokio::net::TcpListener,
|
|
service: AppService,
|
|
) -> Result<(), Box<dyn std::error::Error>> {
|
|
let token_svc = auth::TokenGrpcService::new(service.internal_auth.clone());
|
|
let channel_svc = channel::ChannelGrpcService::new(service.clone());
|
|
let member_svc = member::MemberGrpcService::new(service.clone());
|
|
let permission_svc = permission::PermissionGrpcService::new(service.clone());
|
|
|
|
let cs = channel_settings::ChannelSettingsServices::new(service);
|
|
|
|
let (health_reporter, health_service) = tonic_health::server::health_reporter();
|
|
health_reporter
|
|
.set_service_status("", ServingStatus::Serving)
|
|
.await;
|
|
|
|
tracing::info!(%addr, "gRPC server listening");
|
|
|
|
tonic::transport::Server::builder()
|
|
.add_service(health_service)
|
|
.add_service(TokenServiceServer::new(token_svc))
|
|
.add_service(ChannelServiceServer::new(channel_svc))
|
|
.add_service(MemberServiceServer::new(member_svc))
|
|
.add_service(PermissionServiceServer::new(permission_svc))
|
|
.add_service(ChannelRoleServiceServer::new(cs.channel_role))
|
|
.add_service(ChannelInvitationServiceServer::new(cs.channel_invitation))
|
|
.add_service(ChannelWebhookServiceServer::new(cs.channel_webhook))
|
|
.add_service(ChannelSlashCommandServiceServer::new(cs.channel_slash_command))
|
|
.add_service(ChannelRepoLinkServiceServer::new(cs.channel_repo_link))
|
|
.add_service(ImIntegrationServiceServer::new(cs.im_integration))
|
|
.add_service(CustomEmojiServiceServer::new(cs.custom_emoji))
|
|
.add_service(ForumTagServiceServer::new(cs.forum_tag))
|
|
.add_service(VoiceServiceServer::new(cs.voice))
|
|
.add_service(StageServiceServer::new(cs.stage))
|
|
.add_service(ChannelAuditServiceServer::new(cs.channel_audit))
|
|
.serve_with_incoming(tokio_stream::wrappers::TcpListenerStream::new(listener))
|
|
.await?;
|
|
|
|
Ok(())
|
|
}
|