new: Add parameters to specify the configuration file

This commit is contained in:
yv1ing
2024-12-19 16:27:24 +08:00
parent eee7943d2e
commit d2c7b5444d
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)
}
}