feat: add http_proxy config for fiber

OpenTreeHole · Jul 9, 2024 · e5ec837 · e5ec837
1 parent 5847034
commit e5ec837
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/config.go b/config.go
@@ -16,6 +16,7 @@ var Config struct {
 	ProxyEmail    string `env:"PROXY_EMAIL"`
 	HostRewrite   string `env:"HOST_REWRITE"`
 	Token         string `env:"TOKEN"`
+	HttpProxyUrl  string `env:"HTTP_PROXY_URL"`
 }
 
 var Token struct {

diff --git a/utils.go b/utils.go
@@ -4,6 +4,7 @@ import (
 	"encoding/json"
 	"fmt"
 	"github.com/gofiber/fiber/v2"
+	"github.com/valyala/fasthttp/fasthttpproxy"
 	"time"
 )
 
@@ -45,5 +46,6 @@ func LskyBaseAgent(agent *fiber.Agent, method string, path string) *fiber.Agent
 	req := agent.Request()
 	req.Header.SetMethod(method)
 	req.SetRequestURI(Config.ProxyUrl + path)
+	agent.Dial = fasthttpproxy.FasthttpHTTPDialer(Config.HttpProxyUrl)
 	return agent
 }