Fixed ProtocolConnection::split_conn function
This commit is contained in:
parent
c63ff903cb
commit
4e9b60138e
|
@ -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(),
|
||||||
})
|
}))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue