package proto import ( "encoding/json" "fmt" "os" ) var Config ConfigStruct var Url_map = map[string]bool{} // 不需要token验证的url var SigningKey = []byte{} type ConfigStruct struct { DB int `json:"db"` // 0: mysql, 1: pg MYSQL_DSN string `json:"mysql_dsn"` PG_DSN string `json:"pg_dsn"` TOKEN_SECRET string `json:"token_secret"` 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" // 默认端口 } } //将配置写入redis jsonData, err2 := json.Marshal(Config) if err2 != nil { fmt.Println("ReadConfigToSetSystem Error encoding config,err :", err2) } SigningKey = []byte(Config.TOKEN_SECRET) if Config.TOKEN_SECRET == "" { err = fmt.Errorf("token secret is empty: %s", string(jsonData)) } return err }