feat: support max timeout env (#175)

Co-Authored-By: Minghan Zhang <[email protected]>

globals/variables.go

@@ -3,13 +3,16 @@ package globals
 import (
 	"net/url"
 	"strings"
+	"time"
 
 	"github.com/gin-gonic/gin"
 )
 
 const ChatMaxThread = 5
 const AnonymousMaxThread = 1
 
+var HttpMaxTimeout = 30 * time.Minute
+
 var AllowedOrigins []string
 
 var DebugMode bool

utils/config.go

@@ -3,10 +3,12 @@ package utils
 import (
 	"chat/globals"
 	"fmt"
+	"strings"
+	"time"
+
 	"github.com/gin-contrib/static"
 	"github.com/gin-gonic/gin"
 	"github.com/spf13/viper"
-	"strings"
 )
 
 var configFile = "config/config.yaml"
@@ -34,6 +36,11 @@ func ReadConf() {
 
 	viper.AutomaticEnv()
 	viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_"))
+
+	if timeout := viper.GetInt("max_timeout"); timeout > 0 {
+		globals.HttpMaxTimeout = time.Second * time.Duration(timeout)
+		globals.Debug(fmt.Sprintf("[service] http client timeout set to %ds from env", timeout))
+	}
 }
 
 func NewEngine() *gin.Engine {

utils/net.go

@@ -6,22 +6,20 @@ import (
 	"context"
 	"crypto/tls"
 	"fmt"
-	"github.com/goccy/go-json"
-	"golang.org/x/net/proxy"
 	"io"
 	"net"
 	"net/http"
 	"net/url"
 	"runtime/debug"
 	"strings"
-	"time"
-)
 
-var maxTimeout = 30 * time.Minute
+	"github.com/goccy/go-json"
+	"golang.org/x/net/proxy"
+)
 
 func newClient(c []globals.ProxyConfig) *http.Client {
 	client := &http.Client{
-		Timeout: maxTimeout,
+		Timeout: globals.HttpMaxTimeout,
 		Transport: &http.Transport{
 			TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
 		},