eta_mini_ht_api/main.go

package main

import (
	_ "eta_mini_ht_api/common/component"
	logger "eta_mini_ht_api/common/component/log"
	"eta_mini_ht_api/common/exception"
	"eta_mini_ht_api/middleware"
	_ "eta_mini_ht_api/routers"
	_ "eta_mini_ht_api/task"
	"github.com/beego/beego/v2/server/web"
	"github.com/beego/beego/v2/server/web/filter/cors"
	"time"
)

func main() {
	if web.BConfig.RunMode == "dev" {
		web.BConfig.WebConfig.DirectoryIndex = true
		web.BConfig.WebConfig.StaticDir["/swagger"] = "swagger"
	}
	web.InsertFilter("*", web.BeforeRouter, cors.Allow(&cors.Options{
		AllowAllOrigins:  true, // 允许所有来源的请求
		AllowMethods:     []string{"GET", "POST", "PUT", "DELETE", "OPTIONS"},
		AllowHeaders:     []string{"Origin", "Authorization", "Access-Control-Allow-Origin", "Access-Control-Allow-Headers", "Content-Type"},
		ExposeHeaders:    []string{"Content-Length"},
		AllowCredentials: true,
		MaxAge:           12 * time.Hour,
	}))
	//增加授权拦截
	web.InsertFilter("*", web.BeforeRouter, middleware.AuthMiddleware())
	//web.ErrorHandler("*", exception.ControllerAdvice())
	web.BConfig.RecoverFunc = exception.PanicAdvice
	go func() {
		//内存数据预热预加载
		logger.Info("开始预加载数据")

	}()
	logger.Info("初始化成功")
	web.Run()

}