46 lines
1.6 KiB
Rust
46 lines
1.6 KiB
Rust
use crate::session::{session_service_client::SessionServiceClient, GetSessionRequest, GetSessionResponse, RefreshSessionRequest, RefreshSessionResponse};
|
|
use async_trait::async_trait;
|
|
use chrono::{DateTime, Utc};
|
|
use std::error::Error;
|
|
use tonic::transport::Channel;
|
|
|
|
#[async_trait]
|
|
pub trait SessionClientTrait: Sized {
|
|
async fn connect(endpoint: &str) -> Result<Self, Box<dyn std::error::Error>>;
|
|
async fn get_session(
|
|
&mut self,
|
|
session_id: String,
|
|
) -> Result<GetSessionResponse, Box<dyn std::error::Error>>;
|
|
async fn refresh_session(
|
|
&mut self,
|
|
session_id: String,
|
|
) -> Result<RefreshSessionResponse, Box<dyn std::error::Error>>;
|
|
}
|
|
#[derive(Clone)]
|
|
pub struct SessionClient {
|
|
client: SessionServiceClient<Channel>,
|
|
}
|
|
|
|
#[async_trait]
|
|
impl SessionClientTrait for SessionClient {
|
|
async fn connect(endpoint: &str) -> Result<Self, Box<dyn std::error::Error>> {
|
|
let client = SessionServiceClient::connect(endpoint.to_string()).await?;
|
|
Ok(Self { client })
|
|
}
|
|
|
|
async fn get_session(&mut self, session_id: String) -> Result<GetSessionResponse, Box<dyn Error>> {
|
|
let request = tonic::Request::new(GetSessionRequest {
|
|
session_id,
|
|
});
|
|
let response = self.client.get_session(request).await?;
|
|
Ok(response.into_inner())
|
|
}
|
|
async fn refresh_session(&mut self, session_id: String) -> Result<RefreshSessionResponse, Box<dyn Error>> {
|
|
let request = tonic::Request::new(RefreshSessionRequest {
|
|
session_id,
|
|
});
|
|
let response = self.client.refresh_session(request).await?;
|
|
Ok(response.into_inner())
|
|
}
|
|
}
|