index.go 1.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. package mysql
  2. import (
  3. "fmt"
  4. "os"
  5. "pbx-api-gin/internal/pkg/configs"
  6. "pbx-api-gin/pkg/lfshook"
  7. "syscall"
  8. _ "github.com/go-sql-driver/mysql"
  9. "github.com/sirupsen/logrus"
  10. "xorm.io/xorm"
  11. "xorm.io/xorm/log"
  12. )
  13. var DBOrmInstance *xorm.Engine
  14. func CreateDBInstance() {
  15. var err error
  16. // DBOrmInstance, err = xorm.NewEngine("sqlite3", "playcall.db")
  17. url := fmt.Sprintf("%s:%s@tcp(%s:3306)/%s?charset=utf8",
  18. configs.ConfigGlobal.MysqlDBUser,
  19. // "coovox_admin",
  20. configs.ConfigGlobal.MysqlDBSecret,
  21. // "ZycooCoovoxDba42",
  22. configs.ConfigGlobal.MysqlDBHost,
  23. configs.ConfigGlobal.MysqlDBName,
  24. )
  25. lfshook.NewLogger().Infof("mysql url %s", url)
  26. DBOrmInstance, err = xorm.NewEngine("mysql", url)
  27. if err != nil {
  28. lfshook.NewLogger().Panic(err)
  29. return
  30. }
  31. err = DBOrmInstance.Ping()
  32. if err != nil {
  33. lfshook.NewLogger().Error(err)
  34. syscall.Kill(syscall.Getpid(), syscall.SIGINT)
  35. return
  36. }
  37. //DBOrmInstance.ShowSQL(true)
  38. if configs.ConfigGlobal.LogLevel >= logrus.DebugLevel {
  39. DBOrmInstance.ShowSQL(true)
  40. } else {
  41. info, err := os.Open(configs.ConfigGlobal.LogInfoPath)
  42. if err != nil {
  43. println(err.Error())
  44. return
  45. }
  46. DBOrmInstance.SetLogger(log.NewSimpleLogger(info))
  47. }
  48. }