101 lines
2.8 KiB
Go
101 lines
2.8 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"log"
|
||
|
|
"yuyue/config"
|
||
|
|
"yuyue/database"
|
||
|
|
"yuyue/handlers"
|
||
|
|
"yuyue/middleware"
|
||
|
|
|
||
|
|
"github.com/gofiber/fiber/v2"
|
||
|
|
"github.com/gofiber/fiber/v2/middleware/cors"
|
||
|
|
"github.com/gofiber/fiber/v2/middleware/logger"
|
||
|
|
"github.com/gofiber/fiber/v2/middleware/recover"
|
||
|
|
)
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
// 加载配置
|
||
|
|
cfg := config.LoadConfig()
|
||
|
|
|
||
|
|
// 连接数据库
|
||
|
|
database.ConnectDatabase(cfg)
|
||
|
|
//// 初始化两个月的预约时间槽
|
||
|
|
//if err := utils.GenerateTwoMonthsTimeSlots(); err != nil {
|
||
|
|
// log.Printf("生成时间槽失败: %v", err)
|
||
|
|
//} else {
|
||
|
|
// log.Println("✅ 成功生成最近两个月的预约时间槽")
|
||
|
|
//}
|
||
|
|
// 创建 Fiber 应用
|
||
|
|
app := fiber.New(fiber.Config{
|
||
|
|
AppName: "预约系统",
|
||
|
|
})
|
||
|
|
|
||
|
|
// 中间件
|
||
|
|
app.Use(logger.New())
|
||
|
|
app.Use(recover.New())
|
||
|
|
app.Use(cors.New(cors.Config{
|
||
|
|
AllowOrigins: "*",
|
||
|
|
AllowHeaders: "Origin, Content-Type, Accept, Authorization",
|
||
|
|
AllowMethods: "GET, POST, PUT, DELETE, OPTIONS",
|
||
|
|
}))
|
||
|
|
|
||
|
|
// 设置路由
|
||
|
|
setupRoutes(app)
|
||
|
|
|
||
|
|
// 启动服务器
|
||
|
|
log.Printf("🚀 服务器启动在端口 %s", cfg.ServerPort)
|
||
|
|
log.Fatal(app.Listen(":" + cfg.ServerPort))
|
||
|
|
}
|
||
|
|
|
||
|
|
func setupRoutes(app *fiber.App) {
|
||
|
|
// 公开路由
|
||
|
|
auth := app.Group("/api/auth")
|
||
|
|
auth.Post("/register", handlers.Register)
|
||
|
|
auth.Post("/send-code", handlers.SendVerificationCode) // 发送验证码
|
||
|
|
auth.Post("/verification-login", handlers.VerificationLogin) // 验证码登录
|
||
|
|
auth.Post("/one-click-login", handlers.OneClickLogin) // 一键登录
|
||
|
|
|
||
|
|
// 时间槽相关路由（公开）
|
||
|
|
timeslots := app.Group("/api/timeslots")
|
||
|
|
timeslots.Get("/", handlers.GetTimeSlots)
|
||
|
|
timeslots.Get("/:id", handlers.GetTimeSlot)
|
||
|
|
|
||
|
|
// 需要认证的路由
|
||
|
|
protected := app.Group("/api")
|
||
|
|
protected.Use(middleware.AuthMiddleware())
|
||
|
|
|
||
|
|
// 用户相关路由
|
||
|
|
users := protected.Group("/users")
|
||
|
|
users.Get("/profile", handlers.GetProfile)
|
||
|
|
users.Put("/profile", handlers.UpdateProfile)
|
||
|
|
|
||
|
|
// 预约相关路由
|
||
|
|
appointments := protected.Group("/appointments")
|
||
|
|
appointments.Post("/", handlers.CreateAppointment)
|
||
|
|
appointments.Get("/", handlers.GetAppointments)
|
||
|
|
appointments.Get("/:id", handlers.GetAppointment)
|
||
|
|
appointments.Delete("/:id", handlers.CancelAppointment)
|
||
|
|
|
||
|
|
// 管理员路由
|
||
|
|
admin := protected.Group("/admin")
|
||
|
|
admin.Use(middleware.AdminMiddleware())
|
||
|
|
|
||
|
|
// 管理时间槽
|
||
|
|
adminTimeSlots := admin.Group("/timeslots")
|
||
|
|
adminTimeSlots.Post("/", handlers.CreateTimeSlot)
|
||
|
|
adminTimeSlots.Put("/:id", handlers.UpdateTimeSlot)
|
||
|
|
adminTimeSlots.Delete("/:id", handlers.DeleteTimeSlot)
|
||
|
|
|
||
|
|
// 管理预约
|
||
|
|
adminAppointments := admin.Group("/appointments")
|
||
|
|
adminAppointments.Put("/:id/status", handlers.UpdateAppointmentStatus)
|
||
|
|
|
||
|
|
// 健康检查
|
||
|
|
app.Get("/health", func(c *fiber.Ctx) error {
|
||
|
|
return c.JSON(fiber.Map{
|
||
|
|
"status": "ok",
|
||
|
|
"message": "预约系统运行正常",
|
||
|
|
})
|
||
|
|
})
|
||
|
|
}
|