Q3RCON_DEBUG env var added

This commit is contained in:
onyx-and-iris 2024-03-20 17:03:35 +00:00
parent 7138515904
commit 94a683fb3f

View File

@ -2,10 +2,12 @@ package main
import ( import (
"fmt" "fmt"
"log"
"os" "os"
"strconv"
"strings" "strings"
log "github.com/sirupsen/logrus"
"github.com/onyx-and-iris/q3rcon-proxy/pkg/udpproxy" "github.com/onyx-and-iris/q3rcon-proxy/pkg/udpproxy"
) )
@ -26,9 +28,21 @@ func start(proxy string) {
} }
var ( var (
proxies, host string proxies, host, debug string
) )
func getenvInt(key string) (int, error) {
s := os.Getenv(key)
if s == "" {
return 0, nil
}
v, err := strconv.Atoi(s)
if err != nil {
return 0, err
}
return v, nil
}
func init() { func init() {
proxies = os.Getenv("Q3RCON_PROXY") proxies = os.Getenv("Q3RCON_PROXY")
if proxies == "" { if proxies == "" {
@ -39,6 +53,18 @@ func init() {
if host == "" { if host == "" {
host = "0.0.0.0" host = "0.0.0.0"
} }
debug, err := getenvInt("Q3RCON_DEBUG")
if err != nil {
log.Fatal(err)
}
if debug == 1 {
log.SetLevel(log.DebugLevel)
} else {
log.SetLevel(log.InfoLevel)
}
} }
func main() { func main() {