73 lines
2.2 KiB
Go
73 lines
2.2 KiB
Go
package proto
|
||
|
||
import (
|
||
"encoding/json"
|
||
"fmt"
|
||
"os"
|
||
)
|
||
|
||
var Config ConfigStruct
|
||
|
||
type ConfigStruct struct {
|
||
REDIS_ADDR string `json:"redis_addr"`
|
||
TOKEN_USE_REDIS bool `json:"token_use_redis"`
|
||
REDIS_User_PW bool `json:"redis_user_pw"` // 是否使用密码
|
||
REDIS_PASSWORD string `json:"redis_password"`
|
||
REDIS_DB int `json:"redis_db"`
|
||
SERVER_PORT string `json:"SERVER_PORT" form:"SERVER_PORT"` // 服务器端口
|
||
DeviceInfo []DeviceInfo `json:"device_info" form:"device_info"` // 设备信息
|
||
}
|
||
|
||
type DeviceInfo struct {
|
||
ID int `json:"id" form:"id"`
|
||
Name string `json:"name" form:"name"`
|
||
IP string `json:"ip" form:"ip"`
|
||
Type string `json:"type" form:"type"`
|
||
Stream string `json:"stream" form:"stream"`
|
||
Control string `json:"control" form:"control"`
|
||
NextStop bool `json:"next_stop" form:"next_stop"` // 下一帧，是否停止
|
||
CheckFrameWidth int `json:"check_frame_width" form:"check_frame_width"` // 检测帧宽度
|
||
CheckFrameHeight int `json:"check_frame_height" form:"check_frame_height"` // 检测帧高度
|
||
LogFrame int `json:"log_frame" form:"log_frame"` // 日志帧间隔，默认值为0，即不输出日志
|
||
}
|
||
|
||
// 读取配置文件
|
||
func ReadConfig(path string) error {
|
||
//查看配置文件是否存在,不存在则创建
|
||
_, err := os.Stat(path)
|
||
if err != nil {
|
||
fmt.Println("Config file not found!")
|
||
//写入json文件
|
||
file, err := os.Create(path)
|
||
if err != nil {
|
||
fmt.Println("Error creating config file")
|
||
return err
|
||
}
|
||
defer file.Close()
|
||
encoder := json.NewEncoder(file)
|
||
err = encoder.Encode(&Config)
|
||
if err != nil {
|
||
fmt.Println("Error encoding config")
|
||
}
|
||
return err
|
||
}
|
||
|
||
//读json文件
|
||
file, err := os.Open(path)
|
||
if err != nil {
|
||
fmt.Println("Error opening config file")
|
||
return err
|
||
}
|
||
defer file.Close()
|
||
decoder := json.NewDecoder(file)
|
||
err = decoder.Decode(&Config)
|
||
if err != nil {
|
||
fmt.Println("Error decoding config")
|
||
} else {
|
||
if Config.SERVER_PORT == "" {
|
||
Config.SERVER_PORT = "5002" // 默认端口
|
||
}
|
||
}
|
||
return err
|
||
}
|