use crate::identifiers::UniqueServerId;
use crate::payload::number_of_elements;
use crate::port::client::ClientSharedState;
use crate::port::details::chunk::Chunk;
use crate::port::details::chunk_details::ChunkDetails;
use crate::service;
use crate::service::marker::Flatbuffer;
use crate::unique_id_generator::UniqueId;
use core::fmt::Debug;
use core::marker::PhantomData;
use core::ops::Deref;
use flatbuffers::InvalidFlatbuffer;
use iceoryx2_bb_elementary_traits::iceoryx_send::IceoryxSend;
use iceoryx2_bb_elementary_traits::zero_copy_send::ZeroCopySend;
use iceoryx2_bb_flatbuffers::FlatbufferError;
use iceoryx2_cal::arc_sync_policy::ArcSyncPolicy;
use iceoryx2_cal::zero_copy_connection::ChannelId;
pub struct Response<
Service: crate::service::Service,
ResponsePayload: Debug + IceoryxSend + ?Sized,
ResponseHeader: Debug + ZeroCopySend,
> {
pub(crate) chunk: Chunk,
pub(crate) client_shared_state: Service::ArcThreadSafetyPolicy<ClientSharedState<Service>>,
pub(crate) details: ChunkDetails,
pub(crate) channel_id: ChannelId,
pub(crate) _response_payload: PhantomData<ResponsePayload>,
pub(crate) _response_header: PhantomData<ResponseHeader>,
}
unsafe impl<
Service: crate::service::Service,
ResponsePayload: Debug + IceoryxSend + ?Sized,
ResponseHeader: Debug + ZeroCopySend,
> Send for Response<Service, ResponsePayload, ResponseHeader>
where
Service::ArcThreadSafetyPolicy<ClientSharedState<Service>>: Send + Sync,
{
}
impl<
Service: crate::service::Service,
ResponsePayload: Debug + IceoryxSend + ?Sized,
ResponseHeader: Debug + ZeroCopySend,
> Drop for Response<Service, ResponsePayload, ResponseHeader>
{
fn drop(&mut self) {
self.client_shared_state
.lock()
.response_receiver
.release_offset(&self.details, self.channel_id);
}
}
impl<
Service: crate::service::Service,
ResponsePayload: Debug + IceoryxSend + ?Sized,
ResponseHeader: Debug + ZeroCopySend,
> Debug for Response<Service, ResponsePayload, ResponseHeader>
{
fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {
write!(
f,
"Response<{}, {}, {}> {{ chunk: {:?} }}",
core::any::type_name::<Service>(),
core::any::type_name::<ResponsePayload>(),
core::any::type_name::<ResponseHeader>(),
self.chunk
)
}
}
impl<
Service: crate::service::Service,
ResponsePayload: Debug + IceoryxSend + ZeroCopySend,
ResponseHeader: Debug + ZeroCopySend,
> Deref for Response<Service, ResponsePayload, ResponseHeader>
{
type Target = ResponsePayload;
fn deref(&self) -> &Self::Target {
unsafe { &*self.chunk.payload_ptr().cast() }
}
}
impl<
Service: crate::service::Service,
ResponsePayload: Debug + IceoryxSend + ZeroCopySend,
ResponseHeader: Debug + ZeroCopySend,
> Deref for Response<Service, [ResponsePayload], ResponseHeader>
{
type Target = [ResponsePayload];
fn deref(&self) -> &Self::Target {
let payload_size = self
.client_shared_state
.lock()
.response_receiver
.payload_size();
unsafe {
&*core::ptr::slice_from_raw_parts(
self.chunk.payload_ptr().cast(),
number_of_elements::<ResponsePayload, _>(self.header(), payload_size),
)
}
}
}
impl<
Service: crate::service::Service,
ResponsePayload: Debug + IceoryxSend + ?Sized,
ResponseHeader: Debug + ZeroCopySend,
> Response<Service, ResponsePayload, ResponseHeader>
{
pub fn header(&self) -> &service::header::request_response::ResponseHeader {
unsafe { &*self.chunk.header_ptr().cast() }
}
pub fn user_header(&self) -> &ResponseHeader {
unsafe { &*self.chunk.user_header_ptr().cast() }
}
pub fn origin(&self) -> UniqueServerId {
UniqueServerId(unsafe { UniqueId::from_raw_id(self.details.origin) })
}
}
impl<
Service: crate::service::Service,
ResponsePayload: Debug + IceoryxSend + ZeroCopySend,
ResponseHeader: Debug + ZeroCopySend,
> Response<Service, ResponsePayload, ResponseHeader>
{
pub fn payload(&self) -> &ResponsePayload {
self.deref()
}
}
impl<
Service: crate::service::Service,
ResponsePayload: Debug + IceoryxSend + ZeroCopySend,
ResponseHeader: Debug + ZeroCopySend,
> Response<Service, [ResponsePayload], ResponseHeader>
{
pub fn payload(&self) -> &[ResponsePayload] {
self.deref()
}
}
impl<Service: crate::service::Service, ResponsePayload, ResponseHeader: Debug + ZeroCopySend>
Response<Service, Flatbuffer<ResponsePayload>, ResponseHeader>
{
pub fn payload_bytes(&self) -> &[u8] {
let payload_offset = self.header().payload_offset as usize;
let payload_ptr = self.chunk.payload_ptr();
let payload_len = self.header().number_of_elements as usize;
unsafe {
core::slice::from_raw_parts(
payload_ptr.add(payload_offset),
payload_len - payload_offset,
)
}
}
pub fn payload_root<'a>(
&'a self,
) -> Result<ResponsePayload::Inner, FlatbufferError<InvalidFlatbuffer>>
where
ResponsePayload: flatbuffers::Follow<'a> + flatbuffers::Verifiable,
{
Ok(flatbuffers::root::<ResponsePayload>(self.payload_bytes())?)
}
}