mirror of
https://github.com/weloe/token-go.git
synced 2025-10-05 15:36:50 +08:00
refactor: update config.NewConfig to ReadConfig
This commit is contained in:
@@ -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
|
||||
|
@@ -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)
|
||||
}
|
||||
|
@@ -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
|
||||
}
|
||||
|
Reference in New Issue
Block a user