From b9171ca7725148466f6002dcaaaa950ab08799a7 Mon Sep 17 00:00:00 2001 From: "lilong.129" Date: Tue, 20 Aug 2024 14:46:33 +0800 Subject: [PATCH] refactor: SessionRunner, merge resetSession --- hrp/runner.go | 24 ++++++++++-------------- 1 file changed, 10 insertions(+), 14 deletions(-) diff --git a/hrp/runner.go b/hrp/runner.go index f80f20fa..fd45e2fe 100644 --- a/hrp/runner.go +++ b/hrp/runner.go @@ -412,10 +412,18 @@ func (r *CaseRunner) parseConfig() (parsedConfig *TConfig, err error) { // each boomer task initiates a new session // in order to avoid data racing func (r *CaseRunner) NewSession() *SessionRunner { + log.Info().Msg("create new session runner") sessionRunner := &SessionRunner{ - caseRunner: r, + caseRunner: r, + sessionVariables: make(map[string]interface{}), + summary: newSummary(), + + transactions: make(map[string]map[transactionType]time.Time), + wsConnMap: make(map[string]*websocket.Conn), + inheritWsConnMap: make(map[string]*websocket.Conn), + pongResponseChan: make(chan string, 1), + closeResponseChan: make(chan *wsCloseRespObject, 1), } - sessionRunner.resetSession() return sessionRunner } @@ -437,18 +445,6 @@ type SessionRunner struct { closeResponseChan chan *wsCloseRespObject // channel used to receive close response message } -func (r *SessionRunner) resetSession() { - log.Info().Msg("reset session runner") - r.sessionVariables = make(map[string]interface{}) - r.summary = newSummary() - - r.transactions = make(map[string]map[transactionType]time.Time) - r.wsConnMap = make(map[string]*websocket.Conn) - r.inheritWsConnMap = make(map[string]*websocket.Conn) - r.pongResponseChan = make(chan string, 1) - r.closeResponseChan = make(chan *wsCloseRespObject, 1) -} - func (r *SessionRunner) inheritConnection(src *SessionRunner) { log.Info().Msg("inherit session runner") r.inheritWsConnMap = make(map[string]*websocket.Conn, len(src.wsConnMap)+len(src.inheritWsConnMap))