refact: refactored module SetRunning method (fixes #49)

This commit is contained in:
evilsocket 2018-02-11 01:43:48 +01:00
parent 76d6977967
commit 9c6eb70eb3
15 changed files with 103 additions and 174 deletions

View file

@ -131,17 +131,13 @@ func (p *HttpsProxy) Start() error {
return err
}
p.SetRunning(true)
p.proxy.Start()
return nil
return p.SetRunning(true, func() {
p.proxy.Start()
})
}
func (p *HttpsProxy) Stop() error {
if p.Running() == false {
return session.ErrAlreadyStopped
}
p.SetRunning(false)
return p.proxy.Stop()
return p.SetRunning(false, func() {
p.proxy.Stop()
})
}