64 lines
1.2 KiB
Go
64 lines
1.2 KiB
Go
package database
|
||
|
||
import (
|
||
"goblog/config"
|
||
"goblog/models"
|
||
"log"
|
||
|
||
"github.com/glebarez/sqlite"
|
||
"gorm.io/gorm"
|
||
"gorm.io/gorm/logger"
|
||
)
|
||
|
||
var DB *gorm.DB
|
||
|
||
func Init(cfg *config.DatabaseConfig) error {
|
||
var err error
|
||
|
||
DB, err = gorm.Open(sqlite.Open(cfg.DSN), &gorm.Config{
|
||
Logger: logger.Default.LogMode(logger.Info),
|
||
})
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
// 自动迁移
|
||
err = DB.AutoMigrate(
|
||
&models.User{},
|
||
&models.Post{},
|
||
&models.Category{},
|
||
&models.Tag{},
|
||
&models.Comment{},
|
||
&models.Page{},
|
||
&models.Option{},
|
||
)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
log.Println("数据库连接成功")
|
||
return nil
|
||
}
|
||
|
||
// 创建默认管理员
|
||
func CreateDefaultAdmin() error {
|
||
var count int64
|
||
DB.Model(&models.User{}).Where("role = ?", "admin").Count(&count)
|
||
|
||
if count == 0 {
|
||
admin := &models.User{
|
||
Username: "admin",
|
||
Password: "$2a$10$XrrWkDB.DIpuXOYYrMiZdOBpl0gxrtziSCQ4OOnGFP10C8.xF30qq", // admin123
|
||
Nickname: "管理员",
|
||
Email: "admin@example.com",
|
||
Role: "admin",
|
||
Status: 1,
|
||
}
|
||
if err := DB.Create(admin).Error; err != nil {
|
||
return err
|
||
}
|
||
log.Println("默认管理员创建成功，用户名: admin，密码: admin123")
|
||
}
|
||
return nil
|
||
}
|