Merge pull request #155 from MindHunter86/feature/full-bypass

feature/full-bypass
MindHunter86 · Mar 28, 2024 · 0a9c977 · 0a9c977
2 parents 3680410 + 6079509
commit 0a9c977
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/app/middleware.go b/app/middleware.go
@@ -106,7 +106,7 @@ func (m *App) fbMidAppFakeQuality(ctx *fiber.Ctx) error {
 
 // if return value == true - Balance() will be skipped
 func (m *App) fbMidAppBalancerLottery(_ *fiber.Ctx) bool {
-	return m.runtime.Config.Get(runtime.ParamLottery).(int) < rand.Intn(99)+1 // skipcq: GSC-G404 math/rand is enough
+	return gCli.Bool("balancer-full-bypass") || m.runtime.Config.Get(runtime.ParamLottery).(int) < rand.Intn(99)+1 // skipcq: GSC-G404 math/rand is enough
 }
 
 func (m *App) fbMidAppBalance(ctx *fiber.Ctx) (e error) {

diff --git a/main.go b/main.go
@@ -187,6 +187,10 @@ func main() {
 			Usage: "max fails for one request; max value - 10",
 			Value: 3,
 		},
+		&cli.BoolFlag{
+			Name:  "balancer-full-bypass",
+			Usage: "use X-Server header as a balance target",
+		},
 
 		// ...
 		&cli.DurationFlag{