f8c7c2372da9041243d36dd4d9f7d3630e65bc08.svn-base 3.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156
  1. package setting
  2. import (
  3. "errors"
  4. "github.com/go-ini/ini"
  5. "github.com/jacobsa/go-serial/serial"
  6. "log"
  7. "os"
  8. "os/exec"
  9. "path/filepath"
  10. "runtime"
  11. "strings"
  12. "time"
  13. )
  14. type App struct {
  15. JwtSecret string
  16. PageSize int
  17. PrefixUrl string
  18. RuntimeRootPath string
  19. FileSavePath string
  20. FileMaxSize int // 普通文件上传的限制大小,单位byte, 最大单位1GB
  21. FileAllowType [] string // 允许上传的文件后缀名
  22. ImageSavePath string
  23. ImageMaxSize int
  24. ImageAllowExts []string
  25. ThumbnailSavePath string
  26. ThumbnailMaxWidth int
  27. ThumbnailMaxHeight int
  28. ExportSavePath string
  29. QrCodeSavePath string
  30. FontSavePath string
  31. LogSavePath string
  32. LogSaveName string
  33. LogFileExt string
  34. TimeFormat string
  35. }
  36. var AppSetting = &App{}
  37. type Server struct {
  38. RunMode string
  39. HttpPort int
  40. ReadTimeout time.Duration
  41. WriteTimeout time.Duration
  42. NoAuth int
  43. Mdns_servicename string
  44. Mdns_serviceport int
  45. HttpsPort int
  46. Https int
  47. CrtPath string
  48. KeyPath string
  49. MailStart string
  50. CronFormula string
  51. WXworkStart string
  52. WXworkAgentid string
  53. WXworkSecret string
  54. WXworkCorpid string
  55. ServerName string
  56. DisplayName string
  57. Description string
  58. EasName string
  59. EasPassword string
  60. EasUrl string
  61. }
  62. var ServerSetting = &Server{}
  63. var CommSetting = &serial.OpenOptions{}
  64. type Database struct {
  65. Type string
  66. User string
  67. Password string
  68. Host string
  69. Name string
  70. ShowXormlog bool
  71. ShowGetSqllog bool
  72. CacheApiSql bool
  73. TablePrefix string
  74. }
  75. var DatabaseSetting = &Database{}
  76. var cfg *ini.File
  77. var CurrentPath string
  78. func GetCurrentPath() (string, error) {
  79. file, err := exec.LookPath(os.Args[0])
  80. if err != nil {
  81. return "", err
  82. }
  83. path, err := filepath.Abs(file)
  84. if err != nil {
  85. return "", err
  86. }
  87. //fmt.Println("path111:", path)
  88. if runtime.GOOS == "windows" {
  89. path = strings.Replace(path, "\\", "/", -1)
  90. }
  91. //fmt.Println("path222:", path)
  92. i := strings.LastIndex(path, "/")
  93. if i < 0 {
  94. return "", errors.New(`Can't find "/" or "\".`)
  95. }
  96. //fmt.Println("path333:", path)
  97. return string(path[0 : i+1]), nil
  98. }
  99. func Setup(dir string) {
  100. var err error
  101. if strings.Trim(dir," ")=="" {
  102. CurrentPath, err = GetCurrentPath()
  103. }else {
  104. CurrentPath=dir
  105. }
  106. cfg, err = ini.Load(CurrentPath+"conf/app.ini")
  107. if err != nil {
  108. log.Fatalf("setting.Setup, fail to parse 'conf/app.ini': %v", err)
  109. }
  110. DatabaseSetting.ShowXormlog = false
  111. DatabaseSetting.ShowGetSqllog = false
  112. DatabaseSetting.CacheApiSql = false
  113. mapTo("app", AppSetting)
  114. mapTo("server", ServerSetting)
  115. mapTo("database", DatabaseSetting)
  116. mapTo("comm", CommSetting)
  117. AppSetting.ImageMaxSize = AppSetting.ImageMaxSize * 1024 * 1024
  118. ServerSetting.ReadTimeout = ServerSetting.ReadTimeout * time.Second
  119. ServerSetting.WriteTimeout = ServerSetting.ReadTimeout * time.Second
  120. ServerSetting.CrtPath = CurrentPath + ServerSetting.CrtPath
  121. ServerSetting.KeyPath = CurrentPath + ServerSetting.KeyPath
  122. }
  123. func mapTo(section string, v interface{}) {
  124. err := cfg.Section(section).MapTo(v)
  125. if err != nil {
  126. log.Fatalf("Cfg.MapTo RedisSetting err: %v", err)
  127. }
  128. }