You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
68 lines
1.4 KiB
68 lines
1.4 KiB
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"net"
|
|
"time"
|
|
|
|
"github.com/aceld/zinx/znet"
|
|
)
|
|
|
|
/*
|
|
模拟客户端
|
|
*/
|
|
func main() {
|
|
|
|
fmt.Println("Client Test ... start")
|
|
//3秒之后发起测试请求，给服务端开启服务的机会
|
|
time.Sleep(3 * time.Second)
|
|
|
|
conn, err := net.Dial("tcp", "127.0.0.1:7777")
|
|
if err != nil {
|
|
fmt.Println("client start err, exit!")
|
|
return
|
|
}
|
|
|
|
for {
|
|
//发封包message消息
|
|
dp := znet.NewDataPack()
|
|
msg, _ := dp.Pack(znet.NewMsgPackage(0, []byte("Zinx V0.5 Client Test Message")))
|
|
_, err := conn.Write(msg)
|
|
if err != nil {
|
|
fmt.Println("write error err ", err)
|
|
return
|
|
}
|
|
|
|
//先读出流中的head部分
|
|
headData := make([]byte, dp.GetHeadLen())
|
|
_, err = io.ReadFull(conn, headData) //ReadFull 会把msg填充满为止
|
|
if err != nil {
|
|
fmt.Println("read head error")
|
|
break
|
|
}
|
|
//将headData字节流 拆包到msg中
|
|
msgHead, err := dp.Unpack(headData)
|
|
if err != nil {
|
|
fmt.Println("server unpack err:", err)
|
|
return
|
|
}
|
|
|
|
if msgHead.GetDataLen() > 0 {
|
|
//msg 是有data数据的，需要再次读取data数据
|
|
msg := msgHead.(*znet.Message)
|
|
msg.Data = make([]byte, msg.GetDataLen())
|
|
|
|
//根据dataLen从io中读取字节流
|
|
_, err := io.ReadFull(conn, msg.Data)
|
|
if err != nil {
|
|
fmt.Println("server unpack data err:", err)
|
|
return
|
|
}
|
|
|
|
fmt.Println("==> Recv Msg: ID=", msg.ID, ", len=", msg.DataLen, ", data=", string(msg.Data))
|
|
}
|
|
|
|
time.Sleep(1 * time.Second)
|
|
}
|
|
}
|