读写分离模型
- 之前的框架要发送消息,直接就发送出去了
- 现在需要将写消息单独拿出来,和读消息分离开
- 使用channel管道传递数据
- 新增写数据的方法
func (c *Connection) StartWriter() {
fmt.Println("Writer Goroutine Start...")
defer fmt.Println("Writer is Exit Remote Addr is:", c.RemoteAddr().String())
for true {
select {
case data := <-c.MsgChan:
if _, err := c.Conn.Write(data); err != nil {
fmt.Println("Send Data Error", err)
return
}
case <-c.ExitChan:
return
}
}
}
- 修改发送消息方法
func (c *Connection) SendMsg(msgID uint32, data []byte) error {
if c.isClosed {
return errors.New("connection closed when sendMsg")
}
dp := NewDataPack()
buffer, err := dp.Pack(NewMessage(msgID, data))
if err != nil {
fmt.Println(err)
return err
}
c.MsgChan <- buffer
return nil
}
对于有客户端断开连接,也是使用channel管道进行通讯,在reader中,如果读数据出错,证明有客户端断开了连接,此时需要发送给writer告知writer关闭
测试
和之前的测试代码相同,可以实现读写操作