William Petit
80a1b48966
All checks were successful
Cadoles/bouncer/pipeline/pr-master This commit looks good
127 lines
3.4 KiB
Go
127 lines
3.4 KiB
Go
package proxy
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"os/signal"
|
|
"strings"
|
|
"syscall"
|
|
"time"
|
|
|
|
"forge.cadoles.com/cadoles/bouncer/internal/command/common"
|
|
"forge.cadoles.com/cadoles/bouncer/internal/proxy"
|
|
"forge.cadoles.com/cadoles/bouncer/internal/setup"
|
|
"forge.cadoles.com/cadoles/bouncer/internal/store"
|
|
"github.com/pkg/errors"
|
|
"github.com/urfave/cli/v2"
|
|
"gitlab.com/wpetit/goweb/logger"
|
|
)
|
|
|
|
func RunCommand() *cli.Command {
|
|
flags := common.Flags()
|
|
|
|
return &cli.Command{
|
|
Name: "run",
|
|
Usage: "Run the proxy server",
|
|
Flags: flags,
|
|
Action: func(ctx *cli.Context) error {
|
|
conf, err := common.LoadConfig(ctx)
|
|
if err != nil {
|
|
return errors.Wrap(err, "could not load configuration")
|
|
}
|
|
|
|
logger.SetFormat(logger.Format(conf.Logger.Format))
|
|
logger.SetLevel(logger.Level(conf.Logger.Level))
|
|
|
|
logger.Debug(ctx.Context, "using config", logger.F("config", conf))
|
|
|
|
projectVersion := ctx.String("projectVersion")
|
|
|
|
if conf.Proxy.Sentry.DSN != "" {
|
|
flushSentry, err := setup.SetupSentry(ctx.Context, conf.Proxy.Sentry, projectVersion)
|
|
if err != nil {
|
|
return errors.Wrap(err, "could not initialize sentry client")
|
|
}
|
|
|
|
defer flushSentry()
|
|
}
|
|
|
|
layers, err := setup.GetLayers(ctx.Context, conf)
|
|
if err != nil {
|
|
return errors.Wrap(err, "could not initialize director layers")
|
|
}
|
|
|
|
redisClient := setup.NewSharedClient(conf.Redis)
|
|
|
|
proxyRepository, err := setup.NewProxyRepository(ctx.Context, redisClient)
|
|
if err != nil {
|
|
return errors.Wrap(err, "could not initialize proxy repository")
|
|
}
|
|
|
|
layerRepository, err := setup.NewLayerRepository(ctx.Context, redisClient)
|
|
if err != nil {
|
|
return errors.Wrap(err, "could not initialize layer repository")
|
|
}
|
|
|
|
srv := proxy.NewServer(
|
|
proxy.WithServerConfig(conf.Proxy),
|
|
proxy.WithProxyRepository(proxyRepository),
|
|
proxy.WithLayerRepository(layerRepository),
|
|
proxy.WithDirectorLayers(layers...),
|
|
proxy.WithDirectorCacheTTL(time.Duration(*conf.Proxy.Cache.TTL)),
|
|
)
|
|
|
|
addrs, srvErrs := srv.Start(ctx.Context)
|
|
|
|
if observableProxyRepository, ok := proxyRepository.(store.Observable); ok {
|
|
logger.Info(ctx.Context, "observing proxy repository changes")
|
|
observableProxyRepository.Changes(ctx.Context, func(c store.Change) {
|
|
logger.Info(ctx.Context, "proxy change detected, clearing cache")
|
|
srv.ClearProxyCache()
|
|
})
|
|
}
|
|
|
|
if observableLayerRepository, ok := layerRepository.(store.Observable); ok {
|
|
logger.Info(ctx.Context, "observing layer repository changes")
|
|
observableLayerRepository.Changes(ctx.Context, func(c store.Change) {
|
|
logger.Info(ctx.Context, "layer change detected, clearing cache")
|
|
srv.ClearLayerCache()
|
|
})
|
|
}
|
|
|
|
// Clear director's cache on SIGUSR2
|
|
sig := make(chan os.Signal, 1)
|
|
signal.Notify(sig, syscall.SIGUSR2)
|
|
|
|
go func() {
|
|
for {
|
|
select {
|
|
case <-sig:
|
|
logger.Info(ctx.Context, "received sigusr2, clearing whole cache")
|
|
srv.ClearProxyCache()
|
|
srv.ClearLayerCache()
|
|
case <-ctx.Context.Done():
|
|
return
|
|
}
|
|
}
|
|
}()
|
|
|
|
select {
|
|
case addr := <-addrs:
|
|
url := fmt.Sprintf("http://%s", addr.String())
|
|
url = strings.Replace(url, "0.0.0.0", "127.0.0.1", 1)
|
|
|
|
logger.Info(ctx.Context, "listening", logger.F("url", url))
|
|
case err = <-srvErrs:
|
|
return errors.WithStack(err)
|
|
}
|
|
|
|
if err = <-srvErrs; err != nil {
|
|
return errors.WithStack(err)
|
|
}
|
|
|
|
return nil
|
|
},
|
|
}
|
|
}
|