code refactor

OffchainLabs · Oct 25, 2023 · 70df6d6 · 70df6d6
1 parent 0e9445d
commit 70df6d6
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/wsbroadcastserver/wsbroadcastserver.go b/wsbroadcastserver/wsbroadcastserver.go
@@ -204,11 +204,18 @@ func (s *WSBroadcastServer) Start(ctx context.Context) error {
 		HTTPHeaderChainId:           []string{strconv.FormatUint(s.chainId, 10)},
 	})
 
-	return s.StartWithHeader(ctx, header)
+	startTime := time.Now()
+	err := s.StartWithHeader(ctx, header)
+	elapsed := time.Since(startTime)
+	startWithHeaderTimer.Update(elapsed)
+
+	if err != nil {
+		return err
+	}
+	return nil
 }
 
 func (s *WSBroadcastServer) StartWithHeader(ctx context.Context, header ws.HandshakeHeader) error {
-	startTimeMain := time.Now()
 	s.startMutex.Lock()
 	defer s.startMutex.Unlock()
 	if s.started {
@@ -490,9 +497,6 @@ func (s *WSBroadcastServer) StartWithHeader(ctx context.Context, header ws.Hands
 
 	s.started = true
 
-	elapsedMain := time.Since(startTimeMain)
-	startWithHeaderTimer.Update(elapsedMain)
-
 	return nil
 }