main.go 1.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081
  1. package main
  2. import (
  3. "flag"
  4. "fmt"
  5. "github.com/gin-gonic/gin"
  6. //"github.com/zsais/go-gin-prometheus"
  7. "lzyd-user-api/common"
  8. "lzyd-user-api/router"
  9. "os"
  10. "path"
  11. "github.com/TingYunAPM/go"
  12. )
  13. // @title lzyd user server api
  14. // @version 1.0
  15. // @description lzyd user server api
  16. // @title gin-blog API
  17. // @version 0.0.1
  18. // @description This is a gin blog example
  19. // @securityDefinitions.apikey ApiKeyAuth
  20. // @in header
  21. // @name x-app-id
  22. // @BasePath /
  23. func main() {
  24. var err error
  25. fPath, _ := os.Getwd()
  26. fPath = path.Join(fPath, "conf")
  27. configPath := flag.String("c", fPath, "config file path")
  28. flag.Parse()
  29. err = common.LoadConfigInformation(*configPath)
  30. if err != nil {
  31. return
  32. }
  33. //初始化tingyun: 应用名称、帐号等在tingyun.json中配置
  34. tingyun.AppInit("tingyun.json")
  35. defer tingyun.AppStop()
  36. //原有业务逻辑
  37. if common.ServerInfo.Mode == "debug" {
  38. gin.SetMode(gin.DebugMode)
  39. } else if common.ServerInfo.Mode == "release" {
  40. gin.SetMode(gin.ReleaseMode)
  41. }
  42. defer func() {
  43. _ = common.LogSync()
  44. }()
  45. //router init
  46. router := router.InitRouter()
  47. //p := ginprometheus.NewPrometheus("gin")
  48. //p.Use(router)
  49. server := common.ServerInfo
  50. serverInfo := server.Host + ":" + server.Port
  51. // restart
  52. if server.EnableHttps {
  53. fmt.Println("server start https")
  54. err := router.RunTLS(serverInfo, server.CertFile, server.KeyFile)
  55. fmt.Println(err)
  56. if err != nil {
  57. common.LogError("server start failed: " + err.Error())
  58. }
  59. } else {
  60. fmt.Printf("server start info: %s\n", serverInfo)
  61. common.LogInfo("server start info: " + serverInfo)
  62. common.LogSync()
  63. err := router.Run(serverInfo)
  64. if err != nil {
  65. common.LogError("server start failed: " + err.Error())
  66. }
  67. }
  68. }