mirror of
https://github.com/openimsdk/open-im-server.git
synced 2026-05-01 07:35:58 +08:00
feat: Introduce Language-Specific Comment Detection Tool and Standardize Log Filename Convention (#1992)
* feat: optimize openim reset code * feat: optimize openim reset code * feat: optimize openim reset code * feat: optimize openim reset code * feat: optimize openim reset code * feat: optimize openim reset code * feat: optimize openim reset code * feat: optimize openim reset code
This commit is contained in:
@@ -0,0 +1,35 @@
|
||||
package config
|
||||
|
||||
import (
|
||||
"flag"
|
||||
"log"
|
||||
"os"
|
||||
|
||||
"gopkg.in/yaml.v2"
|
||||
)
|
||||
|
||||
type Config struct {
|
||||
Directory string `yaml:"directory"`
|
||||
FileTypes []string `yaml:"file_types"`
|
||||
Languages []string `yaml:"languages"`
|
||||
}
|
||||
|
||||
func ParseConfig() (Config, error) {
|
||||
var configPath string
|
||||
flag.StringVar(&configPath, "config", "./", "Path to config file")
|
||||
flag.Parse()
|
||||
|
||||
var config Config
|
||||
if configPath != "" {
|
||||
configFile, err := os.ReadFile(configPath)
|
||||
if err != nil {
|
||||
return Config{}, err
|
||||
}
|
||||
if err := yaml.Unmarshal(configFile, &config); err != nil {
|
||||
return Config{}, err
|
||||
}
|
||||
} else {
|
||||
log.Fatal("Config file must be provided")
|
||||
}
|
||||
return config, nil
|
||||
}
|
||||
Reference in New Issue
Block a user