96 lines
1.6 KiB
Go
96 lines
1.6 KiB
Go
|
package ensureSendUtil
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"net"
|
|||
|
|
"testing"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"goutil/debugUtil"
|
|||
|
|
"goutil/zlibUtil"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// 保存接收的数据用于校验
|
|||
|
|
var tcp_recv_msg = make([]byte, 0)
|
|||
|
|
|
|||
|
|
func init() {
|
|||
|
|
debugUtil.SetDebug(true)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 创建socket服务器，保存收到的数据
|
|||
|
|
func server(addr string) net.Listener {
|
|||
|
|
listener, err := net.Listen("tcp", addr)
|
|||
|
|
if err != nil {
|
|||
|
|
panic(err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
go func() {
|
|||
|
|
for {
|
|||
|
|
conn, err := listener.Accept()
|
|||
|
|
if err != nil {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for {
|
|||
|
|
buff := make([]byte, 512)
|
|||
|
|
_, err := conn.Read(buff)
|
|||
|
|
if err != nil {
|
|||
|
|
break
|
|||
|
|
} else {
|
|||
|
|
decompressed, err := zlibUtil.Decompress(buff[4:])
|
|||
|
|
if err != nil {
|
|||
|
|
panic(err)
|
|||
|
|
} else {
|
|||
|
|
tcp_recv_msg = append(tcp_recv_msg, decompressed...)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}()
|
|||
|
|
|
|||
|
|
return listener
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func Test_tcp(t *testing.T) {
|
|||
|
|
// 开启服务器
|
|||
|
|
l := server("127.0.0.1:9559")
|
|||
|
|
|
|||
|
|
tcp, err := NewTCPSender("./test_tcp", "127.0.0.1:9559")
|
|||
|
|
if err != nil {
|
|||
|
|
t.Error(err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 发送消息
|
|||
|
|
tcp.Write("tcp-msg-1")
|
|||
|
|
time.Sleep(time.Millisecond * 50) // 等待协程发送数据
|
|||
|
|
|
|||
|
|
// 关闭连接和服务器
|
|||
|
|
l.Close()
|
|||
|
|
(tcp.(*tcpSender)).conn.Close()
|
|||
|
|
|
|||
|
|
// 发送消息，此数据会失败
|
|||
|
|
tcp.Write("tcp-msg-2")
|
|||
|
|
// time.Sleep(time.Millisecond * 50)
|
|||
|
|
|
|||
|
|
// 保存数据
|
|||
|
|
tcp.Close()
|
|||
|
|
|
|||
|
|
// 重启，检查是否重发tcp-msg-2
|
|||
|
|
l = server("127.0.0.1:9559")
|
|||
|
|
tcp, err = NewTCPSender("./test_tcp", "127.0.0.1:9559")
|
|||
|
|
if err != nil {
|
|||
|
|
t.Error(err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
time.Sleep(time.Second * 2)
|
|||
|
|
|
|||
|
|
if string(tcp_recv_msg) != "tcp-msg-1tcp-msg-2" {
|
|||
|
|
t.Error("message error. got " + string(tcp_recv_msg))
|
|||
|
|
} else {
|
|||
|
|
fmt.Println("TCP OK")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
tcp.Close()
|
|||
|
|
l.Close()
|
|||
|
|
}
|