settings.go 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081
  1. package settings
  2. import (
  3. "gopkg.in/ini.v1"
  4. "log"
  5. )
  6. var Conf *ini.File
  7. type Server struct {
  8. HttpPort string
  9. RunMode string
  10. WebSocketToken string
  11. JwtSecret string
  12. HTTPChallengePort string
  13. Email string
  14. Database string
  15. }
  16. var ServerSettings = &Server{
  17. HttpPort: "9000",
  18. RunMode: "debug",
  19. HTTPChallengePort: "9180",
  20. Database: "database",
  21. }
  22. var ConfPath string
  23. var sections = map[string]interface{}{
  24. "server": ServerSettings,
  25. }
  26. func Init(confPath string) {
  27. ConfPath = confPath
  28. Setup()
  29. }
  30. func Setup() {
  31. var err error
  32. Conf, err = ini.LooseLoad(ConfPath)
  33. if err != nil {
  34. log.Printf("setting.Setup: %v", err)
  35. } else {
  36. MapTo()
  37. }
  38. }
  39. func MapTo() {
  40. for k, v := range sections {
  41. mapTo(k, v)
  42. }
  43. }
  44. func ReflectFrom() {
  45. for k, v := range sections {
  46. reflectFrom(k, v)
  47. }
  48. }
  49. func mapTo(section string, v interface{}) {
  50. err := Conf.Section(section).MapTo(v)
  51. if err != nil {
  52. log.Fatalf("Cfg.MapTo %s err: %v", section, err)
  53. }
  54. }
  55. func reflectFrom(section string, v interface{}) {
  56. err := Conf.Section(section).ReflectFrom(v)
  57. if err != nil {
  58. log.Fatalf("Cfg.ReflectFrom %s err: %v", section, err)
  59. }
  60. }
  61. func Save() (err error) {
  62. err = Conf.SaveTo(ConfPath)
  63. if err != nil {
  64. return
  65. }
  66. Setup()
  67. return
  68. }