Simplify stopLoadTestIfNoWorkerRemaining

pkg/server/server.go

@@ -173,10 +173,10 @@ func (s *Server) acceptWorkerConn(responseWriter http.ResponseWriter, req *http.
 
 	logger.Infow("Worker connected", "name", name)
 
+	defer s.stopLoadTestIfNoWorkerRemaining()
 	defer logger.Infow("Worker removed", "name", name)
 	defer s.workerService.RemoveWorker(conn)
 	defer conn.Close()
-	defer s.stopLoadTestIfNoWorkerRemaining()
 
 	for {
 		_, message, err := conn.ReadMessage()
@@ -188,10 +188,8 @@ func (s *Server) acceptWorkerConn(responseWriter http.ResponseWriter, req *http.
 	}
 }
 
-func (s *Server) stopLoadTestIfNoWorkerRemaining(){
-	val := len(s.workerService.workers) - 1
-	logger.Infow("Number of remaining workers: ", "", val)
-	if val == 0 {
+func (s *Server) stopLoadTestIfNoWorkerRemaining() {
+	if len(s.workerService.workers) == 0 {
 		s.loadTestState = messages.ServerStateStopped
 	}
 }