Maint: improve http proxy loop Run

This commit is contained in:
Quentin McGaw (desktop)
2021-07-26 01:42:37 +00:00
parent 037b43ee10
commit d5ba15c23b
2 changed files with 18 additions and 17 deletions

View File

@@ -2,7 +2,7 @@ package httpproxy
import (
"context"
"fmt"
"strconv"
"github.com/qdm12/gluetun/internal/constants"
)
@@ -14,34 +14,28 @@ type Runner interface {
func (l *Loop) Run(ctx context.Context, done chan<- struct{}) {
defer close(done)
crashed := false
if l.state.GetSettings().Enabled {
go func() {
_, _ = l.statusManager.ApplyStatus(ctx, constants.Running)
}()
}
select {
case <-l.start:
case <-ctx.Done():
return
if !l.state.GetSettings().Enabled {
select {
case <-l.start:
case <-ctx.Done():
return
}
}
for ctx.Err() == nil {
runCtx, runCancel := context.WithCancel(ctx)
settings := l.state.GetSettings()
address := fmt.Sprintf(":%d", settings.Port)
address := ":" + strconv.Itoa(int(settings.Port))
server := New(runCtx, address, l.logger, settings.Stealth, settings.Log, settings.User, settings.Password)
errorCh := make(chan error)
go server.Run(runCtx, errorCh)
// TODO stable timer, check Shadowsocks
if !crashed {
if l.userTrigger {
l.running <- constants.Running
crashed = false
l.userTrigger = false
} else {
l.backoffTime = defaultBackoffTime
l.statusManager.SetStatus(constants.Running)
@@ -53,21 +47,26 @@ func (l *Loop) Run(ctx context.Context, done chan<- struct{}) {
case <-ctx.Done():
runCancel()
<-errorCh
close(errorCh)
return
case <-l.start:
l.userTrigger = true
l.logger.Info("starting")
runCancel()
<-errorCh
close(errorCh)
stayHere = false
case <-l.stop:
l.userTrigger = true
l.logger.Info("stopping")
runCancel()
<-errorCh
// Do not close errorCh or this for loop won't work
l.stopped <- struct{}{}
case err := <-errorCh:
close(errorCh)
l.statusManager.SetStatus(constants.Crashed)
l.logAndWait(ctx, err)
crashed = true
stayHere = false
}
}