refactor: add database initialization entrypoint

This commit is contained in:
2026-07-23 13:55:49 +08:00
parent 50274184dc
commit 51ee510112
3 changed files with 11 additions and 3 deletions

View File

@@ -23,7 +23,7 @@ func main() {
if err != nil {
log.Fatal(err)
}
if err := initdb.InitUser(models.DBService); err != nil {
if err := initdb.New(models.DBService); err != nil {
log.Fatal(err)
}

View File

@@ -0,0 +1,8 @@
package initdb
import "gorm.io/gorm"
// New initializes the default database records.
func New(database *gorm.DB) error {
return InitUser(database)
}

View File

@@ -13,7 +13,7 @@ import (
func TestInitUserCreatesRootUserWhenTableIsEmpty(t *testing.T) {
database := newUserTestDatabase(t)
require.NoError(t, InitUser(database))
require.NoError(t, New(database))
var users []models.SenlinAgentUser
require.NoError(t, database.Find(&users).Error)
@@ -34,7 +34,7 @@ func TestInitUserDoesNothingWhenTableIsNotEmpty(t *testing.T) {
}
require.NoError(t, database.Create(&existing).Error)
require.NoError(t, InitUser(database))
require.NoError(t, New(database))
var users []models.SenlinAgentUser
require.NoError(t, database.Find(&users).Error)