消息长度，错误信息统计

stress/robot/robot.go

@@ -168,15 +168,15 @@ func (r *RobotFormatter) Format(entry *logrus.Entry) ([]byte, error) {
 	return b.Bytes(), nil
 }
 
-func (r *Robot) MessageRsp(mainType, subType string) bool {
+func (r *Robot) MessageRsp(mainType, subType string) (bool, int64) {
-
+	var byteLen int64
 	var msg *pb.UserMessage = &pb.UserMessage{}
 	_, data, err := r.ws.ReadMessage()
 	if err != nil {
 		fmt.Printf("readMessage err:%v", err)
-		return false
+		return false, byteLen
 	}
-
+	byteLen = int64(len(data))
 	if err = proto.Unmarshal(data, msg); err != nil {
 		fmt.Printf("unmarshal err:%v", err)
 	}
@@ -184,19 +184,19 @@ func (r *Robot) MessageRsp(mainType, subType string) bool {
 	if msg.MainType == "chat" && msg.SubType == "message" {
 		resp := &pb.ChatMessagePush{}
 		if !comm.ProtoUnmarshal(msg, resp) {
-			return false
+			return false, byteLen
 		}
-		return false
+		return false, byteLen
 		//fmt.Printf("接收消息=====resp:%v\n", resp.Chat)
 	} else if msg.MainType == mainType && msg.SubType == subType {
 		resp := &pb.UserLoginResp{}
 		if !comm.ProtoUnmarshal(msg, resp) {
-			return false
+			return false, byteLen
 		}
 		//fmt.Printf("接收消息=====loginData:%v,userExpand:%v\n", resp.Data, resp.Ex)
-		return true
+		return true, byteLen
 	}
-	return false
+	return false, byteLen
 }
 
 func (r *Robot) CloseHandler() {
@@ -212,3 +212,42 @@ func (r *Robot) GetMessagedata() []byte {
 	}
 	return data
 }
+func (r *Robot) Login() error {
+	msg := &pb.UserMessage{MainType: "user", SubType: "login"}
+	rsp := &pb.UserLoginReq{
+		Account: "user001",
+		Sid:     "dfmxf",
+	}
+	msg.Sec = r.BuildSecStr()
+	if comm.ProtoMarshal(rsp, msg) {
+		data, _ := proto.Marshal(msg)
+		return r.ws.WriteMessage(websocket.BinaryMessage, data)
+	}
+	return nil
+}
+
+func (r *Robot) Create() bool {
+	mainType := "user"
+	subType := "create"
+	msg := &pb.UserMessage{MainType: mainType, SubType: subType}
+	rsp := &pb.UserCreateReq{
+		NickName: "001",
+		Figure:   100,
+		Gender:   0,
+	}
+	msg.Sec = r.BuildSecStr()
+	if comm.ProtoMarshal(rsp, msg) {
+		data, _ := proto.Marshal(msg)
+		err := r.ws.WriteMessage(websocket.BinaryMessage, data)
+		if err != nil {
+			fmt.Printf("WriteMessage err:%v", err)
+		}
+	}
+
+	for {
+		if b, _ := r.MessageRsp(mainType, subType); b {
+			return true
+		}
+	}
+
+}

stress/server/dispose.go

@@ -29,6 +29,7 @@ func init() {
 
 	// webSocket
 	model.RegisterVerifyWebSocket("json", verify.WebSocketJSON)
+	model.RegisterVerifyWebSocket("pb", verify.WebSocketProto)
 }
 
 // Dispose 处理函数
@@ -52,13 +53,6 @@ func Dispose(ctx context.Context, concurrency, totalNumber uint64, request *mode
 			case 1:
 				// 连接以后再启动协程
 				r := robot.NewRobot(request.URL)
-				// head := &pb.UserMessage{MainType: "user", SubType: "login"}
-				// req := &pb.UserLoginReq{
-				// 	Account: "user001",
-				// 	Sid:     "dfmxf",
-				// }
-				// r.SendToClient(head, req)
-				// r.MessageRsp()
 
 				go golink.WebSocket(ctx, i, ch, totalNumber, &wg, request, r)
 			case 2:

stress/server/golink/websocket_link.go

@@ -72,6 +72,7 @@ func webSocketRequest(chanID uint64, ch chan<- *model.RequestResults, i uint64,
 		startTime = time.Now()
 		isSucceed = false
 		errCode   = model.HTTPOk
+		byteLen   int64 // 协议传输字节数
 	)
 	// 需要发送的数据
 	head := &pb.UserMessage{MainType: "user", SubType: "login"}
@@ -81,18 +82,23 @@ func webSocketRequest(chanID uint64, ch chan<- *model.RequestResults, i uint64,
 	}
 	r.SendToClient(head, req)
 	for {
-		if r.MessageRsp("user", "login") {
+
+		if b, len := r.MessageRsp("user", "login"); b {
+			byteLen += len
 			isSucceed = true
-			errCode, isSucceed = request.GetVerifyWebSocket()(request, "webSocket.json", r.GetMessagedata())
+			//errCode, isSucceed = request.GetVerifyWebSocket()(request, "webSocket.pb", r.GetMessagedata())
 			break
 		}
 	}
 
 	requestTime := uint64(helper.DiffNano(startTime))
 	requestResults := &model.RequestResults{
+		ID:            "",
+		ChanID:        chanID,
 		Time:          requestTime,
 		IsSucceed:     isSucceed,
 		ErrCode:       errCode,
+		ReceivedBytes: byteLen,
 	}
 	requestResults.SetID(chanID, i)
 	ch <- requestResults

stress/server/statistics/statistics.go

@@ -38,7 +38,7 @@ func ReceivingResults(concurrent uint64, ch <-chan *model.RequestResults, wg *sy
 		maxTime        uint64 // 最大时长
 		minTime        uint64 // 最小时长
 		successNum     uint64 // 成功处理数，code为0
-		failureNum     uint64 // 处理失败数，code不为0
+		failureNum     uint64 // 处理失败数，code不为0 0
 		chanIDLen      int    // 并发数
 		chanIDs        = make(map[uint64]bool)
 		receivedBytes  int64

stress/server/verify/websokcet_verify.go

@@ -4,8 +4,11 @@ package verify
 import (
 	"encoding/json"
 	"fmt"
-
+	"go_dreamfactory/comm"
+	"go_dreamfactory/pb"
 	"go_dreamfactory/stress/model"
+
+	"google.golang.org/protobuf/proto"
 )
 
 // WebSocketResponseJSON 返回数据结构体，返回值为json
@@ -47,3 +50,29 @@ func WebSocketJSON(request *model.Request, seq string, msg []byte) (code int, is
 	}
 	return
 }
+
+func WebSocketProto(request *model.Request, seq string, data []byte) (code int, isSucceed bool) {
+
+	var msg *pb.UserMessage = &pb.UserMessage{}
+
+	if err := proto.Unmarshal(data, msg); err != nil {
+		fmt.Printf("unmarshal err:%v", err)
+	}
+	//fmt.Printf("接收消息=====msg.MainType = %s,msg.SubType = %s\n", msg.MainType, msg.SubType)
+	if msg.MainType == "chat" && msg.SubType == "message" {
+		resp := &pb.ChatMessagePush{}
+		if !comm.ProtoUnmarshal(msg, resp) {
+			code = model.ParseError
+		}
+		return
+		//fmt.Printf("接收消息=====resp:%v\n", resp.Chat)
+	} else if msg.MainType == "user" && msg.SubType == "login" {
+		resp := &pb.UserLoginResp{}
+		if !comm.ProtoUnmarshal(msg, resp) {
+			code = model.ParseError
+		}
+		//fmt.Printf("接收消息=====loginData:%v,userExpand:%v\n", resp.Data, resp.Ex)
+		return
+	}
+	return
+}

stress/stress.go

@@ -74,6 +74,7 @@ func main() {
 	totalNumber = 10
 	debugStr = "false"
 	requestURL = "ws://10.0.0.9:7891/gateway"
+	verify = "pb"
 	if concurrency == 0 || totalNumber == 0 || (requestURL == "" && path == "") {
 		fmt.Printf("示例: go run main.go -c 1 -n 1 -u https://www.baidu.com/ \n")
 		fmt.Printf("压测地址或curl路径必填 \n")