Fix connect() to work with external component

This commit is contained in:
Dave Johnston 2020-09-04 10:54:47 +01:00 committed by GitHub
parent ac5b066815
commit ddb8c5a1eb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -113,22 +113,27 @@ func (sm *StreamManager) Stop() {
} }
func (sm *StreamManager) connect() error { func (sm *StreamManager) connect() error {
if sm.client != nil { if sm.client == nil {
if c, ok := sm.client.(*Client); ok { return errors.New("client is not set")
if c.CurrentState.getState() == StateDisconnected { }
sm.Metrics = initMetrics()
err := c.Connect() if c, ok := sm.client.(*Client); ok {
if err != nil { if c.CurrentState.getState() != StateDisconnected {
return err return errors.New("client is not disconnected")
}
if sm.PostConnect != nil {
sm.PostConnect(sm.client)
}
return nil
}
} }
} }
return errors.New("client is not disconnected")
sm.Metrics = initMetrics()
if err := sm.client.Connect(); err != nil {
return err
}
if sm.PostConnect != nil {
sm.PostConnect(sm.client)
}
return nil
} }
// resume manages the reconnection loop and apply the define backoff to avoid overloading the server. // resume manages the reconnection loop and apply the define backoff to avoid overloading the server.