new: Add parameters to specify the configuration file

This commit is contained in:
2024-12-19 16:27:24 +08:00
parent a646b3cb4f
commit 50c365a96a
2 changed files with 30 additions and 27 deletions

View File

@@ -1,13 +1,5 @@
package config
import (
"io"
"log"
"os"
"gopkg.in/yaml.v3"
)
type MConfig struct {
Host string `yaml:"host"`
Port int `yaml:"port"`
@@ -21,21 +13,3 @@ type MConfig struct {
}
var MConfigInstance *MConfig
func init() {
configFile, err := os.OpenFile("config.yaml", os.O_RDONLY, os.ModePerm)
if err != nil {
log.Fatal(err)
}
defer configFile.Close()
configByte, err := io.ReadAll(configFile)
if err != nil {
log.Fatal(err)
}
err = yaml.Unmarshal(configByte, &MConfigInstance)
if err != nil {
log.Fatal(err)
}
}