lz8878
/
iot-spray-equipment
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update

master
李静 2023-07-11 14:35:11 +08:00
parent 230bc9858a
commit 78bacfa47d
1 changed files with 6 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
main.go
View File

@ -1,7 +1,6 @@
package main package main
import ( import (
"bytes"
"encoding/json" "encoding/json"
"flag" "flag"
"fmt" "fmt"
@ -18,19 +17,17 @@ import (
) )
var ( var (
configFilePath string configFile string
sprayEquipment = equipment.NewSprayEquipment("DC73AEE4CD274074BD9D01BCEE379A98") sprayEquipment = equipment.NewSprayEquipment("DC73AEE4CD274074BD9D01BCEE379A98")
) )
func init() { func init() {
flag.StringVar(&configFilePath, "config", "config.yml", "配置文件") flag.StringVar(&configFile, "config", "config.yml", "配置文件")
flag.Parse()
buf, err := os.ReadFile(configFilePath) fmt.Println(configFile)
if err != nil { viper.SetConfigFile(configFile)
panic(err) if err := viper.ReadInConfig(); err != nil {
}
viper.SetConfigType("yaml")
if err := viper.ReadConfig(bytes.NewBuffer(buf)); err != nil {
panic(err) panic(err)
} }