woord-core-service/main.go

package main

import (
	"os"
	"woord-core-service/controller"
	"woord-core-service/global"
	"woord-core-service/model"

	"github.com/gin-contrib/cors"
	"github.com/gin-contrib/sessions"
	gormsessions "github.com/gin-contrib/sessions/gorm"
	"github.com/gin-gonic/gin"
	"gorm.io/driver/sqlite"
	"gorm.io/gorm"
)

func main() {
	var err error
	// 连接数据库
	global.DB, err = gorm.Open(sqlite.Open(os.Getenv("DB_PATH")), &gorm.Config{})
	if err != nil {
		panic(err)
	}
	// 自动迁移 schema
	global.DB.AutoMigrate(&model.User{}, &model.Dict{}, &model.Word{})

	global.SecretKey = []byte(os.Getenv("SECRET_KEY"))
	// 创建 session 存储
	store := gormsessions.NewStore(global.DB, true, global.SecretKey)

	r := gin.Default()
	r.Use(cors.Default())
	r.Use(sessions.Sessions("_session", store))

	r.POST("/user/register", controller.Register)
	r.POST("/user/login", controller.Login)

	auth := r.Group("/", controller.JWTAuth())
	{
		auth.GET("/user/get", controller.GetCurrentUser)

		auth.GET("/dict/list", controller.ListDicts)
		auth.GET("/dict/get", controller.GetDict)
	}

	r.Run(":8080")
}