ccf11990f1
- race conditions between ctx.Done and waitError channel - Sleep for retry cancels on cancelation of context - Stops the any loop at the start if the context was canceled - Mentions when loops exit - Wait for errors on triggered loop restarts
110 lines
2.7 KiB
Go
110 lines
2.7 KiB
Go
package tinyproxy
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/qdm12/golibs/command"
|
|
"github.com/qdm12/golibs/logging"
|
|
"github.com/qdm12/private-internet-access-docker/internal/constants"
|
|
"github.com/qdm12/private-internet-access-docker/internal/firewall"
|
|
"github.com/qdm12/private-internet-access-docker/internal/settings"
|
|
)
|
|
|
|
type Looper interface {
|
|
Run(ctx context.Context, restart <-chan struct{}, wg *sync.WaitGroup)
|
|
}
|
|
|
|
type looper struct {
|
|
conf Configurator
|
|
firewallConf firewall.Configurator
|
|
settings settings.TinyProxy
|
|
logger logging.Logger
|
|
streamMerger command.StreamMerger
|
|
uid int
|
|
gid int
|
|
}
|
|
|
|
func (l *looper) logAndWait(ctx context.Context, err error) {
|
|
l.logger.Error(err)
|
|
l.logger.Info("retrying in 1 minute")
|
|
ctx, cancel := context.WithTimeout(ctx, time.Minute)
|
|
defer cancel() // just for the linter
|
|
<-ctx.Done()
|
|
}
|
|
|
|
func NewLooper(conf Configurator, firewallConf firewall.Configurator, settings settings.TinyProxy,
|
|
logger logging.Logger, streamMerger command.StreamMerger, uid, gid int) Looper {
|
|
return &looper{
|
|
conf: conf,
|
|
firewallConf: firewallConf,
|
|
settings: settings,
|
|
logger: logger.WithPrefix("tinyproxy: "),
|
|
streamMerger: streamMerger,
|
|
uid: uid,
|
|
gid: gid,
|
|
}
|
|
}
|
|
|
|
func (l *looper) Run(ctx context.Context, restart <-chan struct{}, wg *sync.WaitGroup) {
|
|
wg.Add(1)
|
|
defer wg.Done()
|
|
select {
|
|
case <-restart:
|
|
case <-ctx.Done():
|
|
return
|
|
}
|
|
defer l.logger.Warn("loop exited")
|
|
|
|
for ctx.Err() == nil {
|
|
err := l.conf.MakeConf(
|
|
l.settings.LogLevel,
|
|
l.settings.Port,
|
|
l.settings.User,
|
|
l.settings.Password,
|
|
l.uid,
|
|
l.gid)
|
|
if err != nil {
|
|
l.logAndWait(ctx, err)
|
|
continue
|
|
}
|
|
err = l.firewallConf.AllowAnyIncomingOnPort(ctx, l.settings.Port)
|
|
// TODO remove firewall rule on exit below
|
|
if err != nil {
|
|
l.logger.Error(err)
|
|
}
|
|
tinyproxyCtx, tinyproxyCancel := context.WithCancel(context.Background())
|
|
stream, waitFn, err := l.conf.Start(tinyproxyCtx)
|
|
if err != nil {
|
|
tinyproxyCancel()
|
|
l.logAndWait(ctx, err)
|
|
continue
|
|
}
|
|
go l.streamMerger.Merge(tinyproxyCtx, stream,
|
|
command.MergeName("tinyproxy"), command.MergeColor(constants.ColorTinyproxy()))
|
|
waitError := make(chan error)
|
|
go func() {
|
|
err := waitFn() // blocking
|
|
waitError <- err
|
|
}()
|
|
select {
|
|
case <-ctx.Done():
|
|
l.logger.Warn("context canceled: exiting loop")
|
|
tinyproxyCancel()
|
|
<-waitError
|
|
close(waitError)
|
|
return
|
|
case <-restart: // triggered restart
|
|
l.logger.Info("restarting")
|
|
tinyproxyCancel()
|
|
<-waitError
|
|
close(waitError)
|
|
case err := <-waitError: // unexpected error
|
|
tinyproxyCancel()
|
|
close(waitError)
|
|
l.logAndWait(ctx, err)
|
|
}
|
|
}
|
|
}
|