main.go 1.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273
  1. package main
  2. import (
  3. "dsbqj-admin/app/router"
  4. "dsbqj-admin/app/task"
  5. "dsbqj-admin/model/mongo"
  6. "dsbqj-admin/pkg/cache"
  7. "dsbqj-admin/pkg/logger"
  8. "dsbqj-admin/pkg/validator"
  9. "fmt"
  10. "github.com/gin-gonic/gin"
  11. "github.com/joho/godotenv"
  12. "log"
  13. "net/http"
  14. "os"
  15. "time"
  16. )
  17. // @title 后端接口API文档
  18. // @version 1.0
  19. // @description 这里展示所有当前web端API接口信息
  20. // @host localhost:30101
  21. // @BasePath /web/v1
  22. // @securityDefinitions.apikey cxy_token
  23. // @name __CXY_TOKEN_
  24. // @in header
  25. // @securityDefinitions.apikey cxy_uid
  26. // @name __CXY_UID_
  27. // @in header
  28. func main() {
  29. err := godotenv.Load()
  30. if err != nil {
  31. log.Fatal("Error loading .env file")
  32. return
  33. }
  34. validator.Init()
  35. logger.Init("")
  36. cache.Redis()
  37. // 连接数据库
  38. //mysql.Database(os.Getenv("MYSQL_DSN"))
  39. mongo.Database(os.Getenv("MONGO_DSN"))
  40. // 启动任务
  41. //task.VersionInit(task.VersionInit, 5*time.Minute)
  42. task.Run(task.SubscribeInit(), 1*time.Second)
  43. if os.Getenv("GIN_MODE") == "release" {
  44. gin.SetMode(gin.ReleaseMode)
  45. }
  46. // 装载路由
  47. routersInit := router.NewRouter()
  48. readTimeout := 60 * time.Second
  49. writeTimeout := 200 * time.Second
  50. endPoint := fmt.Sprintf(":%s", os.Getenv("LISTEN_PORT"))
  51. maxHeaderBytes := 1 << 20
  52. server := &http.Server{
  53. Addr: endPoint,
  54. Handler: routersInit,
  55. ReadTimeout: readTimeout,
  56. WriteTimeout: writeTimeout,
  57. MaxHeaderBytes: maxHeaderBytes,
  58. }
  59. logger.Info("start http version listening %s", endPoint)
  60. err = server.ListenAndServe()
  61. fmt.Println(err)
  62. }