github.com/easysoft/zendata@v0.0.0-20240513203326-705bd5a7fd67/internal/server/config/db.go (about)

     1  package serverConfig
     2  
     3  import (
     4  	"gorm.io/gorm/logger"
     5  
     6  	consts "github.com/easysoft/zendata/internal/pkg/const"
     7  	"github.com/easysoft/zendata/internal/pkg/model"
     8  	"github.com/easysoft/zendata/pkg/utils/vari"
     9  	"github.com/glebarez/sqlite"
    10  	"gorm.io/gorm"
    11  )
    12  
    13  func NewGormDB() (gormDb *gorm.DB, err error) {
    14  	gormDb, err = gorm.Open(sqlite.Open(consts.SqliteFile), &gorm.Config{
    15  		Logger: logger.Default.LogMode(logger.Info),
    16  	})
    17  
    18  	if vari.Verbose {
    19  		gormDb = gormDb.Debug()
    20  	}
    21  
    22  	err = gormDb.AutoMigrate(
    23  		model.Models...,
    24  	)
    25  
    26  	return
    27  }