liwei_3d/go_dreamfactory
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

websocket返回验证

Browse Source
This commit is contained in:
meixiongfeng 2022-11-28 17:35:28 +08:00
parent 2e9027c655
commit 8ac41f610c
3 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
stress/robot/robot.go
View File

@ -204,3 +204,11 @@ func (r *Robot) CloseHandler() {
r.ws.Close() r.ws.Close()
} }
func (r *Robot) GetMessagedata() []byte {
_, data, err := r.ws.ReadMessage()
if err != nil {
fmt.Printf("readMessage err:%v", err)
}
return data
}

1
stress/server/golink/websocket_link.go
View File

@ -83,6 +83,7 @@ func webSocketRequest(chanID uint64, ch chan<- *model.RequestResults, i uint64,
for { for {
if r.MessageRsp("user", "login") { if r.MessageRsp("user", "login") {
isSucceed = true isSucceed = true
errCode, isSucceed = request.GetVerifyWebSocket()(request, "webSocket.json", r.GetMessagedata())
break break
} }
} }

2
stress/stress.go
View File

@ -81,7 +81,7 @@ func main() {
flag.Usage() flag.Usage()
return return
} }
debug := strings.ToLower(debugStr) == "true" debug := strings.ToLower(debugStr) == "false"
request, err := model.NewRequest(requestURL, verify, code, 0, debug, path, headers, body, maxCon, http2, keepalive) request, err := model.NewRequest(requestURL, verify, code, 0, debug, path, headers, body, maxCon, http2, keepalive)
if err != nil { if err != nil {
fmt.Printf("参数不合法 %v \n", err) fmt.Printf("参数不合法 %v \n", err)