Fix passing Component to StreamManager
0a4acd12c3
, which fixes #160, introduced
a regression as it assumed only Client may implement StreamClient, and
passing a component triggers an unconditional "client is not
disconnected" error.
This commit is contained in:
parent
5f99e1cd06
commit
e55463fc98
|
@ -114,18 +114,23 @@ func (sm *StreamManager) Stop() {
|
|||
|
||||
func (sm *StreamManager) connect() error {
|
||||
if sm.client != nil {
|
||||
if c, ok := sm.client.(*Client); ok {
|
||||
if c.CurrentState.getState() == StateDisconnected {
|
||||
sm.Metrics = initMetrics()
|
||||
err := c.Connect()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if sm.PostConnect != nil {
|
||||
sm.PostConnect(sm.client)
|
||||
}
|
||||
return nil
|
||||
var scs *SyncConnState
|
||||
if client, ok := sm.client.(*Client); ok {
|
||||
scs = &client.CurrentState
|
||||
}
|
||||
if component, ok := sm.client.(*Component); ok {
|
||||
scs = &component.CurrentState
|
||||
}
|
||||
if scs != nil && scs.getState() == StateDisconnected {
|
||||
sm.Metrics = initMetrics()
|
||||
err := sm.client.Connect()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if sm.PostConnect != nil {
|
||||
sm.PostConnect(sm.client)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
}
|
||||
return errors.New("client is not disconnected")
|
||||
|
|
Loading…
Reference in a new issue