libnmdc: assert state parameter is set correctly

This commit is contained in:
mappu 2016-04-03 18:35:05 +12:00
parent 393f156b0e
commit e111c99886

View File

@ -342,10 +342,14 @@ func (this *HubConnection) worker() {
}
if err != nil {
this.OnEvent <- HubEvent{EventType: EVENT_CONNECTION_STATE_CHANGED, StateChange: CONNECTIONSTATE_CONNECTING}
this.State = CONNECTIONSTATE_DISCONNECTED
this.connValid = false
} else {
this.State = CONNECTIONSTATE_CONNECTING
this.connValid = true
this.OnEvent <- HubEvent{EventType: EVENT_CONNECTION_STATE_CHANGED, StateChange: CONNECTIONSTATE_CONNECTING}
}
}
@ -361,7 +365,10 @@ func (this *HubConnection) worker() {
// Maybe we disconnected
if err != nil {
this.OnEvent <- HubEvent{EventType: EVENT_CONNECTION_STATE_CHANGED, StateChange: CONNECTIONSTATE_DISCONNECTED, Message: err.Error()}
this.State = CONNECTIONSTATE_DISCONNECTED
this.conn = nil
this.connValid = false
time.Sleep(30 * time.Second) // Wait before reconnect
continue
}