Go语言连接MySQL数据库详解(单主机版)

一. 使用官方驱动

1. 导入 "database/sql" 与 "github.com/go-sql-driver/mysql" 包

   使用以下命令安装 MySQL驱动:

   go get github.com/go-sql-driver/mysql
   

• 其中 github.com/go-sql-driver/mysql 库需匿名导入, 而不是直接引入, 因为代码中并没有直接使用这个驱动包, 而是通过 import 语句执行驱动包中的 init() 函数.
  init() 调用 database/sql 包中的 Register 函数, 向 drivers 添加了一个名为 mysql 的驱动
  drivers 是 database/sql 包中维护的一个全局变量, 它管理着所有被添加的驱动

  • github.com/go-sql-driver/mysql 中的 init()函数:

  • func init() {
    	if driverName != "" {
    		sql.Register(driverName, &MySQLDriver{})
    	}
    }
    

• 创建数据库连接对象使用的是 database/sql 包中的 Open 函数, 它会返回一个 *sql.DB 和 err, 函数签名如下:

  func Open(driverName, dataSourceName string) (*DB, error)
  

  • driverName: 驱动名称, 即我们导入驱动时向 database/sql 包中注册的驱动名称

  • dataSourceName: 表示数据源, 一般至少包含数据库文件名和其他连接必要的信息

  • 注: Open 函数只是简单验证参数格式是否正确, 实际上并不创建与数据库的连接. 若要检查数据源信息是否真实有效, 那么应该调用 ping 方法

2. 编写初始化函数

   import (
   	"database/sql"
   	"fmt"
   	_ "github.com/go-sql-driver/mysql"
   )
   
   func initDB() (db *sql.DB, err error) {
   	// DSN 全称 data source name, 表示要连接的数据源名称
   	dsn := "数据库用户:数据库密码@tcp(127.0.0.1:3306)/数据库库名?charset=utf8mb4&parseTime=True"
   	db, err = sql.Open("mysql", dsn)
   	if err != nil {
   		fmt.Println("连接数据库失败:", err)
   		return nil, err
   	}
   	defer db.Close()
   
   	err = db.Ping()
   	if err != nil {
   		fmt.Println("连接数据库失败:", err)
   		return nil, err
   	}
   	return db, nil
   }
   

3. 使用初始化函数连接数据库

   func main() {
   	db, err := initDB()
   	if err != nil {
   		fmt.Printf("连接数据库失败: %v\n", err)
   		return
   	}
   	fmt.Println(db)
   }
   

二. 使用 sqlx 第三方库连接数据库

1. sqlx 简介

sqlx 库 可以认为是 database/sql 的超集, 它在 database/sql 的基础上提供了一些易用的方法扩展, 主要包括:

• 支持将查询结果直接解析到 struct、 map、 slice
• 支持命名参数
• 支持使用 Get 和 Select 快捷地查询数据至结构体或切片
• 除此之外, sqlx 还提供了很多强大的功能, 例如各种便捷的查询, 批量插入数据等.

使用以下命令安装 sqlx 库

go get github.com/jmoiron/sqlx

2. 使用 sqlx 连接MySQL数据库

1. 编写MySQL初始化函数

   import (
   	"fmt"
   	_ "github.com/go-sql-driver/mysql"
   	"github.com/jmoiron/sqlx"
   )
   
   func initDB() (db *sqlx.DB, err error) {
   	dsn := "数据库用户:数据库密码@tcp(127.0.0.1:3306)/数据库库名?charset=utf8mb4&parseTime=True"
   
   	// 方法一 使用 sqlx.Connect
   	db, err = sqlx.Connect("mysql", dsn)
   	if err != nil {
   		fmt.Printf("connect DB failed, err:%v\n", err)
   		return db, err
   	}
   
   	 方法二 使用 sqlx.MustConnect
   	//db = sqlx.MustConnect("mysql", dsn)
   
   	// 连接池设置
   	db.SetMaxOpenConns(20)
   	db.SetMaxIdleConns(10)
   	return db, err
   }
   

2. 使用初始化函数连接数据库

   func main() {
   	db, err := initDB()
   	if err != nil {
   		fmt.Printf("连接数据库失败: %v\n", err)
   		return
   	}
   	fmt.Println(db)
   }
   

3. 使用 ORM 框架连接数据库

GORM 是一个神奇的，对开发人员友好的 Golang ORM 库

其支持预编译模式，批量插入，嵌套事务，多态，单表继承等特性

使用以下命令安装 gorm 库

go get -u gorm.io/gorm
go get -u gorm.io/driver/sqlite

1. 编写MySQL初始化函数

   import (
   	"fmt"
   	"gorm.io/driver/mysql"
   	"gorm.io/gorm"
   )
   
   func initDB() (db *gorm.DB, err error) {
   	dsn := "数据库用户:数据库密码@tcp(127.0.0.1:3306)/数据库库名?charset=utf8mb4&parseTime=True&loc=Local"
   
   	// 建立数据库连接
   	db, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
   	if err != nil {
   		panic("failed to connect database")
   	}
   
   	// 配置连接池
   	sqlDB, err := db.DB()
   	if err != nil {
   		panic("failed to get database instance")
   	}
   
   	sqlDB.SetMaxIdleConns(10)  // 最大空闲连接数
   	sqlDB.SetMaxOpenConns(100) // 最大打开连接数
   	return db, err
   }
   

2. 使用初始化函数连接数据库

   func main() {
   	db, err := initDB()
   	if err != nil {
   		fmt.Println(err)
   	}
   
   	fmt.Println(db)
   }