Exemplo n.º 1
0
func main() {
	tcpAddr, err := net.ResolveTCPAddr("tcp4", "127.0.0.1:8989")
	checkError(err)
	conn, err := net.DialTCP("tcp", nil, tcpAddr)
	checkError(err)

	echoProtocol := &protocol.EchoProtocol{}

	// ping <--> pong
	go func() {
		for {
			// write
			if _, err := conn.Write(protocol.NewEchoPacket([]byte("hello"), false).Serialize()); err != nil {
				if neterr, ok := err.(net.Error); ok && neterr.Timeout() {
					fmt.Println("=====>> ", err)
				}
			}
			time.Sleep(2 * time.Second)
		}
	}()
	for {
		//check connect
		conn.SetReadDeadline(time.Now())
		zero := make([]byte, 0)
		if _, err := conn.Read(zero); err == io.EOF {
			fmt.Println("Connect close")
			conn.Close()
			conn = nil
			panic("Connect close!")
		} else {
			conn.SetReadDeadline(time.Time{})
		}

		// read
		p, err := echoProtocol.ReadPacket(conn)
		if err == nil {
			echoPacket := p.(*protocol.EchoPacket)
			fmt.Printf("Server reply:[%v] [%v]\n", echoPacket.GetLength(), string(echoPacket.GetBody()))
		} else {
			if neterr, ok := err.(net.Error); ok && neterr.Timeout() {
				fmt.Println("=====>> ", err)
			}
		}
	}

	conn.Close()
}
Exemplo n.º 2
0
func (this *Callback) OnMessage(c *gotcp.Conn, p gotcp.Packet) bool {
	echoPacket := p.(*protocol.EchoPacket)
	fmt.Printf("OnMessage:[%v] [%v]\n", echoPacket.GetLength(), string(echoPacket.GetBody()))
	c.AsyncWritePacket(protocol.NewEchoPacket(echoPacket.Serialize(), true), time.Second)
	return true
}