This commit is contained in:
ginuerzh
2023-12-20 22:00:23 +08:00
parent 7264fca4f8
commit ca077d418a
3 changed files with 64 additions and 394 deletions

View File

@ -4,6 +4,7 @@ import (
"encoding/json"
"net/http"
"os"
"strings"
"github.com/go-gost/core/logger"
"github.com/go-gost/x/config"
@ -20,8 +21,14 @@ type program struct {
func (p *program) Init(env svc.Environment) error {
cfg := &config.Config{}
if cfgFile != "" {
if err := json.Unmarshal([]byte(cfgFile), cfg); err != nil {
cfgFile = strings.TrimSpace(cfgFile)
if strings.HasPrefix(cfgFile, "{") && strings.HasSuffix(cfgFile, "}") {
if err := json.Unmarshal([]byte(cfgFile), cfg); err != nil {
return err
}
} else {
if err := cfg.ReadFile(cfgFile); err != nil {
logger.Default().Error(err)
return err
}
}