shudao-main/shudao-go-backend/main.go

package main

import (
	_ "shudao-chat-go/routers"
	"shudao-chat-go/controllers"
	"shudao-chat-go/utils"

	beego "github.com/beego/beego/v2/server/web"
	"github.com/beego/beego/v2/server/web/filter/cors"
)

func main() {
	beego.BConfig.CopyRequestBody = true

	// 注册认证中间件
	beego.InsertFilter("*", beego.BeforeRouter, utils.AuthMiddleware)
	beego.InsertFilter("*", beego.BeforeExec, utils.AuthMiddleware)

	// CORS配置
	beego.InsertFilter("*", beego.BeforeRouter, cors.Allow(&cors.Options{
		AllowOrigins:     []string{"*"},
		AllowMethods:     []string{"GET", "POST", "PUT", "DELETE", "OPTIONS"},
		AllowHeaders:     []string{"Origin", "Authorization", "Access-Control-Allow-Origin", "Access-Control-Allow-Headers", "Content-Type", "token"},
		ExposeHeaders:    []string{"Content-Length", "Access-Control-Allow-Origin", "Access-Control-Allow-Headers", "Content-Type"},
		AllowCredentials: true,
	}))

	// 静态文件路径
	beego.SetStaticPath("/assets", "assets")
	beego.SetStaticPath("/static", "static")
	beego.SetStaticPath("/src", "assets")

	// 启动心跳任务
	controllers.StartChromaSearchHeartbeatTask()

	beego.Run()
}