github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-08-20 05:23:19 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

refact: refactored module SetRunning method (fixes #49)

Browse source
This commit is contained in:
evilsocket 2018-02-11 01:43:48 +01:00
commit 9c6eb70eb3
15 changed files with 103 additions and 174 deletions
Download patch file Download diff file Expand all files Collapse all files

21
modules/events_stream.go
View file

@ -79,15 +79,11 @@ func (s *EventsStream) Configure() error {
}
func (s *EventsStream) Start() error {
if s.Running() == true {
return session.ErrAlreadyStarted
} else if err := s.Configure(); err != nil {
if err := s.Configure(); err != nil {
return err
}
s.SetRunning(true)
go func() {
return s.SetRunning(true, func() {
for {
var e session.Event
select {
@ -99,9 +95,7 @@ func (s *EventsStream) Start() error {
return
}
}
}()
return nil
})
}
func (s *EventsStream) Show(limit int) error {
@ -123,10 +117,7 @@ func (s *EventsStream) Show(limit int) error {
}
func (s *EventsStream) Stop() error {
if s.Running() == false {
return session.ErrAlreadyStopped
}
s.SetRunning(false)
s.quit <- true
return nil
return s.SetRunning(false, func() {
s.quit <- true
})
}