GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-07-21 12:33:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

optimize websocket step

Browse Source
This commit is contained in:
buyuxiang
2022-07-11 18:25:43 +08:00
parent 28d5f414d0
commit aa5de4ec83
6 changed files with 278 additions and 119 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
hrp/session.go
View File

@@ -17,11 +17,11 @@ type SessionRunner struct {
// transactions stores transaction timing info.
// key is transaction name, value is map of transaction type and time, e.g. start time and end time.
transactions map[string]map[transactionType]time.Time
startTime time.Time // record start time of the testcase
summary *TestCaseSummary // record test case summary
wsConn *websocket.Conn // one websocket connection each session
pongResponseChan chan string // channel used to receive pong response message
closeResponseChan chan *wsCloseRespObject // channel used to receive close response message
startTime time.Time // record start time of the testcase
summary *TestCaseSummary // record test case summary
wsConnMap map[string]*websocket.Conn // save all websocket connections
pongResponseChan chan string // channel used to receive pong response message
closeResponseChan chan *wsCloseRespObject // channel used to receive close response message
}
func (r *SessionRunner) resetSession() {
@@ -30,6 +30,7 @@ func (r *SessionRunner) resetSession() {
r.transactions = make(map[string]map[transactionType]time.Time)
r.startTime = time.Now()
r.summary = newSummary()
r.wsConnMap = make(map[string]*websocket.Conn)
r.pongResponseChan = make(chan string, 1)
r.closeResponseChan = make(chan *wsCloseRespObject, 1)
}
@@ -102,11 +103,13 @@ func (r *SessionRunner) Start(givenVars map[string]interface{}) error {
// close websocket connection after all steps done
defer func() {
if r.wsConn != nil {
log.Info().Str("testcase", config.Name).Msg("websocket disconnected")
err := r.wsConn.Close()
if err != nil {
log.Error().Err(err).Msg("websocket disconnection failed")
for _, wsConn := range r.wsConnMap {
if wsConn != nil {
log.Info().Str("testcase", config.Name).Msg("websocket disconnected")
err := wsConn.Close()
if err != nil {
log.Error().Err(err).Msg("websocket disconnection failed")
}
}
}
}()