main.go 1.5 KB

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