从零开始写Go网络通信框架（4）——全双工收发消息

经过上面三篇文章，我们已经写了一个简单的C/S框架，可以实现客户端与服务端进行发送消息。但这是一个单向通讯的。我们要想让客户端接收服务端发来的命令，并执行响应的动作，就需要写成全双工收发消息。即客户端也能接收服务端发来的指令。

下面我在客户端定义一个方法，用来接收服务端的消息。

//接收服务端发来的消息func ReadMsg(conn net.Conn)  {//存储被截断的数据tmpbuf:=make([] byte,0)buf:=make([] byte,1024)//将信息解包n,_:=conn.Read(buf)tmpbuf = protocol.Depack(append(tmpbuf,buf[:n]...))msg:=string(tmpbuf)fmt.Println("server say:",msg)}

客户端完整代码如下：

package mainimport ("fmt""os""net""strconv""time""github.com/mxi4oyu/MoonSocket/protocol")//定义CheckError方法，避免写太多到 if err!=nilfunc CheckError(err error)  {if err!=nil{fmt.Fprintf(os.Stderr,"Fatal error:%s",err.Error())os.Exit(1)}}func main()  {if len(os.Args) !=2 {fmt.Fprintf(os.Stderr,"Usage:%s IP:Port\n",os.Args[0])os.Exit(1)}//动态传入服务端IP和端口号service:=os.Args[1]tcpAddr,err:=net.ResolveTCPAddr("tcp4",service)CheckError(err)conn,err:=net.DialTCP("tcp",nil,tcpAddr)CheckError(err)ch:=make(chan int,100)ticker := time.NewTicker(time.Second)defer ticker.Stop()for{select {case <-ticker.C:ch<-1go ClientMsgHandler(conn,ch)case <-time.After(time.Second*10):defer conn.Close()fmt.Println("timeout")}}}//客户端消息处理func ClientMsgHandler(conn net.Conn,ch chan int)  {<-ch//获取当前时间msg:=time.Now().String()go SendMsg(conn,msg)go ReadMsg(conn)}func GetSession() string{gs1:=time.Now().Unix()gs2:=strconv.FormatInt(gs1,10)return gs2}//接收服务端发来的消息func ReadMsg(conn net.Conn)  {//存储被截断的数据tmpbuf:=make([] byte,0)buf:=make([] byte,1024)//将信息解包n,_:=conn.Read(buf)tmpbuf = protocol.Depack(append(tmpbuf,buf[:n]...))msg:=string(tmpbuf)fmt.Println("server say:",msg)}//向服务端发送消息func SendMsg(conn net.Conn,msg string)  {session:=GetSession()words := "{\"Session\":"+session +",\"Meta\":\"Monitor\",\"Message\":\""+msg+"\"}"//将信息封包smsg:=protocol.Enpack([]byte(words))conn.Write(smsg)}

在服务端，我们也做了响应的调整。定义一个方法，专门向客户端发送消息。

//服务端向客户端发送消息func WriteMsgToClient(conn net.Conn)  {talk:="wordpress"//将信息封包smsg:=protocol.Enpack([]byte(talk))conn.Write(smsg)}

服务端完整代码如下：

package mainimport ("fmt""os""net""log""github.com/mxi4oyu/MoonSocket/protocol""time")//定义CheckError方法，避免写太多到 if err!=nilfunc CheckError(err error)  {if err!=nil{fmt.Fprintf(os.Stderr,"Fatal error:%s",err.Error())os.Exit(1)}}//自定义logfunc Log(v... interface{})  {log.Println(v...)}func main()  {    server_listener,err:=net.Listen("tcp","localhost:8848")CheckError(err)defer server_listener.Close()Log("Waiting for clients connect")for{new_conn,err:=server_listener.Accept()CheckError(err)go ServerMsgHandler(new_conn)}}//服务端消息处理func ServerMsgHandler(conn net.Conn)  {//存储被截断的数据tmpbuf:=make([] byte,0)buf:=make([] byte,1024)defer conn.Close()//接收解包readchan:=make(chan [] byte,16)go ReadChan(readchan)for{//读取客户端发来的消息n,err:=conn.Read(buf)if err!=nil{fmt.Println("connection close")return}//解包tmpbuf = protocol.Depack(append(tmpbuf,buf[:n]...))fmt.Println("client say:",string(tmpbuf))Msg:=tmpbuf//向客户端发送消息go WriteMsgToClient(conn)beatch :=make(chan byte)//心跳计时，默认30秒go HeartBeat(conn,beatch,30)//检测每次Client是否有数据传来go HeartChanHandler(Msg,beatch)}}//处理心跳,根据HeartChanHandler判断Client是否在设定时间内发来信息func HeartBeat(conn net.Conn,heartChan chan byte,timeout int)  {select {case hc:=<-heartChan:Log("<-heartChan:",string(hc))conn.SetDeadline(time.Now().Add(time.Duration(timeout)*time.Second))breakcase <-time.After(time.Second*30):Log("timeout")conn.Close()}}//服务端向客户端发送消息func WriteMsgToClient(conn net.Conn)  {talk:="wordpress"//将信息封包smsg:=protocol.Enpack([]byte(talk))conn.Write(smsg)}//处理心跳channelfunc HeartChanHandler( n [] byte,beatch chan byte)  {for _,v:=range n{beatch<-v}close(beatch)}//从channell中读取数据func ReadChan(readchan chan [] byte)  {for{select {case data:=<-readchan:Log(string(data))}}}

现在配上一张图，Server端不断给Client端发送消息"wordpress"，Client不断向Server端汇报当前时间。