27 lines
622 B
Go
27 lines
622 B
Go
package server
|
|
|
|
import (
|
|
"context"
|
|
|
|
"tss-rocks-be/ent"
|
|
"tss-rocks-be/internal/config"
|
|
|
|
_ "github.com/mattn/go-sqlite3"
|
|
"github.com/rs/zerolog/log"
|
|
)
|
|
|
|
// NewEntClient creates a new ent client
|
|
func NewEntClient(cfg *config.Config) *ent.Client {
|
|
// 使用配置文件中的数据库设置
|
|
client, err := ent.Open(cfg.Database.Driver, cfg.Database.DSN)
|
|
if err != nil {
|
|
log.Fatal().Err(err).Msg("Failed to connect to database")
|
|
}
|
|
|
|
// Run the auto migration tool
|
|
if err := client.Schema.Create(context.Background()); err != nil {
|
|
log.Fatal().Err(err).Msg("Failed to create schema resources")
|
|
}
|
|
|
|
return client
|
|
}
|