初始化项目文件

api_file/database/database.go

@@ -0,0 +1,48 @@
+package database
+
+import (
+	"github.com/sirupsen/logrus"
+	"gorm.io/driver/mysql"
+	"gorm.io/gorm"
+	"gorm.io/gorm/logger"
+	"main/config"
+	"sync"
+)
+
+var instance *MysqlConnectionPool
+var once sync.Once
+var db *gorm.DB
+var err error
+
+// GetInstance 获取数据库连接池
+func GetInstance() *MysqlConnectionPool {
+	once.Do(func() {
+		instance = &MysqlConnectionPool{}
+	})
+	return instance
+}
+
+// InitDataPool 数据库初始化连接
+func (m *MysqlConnectionPool) InitDataPool() (isSuccess bool) {
+	dbConfig := config.Config.Database
+	if dbConfig.Prd {
+		db, err = gorm.Open(mysql.Open(dbConfig.Dsn), &gorm.Config{})
+	} else {
+		db, err = gorm.Open(mysql.Open(dbConfig.Dsn), &gorm.Config{
+			Logger: logger.Default.LogMode(logger.Info),
+		})
+	}
+	if err != nil {
+		logrus.Errorln("数据库配置失败：", err)
+		return false
+	}
+	return true
+}
+
+// GetMysqlDb 获取数据库连接
+func (m *MysqlConnectionPool) GetMysqlDb() (dbCon *gorm.DB) {
+	return db
+}
+
+type MysqlConnectionPool struct {
+}