-
Notifications
You must be signed in to change notification settings - Fork 1
/
socket_client_ab.go
73 lines (66 loc) · 1.65 KB
/
socket_client_ab.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
package main
import (
"log"
"net"
"strconv"
"sync"
"sync/atomic"
"time"
"github.com/henrylee2cn/teleport/codec"
"github.com/henrylee2cn/teleport/socket"
"github.com/henrylee2cn/teleport/socket/example/pb"
)
//go:generate go build $GOFILE
func main() {
socket.SetNoDelay(false)
conn, err := net.Dial("tcp", "127.0.0.1:8000")
if err != nil {
log.Fatalf("[CLI] dial err: %v", err)
}
s := socket.GetSocket(conn)
defer s.Close()
var count sync.WaitGroup
t := time.Now()
loop := 30
group := 10000
var failNum uint32
defer func() {
cost := time.Since(t)
times := time.Duration(loop * group)
log.Printf("------------------- call times: %d ok: %d fail: %d | cost time: %v | QPS: %d -----------------", times, uint32(times)-failNum, failNum, cost, time.Second*times/cost)
}()
for j := 0; j < loop; j++ {
count.Add(group)
for i := 0; i < group; i++ {
go func(a int) {
var message = socket.GetMessage()
defer func() {
socket.PutMessage(message)
count.Done()
}()
// write request
message.Reset()
message.SetBodyCodec(codec.ID_PROTOBUF)
message.SetSeq(strconv.Itoa(a))
message.SetUri("/a/b")
message.SetBody(&pb.PbTest{A: 10, B: 2})
err = s.WriteMessage(message)
if err != nil {
atomic.AddUint32(&failNum, 1)
log.Printf("[CLI] write request err: %v", err)
return
}
// read response
message.Reset(socket.WithNewBody(func(header socket.Header) interface{} {
return new(pb.PbTest)
}))
err = s.ReadMessage(message)
if err != nil {
atomic.AddUint32(&failNum, 1)
log.Printf("[CLI] read response err: %v", err)
}
}(i * group)
}
count.Wait()
}
}