优化 yaml本身就兼容json....

This commit is contained in:
xing 2023-06-01 22:46:46 +08:00
parent a08b9538ae
commit e4fa93e893

View File

@ -2,14 +2,12 @@ package config
import (
"encoding/json"
"errors"
"fmt"
"github.com/fthvgb1/wp-go/safety"
"gopkg.in/yaml.v2"
"io"
"net/http"
"os"
"path/filepath"
"strings"
"time"
)
@ -106,23 +104,7 @@ func InitConfig(conf string) error {
return err
}
var c Config
switch strings.ToLower(filepath.Ext(conf)) {
case ".yaml":
err = yaml.Unmarshal(file, &c)
case ".json":
err = jsonToYaml(file, &c)
default:
err = yaml.Unmarshal(file, &c)
if err == nil {
break
}
err = jsonToYaml(file, &c)
if err == nil {
break
}
return errors.Join(errors.New("can't parse the config"), err)
}
if err != nil {
return err
}
@ -146,7 +128,7 @@ func jsonToYaml[T any](b []byte, c T) error {
type Dsn struct {
Host string `yaml:"host" json:"host,omitempty"`
Port json.Number `yaml:"port" json:"port,omitempty"`
Port string `yaml:"port" json:"port,omitempty"`
Db string `yaml:"db" json:"db,omitempty"`
User string `yaml:"user" json:"user,omitempty"`
Password string `yaml:"password" json:"password,omitempty"`