Compare commits
2 Commits
a08b9538ae
...
3493eee919
Author | SHA1 | Date | |
---|---|---|---|
3493eee919 | |||
e4fa93e893 |
|
@ -1,15 +1,12 @@
|
||||||
package config
|
package config
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
|
||||||
"errors"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/fthvgb1/wp-go/safety"
|
"github.com/fthvgb1/wp-go/safety"
|
||||||
"gopkg.in/yaml.v2"
|
"gopkg.in/yaml.v2"
|
||||||
"io"
|
"io"
|
||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
@ -106,23 +103,7 @@ func InitConfig(conf string) error {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
var c Config
|
var c Config
|
||||||
switch strings.ToLower(filepath.Ext(conf)) {
|
err = yaml.Unmarshal(file, &c)
|
||||||
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 {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
@ -130,27 +111,13 @@ func InitConfig(conf string) error {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func jsonToYaml[T any](b []byte, c T) error {
|
|
||||||
var v map[string]any
|
|
||||||
err := json.Unmarshal(b, &v)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
marshal, er := yaml.Marshal(v)
|
|
||||||
if er != nil {
|
|
||||||
return er
|
|
||||||
}
|
|
||||||
err = yaml.Unmarshal(marshal, c)
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
type Dsn struct {
|
type Dsn struct {
|
||||||
Host string `yaml:"host" json:"host,omitempty"`
|
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"`
|
Db string `yaml:"db" json:"db,omitempty"`
|
||||||
User string `yaml:"user" json:"user,omitempty"`
|
User string `yaml:"user" json:"user,omitempty"`
|
||||||
Password string `yaml:"password" json:"password,omitempty"`
|
Password string `yaml:"password" json:"password,omitempty"`
|
||||||
Charset string `yaml:"charset" json:"charset,omitempty"`
|
Charset string `yaml:"charset" json:"charset,omitempty"`
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m Dsn) GetDsn() string {
|
func (m Dsn) GetDsn() string {
|
||||||
|
|
Loading…
Reference in New Issue
Block a user