From 25532520499f51fc2d7a4eeea604750e7fa1adc1 Mon Sep 17 00:00:00 2001 From: yanghao05 Date: Mon, 30 Jan 2023 18:25:03 +0800 Subject: [PATCH] auto create database --- config.toml | 2 +- providers/database/mysql.go | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/config.toml b/config.toml index 2ccca92..a4b416c 100644 --- a/config.toml +++ b/config.toml @@ -87,7 +87,7 @@ Driver = "mysql" [Database.MySQL] Host = "localhost" Port = 3306 -Database = "demos" +Database = "demos11" Username = "root" Password = "root" Prefix="" diff --git a/providers/database/mysql.go b/providers/database/mysql.go index 84e7880..8f623fb 100644 --- a/providers/database/mysql.go +++ b/providers/database/mysql.go @@ -19,6 +19,10 @@ func init() { } func NewDatabase(config *config.Config) (*gorm.DB, error) { + if err := createDatabase(config.Database.MySQL.EmptyDsn(), "mysql", config.Database.MySQL.CreateDatabaseSql()); err != nil { + return nil, err + } + mysqlConfig := mysql.Config{ DSN: config.Database.MySQL.DSN(), // DSN data source name DefaultStringSize: 191, // string 类型字段的默认长度 @@ -53,10 +57,6 @@ func NewDatabase(config *config.Config) (*gorm.DB, error) { sqlDB.SetMaxIdleConns(config.Database.MySQL.MaxIdleConns) sqlDB.SetMaxOpenConns(config.Database.MySQL.MaxOpenConns) - if err := createDatabase(config.Database.MySQL.EmptyDsn(), "mysql", config.Database.MySQL.CreateDatabaseSql()); err != nil { - return nil, err - } - return db, err }