refactor: update config.NewConfig to ReadConfig

This commit is contained in:
weloe
2023-06-09 19:59:56 +08:00
parent 446789e261
commit 4aa6df57df
3 changed files with 8 additions and 8 deletions

View File

@@ -44,8 +44,8 @@ func (c *FileConfig) parse(confName string) (err error) {
return err
}
// NewConfig create from file.
func NewConfig(confName string) (ConfigInterface, error) {
// ReadConfig create from file.
func ReadConfig(confName string) (*FileConfig, error) {
c := &FileConfig{}
err := c.parse(confName)
return c, err

View File

@@ -3,21 +3,21 @@ package config
import "testing"
func TestNewConfig(t *testing.T) {
config, err := NewConfig("../examples/token_conf.yaml")
config, err := ReadConfig("../examples/token_conf.yaml")
if err != nil {
t.Errorf("read error: %v", err)
}
tokenConfig := config.(*FileConfig).TokenConfig
tokenConfig := config.TokenConfig
t.Log(tokenConfig)
}
func TestNewIniConfig(t *testing.T) {
config, err := NewConfig("../examples/token_conf.ini")
config, err := ReadConfig("../examples/token_conf.ini")
if err != nil {
t.Errorf("read error: %v", err)
}
tokenConfig := config.(*FileConfig).TokenConfig
tokenConfig := config.TokenConfig
t.Log(tokenConfig)
}

View File

@@ -71,11 +71,11 @@ func InitWithFile(conf string, adapter persist.Adapter) (*Enforcer, error) {
if conf == "" || adapter == nil {
return nil, errors.New("InitWithFile() failed: parameters cannot be nil")
}
newConfig, err := config.NewConfig(conf)
newConfig, err := config.ReadConfig(conf)
if err != nil {
return nil, err
}
enforcer, err := InitWithConfig(newConfig.(*config.FileConfig).TokenConfig, adapter)
enforcer, err := InitWithConfig(newConfig.TokenConfig, adapter)
enforcer.conf = conf
return enforcer, err
}