From 706c72b973b49e69d6b6eb0730f119ae854f0b34 Mon Sep 17 00:00:00 2001
From: linwenxiang <991154416@qq.com>
Date: Thu, 29 Oct 2020 14:48:25 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E6=B3=A8=E9=87=8A?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 common/database/initialize.go  | 1 +
 common/database/interface.go   | 1 +
 common/database/mysql_drive.go | 9 ++++++---
 3 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/common/database/initialize.go b/common/database/initialize.go
index f78fc0ca..f59ac3bb 100644
--- a/common/database/initialize.go
+++ b/common/database/initialize.go
@@ -2,6 +2,7 @@
 
 package database
 
+// Setup 配置数据库
 func Setup(driver string) {
 	dbType := driver
 	if dbType == "mysql" {
diff --git a/common/database/interface.go b/common/database/interface.go
index 0fc32337..59c5e1ce 100644
--- a/common/database/interface.go
+++ b/common/database/interface.go
@@ -2,6 +2,7 @@ package database
 
 import "gorm.io/gorm"
 
+// Database 数据库配置
 type Database interface {
 	Setup()
 	Open(conn string, cfg *gorm.Config) (db *gorm.DB, err error)
diff --git a/common/database/mysql_drive.go b/common/database/mysql_drive.go
index c30d7a2b..faa4d977 100644
--- a/common/database/mysql_drive.go
+++ b/common/database/mysql_drive.go
@@ -5,12 +5,12 @@ import (
 	. "log"
 	"time"
 
+	goAdminLogger "github.com/go-admin-team/go-admin-core/logger"
 	"gorm.io/driver/mysql"
 	"gorm.io/gorm"
 	"gorm.io/gorm/logger"
 	"gorm.io/gorm/schema"
 
-	goAdminLogger "github.com/go-admin-team/go-admin-core/logger"
 	"go-admin/common/config"
 	"go-admin/common/global"
 	"go-admin/common/log"
@@ -18,9 +18,11 @@ import (
 	toolsConfig "go-admin/tools/config"
 )
 
+// Mysql mysql配置结构体
 type Mysql struct {
 }
 
+// Setup 配置步骤
 func (e *Mysql) Setup() {
 	global.Source = e.GetConnect()
 	log.Info(tools.Green(global.Source))
@@ -59,16 +61,17 @@ func (e *Mysql) Setup() {
 	}
 }
 
-// 打开数据库连接
+// Open 打开数据库连接
 func (e *Mysql) Open(db *sql.DB, cfg *gorm.Config) (*gorm.DB, error) {
 	return gorm.Open(mysql.New(mysql.Config{Conn: db}), cfg)
 }
 
-// 获取数据库连接
+// GetConnect 获取数据库连接
 func (e *Mysql) GetConnect() string {
 	return toolsConfig.DatabaseConfig.Source
 }
 
+// GetDriver 获取连接
 func (e *Mysql) GetDriver() string {
 	return toolsConfig.DatabaseConfig.Driver
 }