config.go 1.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990
  1. package config
  2. import (
  3. "encoding/json"
  4. "fmt"
  5. "gopkg.in/yaml.v3"
  6. "io/ioutil"
  7. "os"
  8. "sync"
  9. )
  10. var (
  11. instance *config
  12. once sync.Once
  13. DevConfig Devs
  14. )
  15. // 初始化配置,确保只初始化一次
  16. func init() {
  17. once.Do(func() {
  18. loadConfig() // 加载配置文件
  19. })
  20. }
  21. // 重新加载配置文件
  22. func ReloadConfig() {
  23. loadConfig() // 加载配置文件
  24. }
  25. // 加载配置文件
  26. func loadConfig() {
  27. var conf config
  28. filePath := "./config/config.yaml"
  29. if f, err := os.Open(filePath); err != nil {
  30. panic(err)
  31. } else {
  32. err := yaml.NewDecoder(f).Decode(&conf)
  33. if err != nil {
  34. panic(err)
  35. }
  36. }
  37. instance = &conf
  38. }
  39. // 获取配置文档实例
  40. func Instance() *config {
  41. return instance
  42. }
  43. type config struct {
  44. ProjectName string `yaml:"projectName"`
  45. Logger logger `yaml:"logger"`
  46. Mqtt mqtt `yaml:"mqtt"`
  47. }
  48. type logger struct {
  49. Path string `yaml:"path"`
  50. Name string `yaml:"name"`
  51. Level string `yaml:"level"`
  52. }
  53. type mqtt struct {
  54. Server string `yaml:"server"`
  55. Id string `yaml:"id"`
  56. User string `yaml:"user"`
  57. Password string `yaml:"password"`
  58. }
  59. func LoadDevConfig() error {
  60. // 读取配置文件
  61. data, err := ioutil.ReadFile("./dev/devs.json")
  62. if err != nil {
  63. return fmt.Errorf("读取配置文件失败: %v", err)
  64. }
  65. // 解析 JSON 数据到结构体
  66. if err := json.Unmarshal(data, &DevConfig); err != nil {
  67. return fmt.Errorf("解析 JSON 配置失败: %v", err)
  68. }
  69. return nil
  70. }
  71. type Devs struct {
  72. Screens []*Screens `json:"ledScreens"`
  73. }
  74. type Screens struct {
  75. Sn string `json:"sn"`
  76. ScreensName string `json:"screensName"`
  77. }