raven
a8755bd3de
Documentation: - Add detailed README files for all services (auth, character, database, launcher, packet, utils, world) - Create API documentation for the database service with detailed endpoint specifications - Document database schema and relationships - Add service architecture overviews and configuration instructions Unit Tests: - Implement comprehensive test suite for database repositories (user, character, session) - Add gRPC service tests for database interactions - Create tests for packet service components (bufferpool, connection, packets) - Add utility service tests (health check, logging, load balancer, redis cache, service discovery) - Implement auth service user tests - Add character service tests Code Structure: - Reorganize test files into a more consistent structure - Create a dedicated tests crate for integration testing - Add test helpers and mock implementations for easier testing
50 lines
2.0 KiB
Rust
50 lines
2.0 KiB
Rust
use crate::database::{
|
|
user_service_client::UserServiceClient, GetUserByEmailRequest, GetUserByUsernameRequest, GetUserRequest,
|
|
GetUserResponse,
|
|
};
|
|
use async_trait::async_trait;
|
|
use std::error::Error;
|
|
use tonic::transport::Channel;
|
|
|
|
#[async_trait]
|
|
pub trait DatabaseClientTrait: Sized {
|
|
async fn connect(endpoint: &str) -> Result<Self, Box<dyn std::error::Error>>;
|
|
async fn get_user_by_userid(&mut self, user_id: i32) -> Result<GetUserResponse, Box<dyn std::error::Error>>;
|
|
async fn get_user_by_username(&mut self, user_id: &str) -> Result<GetUserResponse, Box<dyn std::error::Error>>;
|
|
async fn get_user_by_email(&mut self, email: &str) -> Result<GetUserResponse, Box<dyn std::error::Error>>;
|
|
}
|
|
#[derive(Clone)]
|
|
pub struct DatabaseClient {
|
|
client: UserServiceClient<Channel>,
|
|
}
|
|
|
|
#[async_trait]
|
|
impl DatabaseClientTrait for DatabaseClient {
|
|
async fn connect(endpoint: &str) -> Result<Self, Box<dyn std::error::Error>> {
|
|
let client = UserServiceClient::connect(endpoint.to_string()).await?;
|
|
Ok(Self { client })
|
|
}
|
|
|
|
async fn get_user_by_userid(&mut self, user_id: i32) -> Result<GetUserResponse, Box<dyn std::error::Error>> {
|
|
let request = tonic::Request::new(GetUserRequest { user_id });
|
|
let response = self.client.get_user(request).await?;
|
|
Ok(response.into_inner())
|
|
}
|
|
|
|
async fn get_user_by_username(&mut self, username: &str) -> Result<GetUserResponse, Box<dyn std::error::Error>> {
|
|
let request = tonic::Request::new(GetUserByUsernameRequest {
|
|
username: username.to_string(),
|
|
});
|
|
let response = self.client.get_user_by_username(request).await?;
|
|
Ok(response.into_inner())
|
|
}
|
|
|
|
async fn get_user_by_email(&mut self, email: &str) -> Result<GetUserResponse, Box<dyn Error>> {
|
|
let request = tonic::Request::new(GetUserByEmailRequest {
|
|
email: email.to_string(),
|
|
});
|
|
let response = self.client.get_user_by_email(request).await?;
|
|
Ok(response.into_inner())
|
|
}
|
|
}
|