功能设计
client实现功能相对比较单一,就是能够向服务器pub消息,然后就会说订阅消息,订阅的主题收到消息以后能够得到通知.因此总结起来就是下面三个功能:
提供pub接口
提供sub接口
处理sub后收到的消息
数据结构定义
提供给用户的接口是上面的三个,
为了实现这三个接口,client一定要有的就是writer
以及handler
. 而sid则是因为一个client可以有多个sub,每一个sub要有唯一的id,主要是编号用. stop
则是为了client正常关闭使用.
#[derive(Debug)]
pub struct Client {
addr: String,
writer: Arc<Mutex<WriteHalf<TcpStream>>>,
pub stop: oneshot::Sender<()>,
sid: u64,
handler: Arc<Mutex<HashMap<String,
mpsc::UnboundedSender<Vec<u8>>>>>,
}
接口-connect
connect的功能非常直白就是创建和服务器的连接,同时后台会启动一个任务来处理tcp连接,主要是接收消息.
pub async fn connect(addr: &str) -> std::io::Result<Client> {}
接口-pub_message
向服务器发布一条pub消息
pub async fn pub_message(&mut self,
subject: &str,
msg: &[u8])
-> std::io::Result<()> {}
接口-sub_message
向服务器发布一条sub消息,然后等待服务器推送相关消息.
需要说明的是这里的参数subject
和queue
完全没有必要使用String,&str即可. 这应该是rust的一个bug,在1.41和nightly 1.43都是编译不过去的.所以退而求其次,使用了String.
//sub消息格式为SUB subject {queue} {sid}\r\n
pub async fn sub_message(
&mut self,
subject: String,
queue: Option<String>,
handler: MessageHandler,
) -> std::io::