Bogdan-Ștefan Neacşu
fbdf31b879
* Replace client address with service address * Copy over the server parts of the statistics server * Separate API in module * Rename struct * Add insertion endpoint * Remove unused code from network requester * Box big rocket error * Remove the feature-specific code * Construct http req * Clippy + Cargo.lock update * Re-added needed sqlx feature * Wrap http req in ordered msg * Add http headers * Move common api functionality into the separate crate * Make stats server address configurable, especially for testing * Update changelog * Fix clippy * Help command update
84 lines
2.4 KiB
Rust
84 lines
2.4 KiB
Rust
// Copyright 2022 - Nym Technologies SA <contact@nymtech.net>
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
use crate::request::{Request, RequestError};
|
|
use crate::response::{Response, ResponseError};
|
|
|
|
#[derive(Debug)]
|
|
pub enum MessageError {
|
|
Request(RequestError),
|
|
Response(ResponseError),
|
|
NoData,
|
|
UnknownMessageType,
|
|
}
|
|
|
|
impl std::fmt::Display for MessageError {
|
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
match self {
|
|
MessageError::Request(r) => write!(f, "{}", r),
|
|
MessageError::Response(r) => write!(f, "{:?}", r),
|
|
MessageError::NoData => write!(f, "no data provided"),
|
|
MessageError::UnknownMessageType => write!(f, "unknown message type received"),
|
|
}
|
|
}
|
|
}
|
|
|
|
pub enum Message {
|
|
Request(Request),
|
|
Response(Response),
|
|
}
|
|
|
|
impl Message {
|
|
const REQUEST_FLAG: u8 = 0;
|
|
const RESPONSE_FLAG: u8 = 1;
|
|
|
|
pub fn conn_id(&self) -> u64 {
|
|
match self {
|
|
Message::Request(req) => match req {
|
|
Request::Connect(c) => c.conn_id,
|
|
Request::Send(conn_id, _, _) => *conn_id,
|
|
},
|
|
Message::Response(resp) => resp.connection_id,
|
|
}
|
|
}
|
|
|
|
pub fn size(&self) -> usize {
|
|
match self {
|
|
Message::Request(req) => match req {
|
|
Request::Connect(_) => 0,
|
|
Request::Send(_, data, _) => data.len(),
|
|
},
|
|
Message::Response(resp) => resp.data.len(),
|
|
}
|
|
}
|
|
|
|
pub fn try_from_bytes(b: &[u8]) -> Result<Message, MessageError> {
|
|
if b.is_empty() {
|
|
return Err(MessageError::NoData);
|
|
}
|
|
|
|
if b[0] == Self::REQUEST_FLAG {
|
|
Request::try_from_bytes(&b[1..])
|
|
.map(Message::Request)
|
|
.map_err(MessageError::Request)
|
|
} else if b[0] == Self::RESPONSE_FLAG {
|
|
Response::try_from_bytes(&b[1..])
|
|
.map(Message::Response)
|
|
.map_err(MessageError::Response)
|
|
} else {
|
|
Err(MessageError::UnknownMessageType)
|
|
}
|
|
}
|
|
|
|
pub fn into_bytes(self) -> Vec<u8> {
|
|
match self {
|
|
Self::Request(r) => std::iter::once(Self::REQUEST_FLAG)
|
|
.chain(r.into_bytes().iter().cloned())
|
|
.collect(),
|
|
Self::Response(r) => std::iter::once(Self::RESPONSE_FLAG)
|
|
.chain(r.into_bytes().iter().cloned())
|
|
.collect(),
|
|
}
|
|
}
|
|
}
|