more cleanup

This commit is contained in:
nora 2022-02-26 23:24:08 +01:00
parent 6d944e1265
commit de027d9f5a
9 changed files with 46 additions and 14 deletions

View file

@ -162,7 +162,7 @@ impl Connection {
ensure_conn(mechanism == "PLAIN")?;
ensure_conn(locale == "en_US")?;
let plain_user = sasl::parse_sasl_plain_response(&response)?;
info!(username = %plain_user.authentication_identity, "SASL Authentication successful")
info!(username = %plain_user.authentication_identity, "SASL Authentication successful");
} else {
return Err(ConException::Todo.into());
}
@ -257,7 +257,7 @@ impl Connection {
Method::ChannelClose { .. } => self.channel_close(frame.channel, method).await?,
Method::BasicPublish { .. } => match self.channels.get_mut(&frame.channel) {
Some(channel) => {
channel.status = ChannelStatus::NeedHeader(BASIC_CLASS_ID, Box::new(method))
channel.status = ChannelStatus::NeedHeader(BASIC_CLASS_ID, Box::new(method));
}
None => return Err(ConException::Todo.into()),
},

View file

@ -138,7 +138,7 @@ impl ContentHeader {
pub async fn write_frame<W>(frame: &Frame, mut w: W) -> Result<()>
where
W: AsyncWriteExt + Unpin,
W: AsyncWriteExt + Unpin + Send,
{
trace!(?frame, "Sending frame");
@ -154,7 +154,7 @@ where
pub async fn read_frame<R>(r: &mut R, max_frame_size: usize) -> Result<Frame>
where
R: AsyncReadExt + Unpin,
R: AsyncReadExt + Unpin + Send,
{
let kind = r.read_u8().await.context("read type")?;
let channel = r.read_u16().await.context("read channel")?;

View file

@ -2,7 +2,6 @@ use crate::error::TransError;
use amqp_core::error::ConException;
use amqp_core::methods::{FieldValue, Method, Table};
use rand::Rng;
use std::collections::HashMap;
mod generated;
pub mod parse_helper;
@ -65,7 +64,9 @@ rand_random_method!(bool, u8, i8, u16, i16, u32, i32, u64, i64, f32, f64);
impl<R: Rng> RandomMethod<R> for Table {
fn random(rng: &mut R) -> Self {
let len = rng.gen_range(0..3);
HashMap::from_iter((0..len).map(|_| (String::random(rng), FieldValue::random(rng))))
(0..len)
.map(|_| (String::random(rng), FieldValue::random(rng)))
.collect()
}
}

View file

@ -12,7 +12,6 @@ use nom::multi::{count, many0};
use nom::number::complete::{f32, f64, i16, i32, i64, i8, u16, u32, u64, u8};
use nom::number::Endianness::Big;
use nom::Err;
use std::collections::HashMap;
impl<T> nom::error::ParseError<T> for TransError {
fn from_error_kind(_input: T, _kind: ErrorKind) -> Self {
@ -37,7 +36,7 @@ pub fn fail_err<S: Into<String>>(msg: S) -> impl FnOnce(Err<TransError>) -> Err<
}
_ => vec![msg],
},
_ => vec![msg],
Err::Incomplete(_) => vec![msg],
};
Err::Failure(ConException::SyntaxError(stack).into())
}
@ -133,7 +132,7 @@ pub fn table(input: &[u8]) -> IResult<'_, Table> {
));
}
let table = HashMap::from_iter(values.into_iter());
let table = values.into_iter().collect();
Ok((rest_input, table))
}

View file

@ -1,5 +1,6 @@
use crate::frame::{ChannelNum, FrameType};
use crate::frame::FrameType;
use crate::{frame, methods};
use amqp_core::connection::ChannelNum;
use amqp_core::methods::{FieldValue, Method};
use std::collections::HashMap;