blackteay/gf
1
0
Fork 0
mirror of https://github.com/gogf/gf.git synced 2025-04-05 03:05:05 +08:00
Code Issues Projects Releases Wiki Activity

os/gcfg: fix file searching issue always returning the configuration file of pwd (#3592)

Browse Source
This commit is contained in:
John Guo 2024-05-22 21:14:16 +08:00 committed by GitHub
parent 1e8f4287af
commit 2e471662f4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 71 additions and 72 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
os/gcfg/gcfg_adapter_file.go
View File

@ -25,10 +25,10 @@ import (
// AdapterFile implements interface Adapter using file.
type AdapterFile struct {
defaultName string // Default configuration file name.
searchPaths *garray.StrArray // Searching path array.
jsonMap *gmap.StrAnyMap // The pared JSON objects for configuration files.
violenceCheck bool // Whether it does violence check in value index searching. It affects the performance when set true(false in default).
defaultFileNameOrPath string // Default configuration file name or file path.
searchPaths *garray.StrArray // Searching path array.
jsonMap *gmap.StrAnyMap // The pared JSON objects for configuration files.
violenceCheck bool // Whether it does violence check in value index searching. It affects the performance when set true(false in default).
}
const (
@ -53,23 +53,23 @@ var (
// NewAdapterFile returns a new configuration management object.
// The parameter `file` specifies the default configuration file name for reading.
func NewAdapterFile(file ...string) (*AdapterFile, error) {
func NewAdapterFile(fileNameOrPath ...string) (*AdapterFile, error) {
var (
err error
name = DefaultConfigFileName
err error
usedFileNameOrPath = DefaultConfigFileName
)
if len(file) > 0 {
name = file[0]
if len(fileNameOrPath) > 0 {
usedFileNameOrPath = fileNameOrPath[0]
} else {
// Custom default configuration file name from command line or environment.
if customFile := command.GetOptWithEnv(commandEnvKeyForFile); customFile != "" {
name = customFile
usedFileNameOrPath = customFile
}
}
config := &AdapterFile{
defaultName: name,
searchPaths: garray.NewStrArray(true),
jsonMap: gmap.NewStrAnyMap(true),
defaultFileNameOrPath: usedFileNameOrPath,
searchPaths: garray.NewStrArray(true),
jsonMap: gmap.NewStrAnyMap(true),
}
// Customized dir path from env/cmd.
if customPath := command.GetOptWithEnv(commandEnvKeyForPath); customPath != "" {
@ -120,13 +120,13 @@ func (a *AdapterFile) SetViolenceCheck(check bool) {
}
// SetFileName sets the default configuration file name.
func (a *AdapterFile) SetFileName(name string) {
a.defaultName = name
func (a *AdapterFile) SetFileName(fileNameOrPath string) {
a.defaultFileNameOrPath = fileNameOrPath
}
// GetFileName returns the default configuration file name.
func (a *AdapterFile) GetFileName() string {
return a.defaultName
return a.defaultFileNameOrPath
}
// Get retrieves and returns value by specified `pattern`.
@ -200,7 +200,7 @@ func (a *AdapterFile) Dump() {
// Available checks and returns whether configuration of given `file` is available.
func (a *AdapterFile) Available(ctx context.Context, fileName ...string) bool {
checkFileName := gutil.GetOrDefaultStr(a.defaultName, fileName...)
checkFileName := gutil.GetOrDefaultStr(a.defaultFileNameOrPath, fileName...)
// Custom configuration content exists.
if a.GetContent(checkFileName) != "" {
return true
@ -227,26 +227,26 @@ func (a *AdapterFile) autoCheckAndAddMainPkgPathToSearchPaths() {
// getJson returns a *gjson.Json object for the specified `file` content.
// It would print error if file reading fails. It returns nil if any error occurs.
func (a *AdapterFile) getJson(fileName ...string) (configJson *gjson.Json, err error) {
func (a *AdapterFile) getJson(fileNameOrPath ...string) (configJson *gjson.Json, err error) {
var (
usedFileName = a.defaultName
usedFileNameOrPath = a.defaultFileNameOrPath
)
if len(fileName) > 0 && fileName[0] != "" {
usedFileName = fileName[0]
if len(fileNameOrPath) > 0 && fileNameOrPath[0] != "" {
usedFileNameOrPath = fileNameOrPath[0]
} else {
usedFileName = a.defaultName
usedFileNameOrPath = a.defaultFileNameOrPath
}
// It uses json map to cache specified configuration file content.
result := a.jsonMap.GetOrSetFuncLock(usedFileName, func() interface{} {
result := a.jsonMap.GetOrSetFuncLock(usedFileNameOrPath, func() interface{} {
var (
content string
filePath string
)
// The configured content can be any kind of data type different from its file type.
isFromConfigContent := true
if content = a.GetContent(usedFileName); content == "" {
if content = a.GetContent(usedFileNameOrPath); content == "" {
isFromConfigContent = false
filePath, err = a.GetFilePath(usedFileName)
filePath, err = a.GetFilePath(usedFileNameOrPath)
if err != nil {
return nil
}
@ -279,7 +279,7 @@ func (a *AdapterFile) getJson(fileName ...string) (configJson *gjson.Json, err e
// any changes of this file will refresh its cache in Config object.
if filePath != "" && !gres.Contains(filePath) {
_, err = gfsnotify.Add(filePath, func(event *gfsnotify.Event) {
a.jsonMap.Remove(usedFileName)
a.jsonMap.Remove(usedFileNameOrPath)
})
if err != nil {
return nil

40
os/gcfg/gcfg_adapter_file_content.go
View File

@ -14,58 +14,58 @@ import (
// SetContent sets customized configuration content for specified `file`.
// The `file` is unnecessary param, default is DefaultConfigFile.
func (a *AdapterFile) SetContent(content string, file ...string) {
name := DefaultConfigFileName
if len(file) > 0 {
name = file[0]
func (a *AdapterFile) SetContent(content string, fileNameOrPath ...string) {
var usedFileNameOrPath = DefaultConfigFileName
if len(fileNameOrPath) > 0 {
usedFileNameOrPath = fileNameOrPath[0]
}
// Clear file cache for instances which cached `name`.
localInstances.LockFunc(func(m map[string]interface{}) {
if customConfigContentMap.Contains(name) {
if customConfigContentMap.Contains(usedFileNameOrPath) {
for _, v := range m {
if configInstance, ok := v.(*Config); ok {
if fileConfig, ok := configInstance.GetAdapter().(*AdapterFile); ok {
fileConfig.jsonMap.Remove(name)
fileConfig.jsonMap.Remove(usedFileNameOrPath)
}
}
}
}
customConfigContentMap.Set(name, content)
customConfigContentMap.Set(usedFileNameOrPath, content)
})
}
// GetContent returns customized configuration content for specified `file`.
// The `file` is unnecessary param, default is DefaultConfigFile.
func (a *AdapterFile) GetContent(file ...string) string {
name := DefaultConfigFileName
if len(file) > 0 {
name = file[0]
func (a *AdapterFile) GetContent(fileNameOrPath ...string) string {
var usedFileNameOrPath = DefaultConfigFileName
if len(fileNameOrPath) > 0 {
usedFileNameOrPath = fileNameOrPath[0]
}
return customConfigContentMap.Get(name)
return customConfigContentMap.Get(usedFileNameOrPath)
}
// RemoveContent removes the global configuration with specified `file`.
// If `name` is not passed, it removes configuration of the default group name.
func (a *AdapterFile) RemoveContent(file ...string) {
name := DefaultConfigFileName
if len(file) > 0 {
name = file[0]
func (a *AdapterFile) RemoveContent(fileNameOrPath ...string) {
var usedFileNameOrPath = DefaultConfigFileName
if len(fileNameOrPath) > 0 {
usedFileNameOrPath = fileNameOrPath[0]
}
// Clear file cache for instances which cached `name`.
localInstances.LockFunc(func(m map[string]interface{}) {
if customConfigContentMap.Contains(name) {
if customConfigContentMap.Contains(usedFileNameOrPath) {
for _, v := range m {
if configInstance, ok := v.(*Config); ok {
if fileConfig, ok := configInstance.GetAdapter().(*AdapterFile); ok {
fileConfig.jsonMap.Remove(name)
fileConfig.jsonMap.Remove(usedFileNameOrPath)
}
}
}
customConfigContentMap.Remove(name)
customConfigContentMap.Remove(usedFileNameOrPath)
}
})
intlog.Printf(context.TODO(), `RemoveContent: %s`, name)
intlog.Printf(context.TODO(), `RemoveContent: %s`, usedFileNameOrPath)
}
// ClearContent removes all global configuration contents.

51
os/gcfg/gcfg_adapter_file_path.go
View File

@ -174,10 +174,9 @@ func (a *AdapterFile) GetPaths() []string {
return a.searchPaths.Slice()
}
// doGetFilePath returns the absolute configuration file path for the given filename by `file`.
// If `file` is not passed, it returns the configuration file path of the default name.
// It returns an empty `path` string and an error if the given `file` does not exist.
func (a *AdapterFile) doGetFilePath(fileName string) (filePath string) {
// doGetFilePath returns the absolute configuration file path for the given filename by `fileNameOrPath`.
// The `fileNameOrPath` can be either a file name or the file path.
func (a *AdapterFile) doGetFilePath(fileNameOrPath string) (filePath string) {
var (
tempPath string
resFile *gres.File
@ -186,7 +185,7 @@ func (a *AdapterFile) doGetFilePath(fileName string) (filePath string) {
// Searching resource manager.
if !gres.IsEmpty() {
for _, tryFolder := range resourceTryFolders {
tempPath = tryFolder + fileName
tempPath = tryFolder + fileNameOrPath
if resFile = gres.Get(tempPath); resFile != nil {
fileInfo, _ = resFile.Stat()
if fileInfo != nil && !fileInfo.IsDir() {
@ -198,7 +197,7 @@ func (a *AdapterFile) doGetFilePath(fileName string) (filePath string) {
a.searchPaths.RLockFunc(func(array []string) {
for _, searchPath := range array {
for _, tryFolder := range resourceTryFolders {
tempPath = searchPath + tryFolder + fileName
tempPath = searchPath + tryFolder + fileNameOrPath
if resFile = gres.Get(tempPath); resFile != nil {
fileInfo, _ = resFile.Stat()
if fileInfo != nil && !fileInfo.IsDir() {
@ -215,16 +214,12 @@ func (a *AdapterFile) doGetFilePath(fileName string) (filePath string) {
// Searching local file system.
if filePath == "" {
// Absolute path.
/*if filePath = gfile.RealPath(fileName); filePath != "" && !gfile.IsDir(filePath) {
return
}*/
a.searchPaths.RLockFunc(func(array []string) {
for _, searchPath := range array {
searchPath = gstr.TrimRight(searchPath, `\/`)
for _, tryFolder := range localSystemTryFolders {
relativePath := gstr.TrimRight(
gfile.Join(tryFolder, fileName),
gfile.Join(tryFolder, fileNameOrPath),
`\/`,
)
if filePath, _ = gspath.Search(searchPath, relativePath); filePath != "" &&
@ -235,9 +230,9 @@ func (a *AdapterFile) doGetFilePath(fileName string) (filePath string) {
}
})
}
// The `fileNameOrPath` can be a file path.
if filePath == "" {
// Absolute path.
if filePath = gfile.RealPath(fileName); filePath != "" && !gfile.IsDir(filePath) {
if filePath = gfile.RealPath(fileNameOrPath); filePath != "" && !gfile.IsDir(filePath) {
return
}
}
@ -247,23 +242,24 @@ func (a *AdapterFile) doGetFilePath(fileName string) (filePath string) {
// GetFilePath returns the absolute configuration file path for the given filename by `file`.
// If `file` is not passed, it returns the configuration file path of the default name.
// It returns an empty `path` string and an error if the given `file` does not exist.
func (a *AdapterFile) GetFilePath(fileName ...string) (filePath string, err error) {
func (a *AdapterFile) GetFilePath(fileNameOrPath ...string) (filePath string, err error) {
var (
fileExtName string
tempFileName string
usedFileName = a.defaultName
fileExtName string
tempFileNameOrPath string
usedFileNameOrPath = a.defaultFileNameOrPath
)
if len(fileName) > 0 {
usedFileName = fileName[0]
if len(fileNameOrPath) > 0 {
usedFileNameOrPath = fileNameOrPath[0]
}
fileExtName = gfile.ExtName(usedFileName)
if filePath = a.doGetFilePath(usedFileName); (filePath == "" || gfile.IsDir(filePath)) && !gstr.InArray(supportedFileTypes, fileExtName) {
fileExtName = gfile.ExtName(usedFileNameOrPath)
if filePath = a.doGetFilePath(usedFileNameOrPath); (filePath == "" || gfile.IsDir(filePath)) &&
!gstr.InArray(supportedFileTypes, fileExtName) {
// If it's not using default configuration or its configuration file is not available,
// it searches the possible configuration file according to the name and all supported
// file types.
for _, fileType := range supportedFileTypes {
tempFileName = fmt.Sprintf(`%s.%s`, usedFileName, fileType)
if filePath = a.doGetFilePath(tempFileName); filePath != "" {
tempFileNameOrPath = fmt.Sprintf(`%s.%s`, usedFileNameOrPath, fileType)
if filePath = a.doGetFilePath(tempFileNameOrPath); filePath != "" {
break
}
}
@ -275,12 +271,12 @@ func (a *AdapterFile) GetFilePath(fileName ...string) (filePath string, err erro
if !gstr.InArray(supportedFileTypes, fileExtName) {
buffer.WriteString(fmt.Sprintf(
`possible config files "%s" or "%s" not found in resource manager or following system searching paths:`,
usedFileName, fmt.Sprintf(`%s.%s`, usedFileName, gstr.Join(supportedFileTypes, "/")),
usedFileNameOrPath, fmt.Sprintf(`%s.%s`, usedFileNameOrPath, gstr.Join(supportedFileTypes, "/")),
))
} else {
buffer.WriteString(fmt.Sprintf(
`specified config file "%s" not found in resource manager or following system searching paths:`,
usedFileName,
usedFileNameOrPath,
))
}
a.searchPaths.RLockFunc(func(array []string) {
@ -297,7 +293,10 @@ func (a *AdapterFile) GetFilePath(fileName ...string) (filePath string, err erro
}
})
} else {
buffer.WriteString(fmt.Sprintf(`cannot find config file "%s" with no filePath configured`, usedFileName))
buffer.WriteString(fmt.Sprintf(
`cannot find config file "%s" with no filePath configured`,
usedFileNameOrPath,
))
}
err = gerror.NewCode(gcode.CodeNotFound, buffer.String())
}