Fixed ProtocolConnection::split_conn function

This commit is contained in:
Kyler 2024-05-31 19:23:20 -06:00
parent c63ff903cb
commit 4e9b60138e
1 changed files with 8 additions and 7 deletions

View File

@ -4,6 +4,7 @@ use std::error::Error;
use std::fmt; use std::fmt;
use tokio::net::tcp::{OwnedReadHalf, OwnedWriteHalf}; use tokio::net::tcp::{OwnedReadHalf, OwnedWriteHalf};
use tokio::net::TcpStream;
use tokio::io::{AsyncReadExt, AsyncWriteExt}; use tokio::io::{AsyncReadExt, AsyncWriteExt};
use serde::{Serialize, Deserialize}; use serde::{Serialize, Deserialize};
use async_trait::async_trait; use async_trait::async_trait;
@ -193,15 +194,15 @@ impl<'a> ProtocolConnection<'a> {
pub fn split_conn( pub fn split_conn(
&mut self &mut self
) -> (WriteHaftProtocolConnection, ReadHaftProtocolConnection) { ) -> Result<(WriteHaftProtocolConnection, ReadHaftProtocolConnection)> {
(WriteHaftProtocolConnection { Ok((WriteHaftProtocolConnection {
stream_write: self.stream_write, stream_write: &mut self.stream_write,
aes_encryption_key: self.aes_encryption_key, aes_encryption_key: self.aes_encryption_key.clone(),
}, },
ReadHaftProtocolConnection { ReadHaftProtocolConnection {
stream_read: self.stream_read, stream_read: &mut self.stream_read,
aes_encryption_key: self.aes_encryption_key, aes_encryption_key: self.aes_encryption_key.clone(),
}) }))
} }
} }