Gateway: Minor bug fixes

gateway/gateway.go

@@ -329,7 +329,7 @@ func (g *Gateway) eventLoop() error {
 				return nil
 			}
 
-			return errors.New("Pacemaker died, reconnecting.")
+			return errors.Wrap(err, "Pacemaker died, reconnecting")
 
 		case ev := <-ch:
 			// Check for error

gateway/pacemaker.go

@@ -77,21 +77,22 @@ func (p *Pacemaker) start(stop chan struct{}, wg *sync.WaitGroup) error {
 	p.Echo()
 
 	for {
+		if err := p.Pace(); err != nil {
+			return err
+		}
+
+		// Paced, save:
+		atomic.StoreInt64(&p.SentBeat, time.Now().UnixNano())
+
+		if p.Dead() {
+			return ErrDead
+		}
+
 		select {
 		case <-stop:
 			return nil
 
 		case <-tick.C:
-			if err := p.Pace(); err != nil {
-				return err
-			}
-
-			// Paced, save:
-			atomic.StoreInt64(&p.SentBeat, time.Now().UnixNano())
-
-			if p.Dead() {
-				return ErrDead
-			}
 		}
 	}
 }

internal/wsutil/ws.go

@@ -58,6 +58,9 @@ func (ws *Websocket) Dial(ctx context.Context) error {
 		return errors.Wrap(err, "Failed to dial")
 	}
 
+	// Reset the SendLimiter:
+	ws.SendLimiter = NewSendLimiter()
+
 	return nil
 }