fix: misc fixes

This commit is contained in:
evilsocket 2018-01-08 08:30:49 +01:00
commit 7203df042b

View file

@ -77,10 +77,9 @@ func (s *EventsStream) Start() error {
}() }()
return nil return nil
} else {
return fmt.Errorf("Events stream already started.")
} }
return fmt.Errorf("Events stream already started.")
} }
func (s *EventsStream) Stop() error { func (s *EventsStream) Stop() error {
@ -88,13 +87,12 @@ func (s *EventsStream) Stop() error {
s.SetRunning(false) s.SetRunning(false)
s.quit <- true s.quit <- true
return nil return nil
} else {
return fmt.Errorf("Events stream already stopped.")
} }
return fmt.Errorf("Events stream already stopped.")
} }
func (p EventsStream) OnSessionEnded(s *session.Session) { func (s *EventsStream) OnSessionEnded(sess *session.Session) {
if p.Running() { if s.Running() {
p.Stop() s.Stop()
} }
} }