引言

Go 语言的 database/sql 包为开发者提供了一个抽象层，使得与 MySQL 数据库的交互变得更加容易。

1. 安装 MySQL 驱动

为了能够与 MySQL 数据库通信，我们需要安装 go-sql-driver/mysql 驱动。可以通过 Go 的包管理工具 go get 安装：

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

2. 创建数据库连接

使用 sql.Open 方法来打开一个数据库连接，并使用 Ping 方法测试连接。

package main

import (
	"database/sql"
	"log"

	_ "github.com/go-sql-driver/mysql"
)

func main() {
	db, err := sql.Open("mysql", "user:password@tcp(127.0.0.1:3306)/dbname")
	if err != nil {
		log.Fatal(err)
	}
	defer db.Close()

	err = db.Ping()
	if err != nil {
		log.Fatal(err)
	}
	log.Println("Connected to the database successfully.")
}

3. 执行SQL语句

3.1 查询单条记录

使用 db.QueryRow 方法来查询一条记录，并通过 Scan 方法将结果赋值给变量。

var id int
var name string
err := db.QueryRow("SELECT id, name FROM users WHERE id = ?", 1).Scan(&id, &name)
if err != nil {
  if errors.Is(err, sql.ErrNoRows) {
    log.Println("No rows were returned!")
  } else {
    log.Fatal(err)
  }
}
log.Printf("Found user with ID %d and name %s\n", id, name)

3.2 查询多条记录

使用 db.Query 方法来查询多条记录，并遍历结果集。

rows, err := db.Query("SELECT id, name FROM users")
if err != nil {
  log.Fatal(err)
}
defer rows.Close()

for rows.Next() {
  var id int
  var name string
  err := rows.Scan(&id, &name)
  if err != nil {
    log.Fatal(err)
  }
  log.Printf("User ID: %d, Name: %s\n", id, name)
}

3.3 插入记录

使用 db.Exec 方法来插入记录，并获取最后插入的 ID 和受影响的行数。

result, err := db.Exec("INSERT INTO users (name, email) VALUES (?, ?)", "John Doe", "john.doe@example.com")
if err != nil {
  log.Fatal(err)
}

lastInsertId, err := result.LastInsertId()
if err != nil {
  log.Fatal(err)
}

rowsAffected, err := result.RowsAffected()
if err != nil {
  log.Fatal(err)
}

log.Printf("Last Insert ID: %d, Rows Affected: %d\n", lastInsertId, rowsAffected)

3.4 更新记录

同样使用 db.Exec 方法来更新记录。

_, err := db.Exec("UPDATE users SET name = ? WHERE id = ?", "Jane Doe", 1)
if err != nil {
  log.Fatal(err)
}

3.5 删除记录

还是使用 db.Exec 方法来删除记录。

_, err := db.Exec("DELETE FROM users WHERE id = ?", 1)
if err != nil {
  log.Fatal(err)
}

3.6 事务处理

在进行涉及多个操作的业务逻辑时，事务处理是必不可少的。事务保证了操作的原子性、一致性、隔离性和持久性（ACID）。

tx, err := db.Begin()
if err != nil {
  log.Fatal(err)
}

// 执行多个操作
_, err = tx.Exec("UPDATE users SET balance = balance - 10 WHERE id = ?", 1)
if err != nil {
  tx.Rollback()
  log.Fatal(err)
}

_, err = tx.Exec("INSERT INTO logs (user_id, action) VALUES (?, 'debit')", 1)
if err != nil {
  tx.Rollback()
  log.Fatal(err)
}

err = tx.Commit()
if err != nil {
  log.Fatal(err)
}

3.7 错误处理

在与数据库交互时，错误处理非常重要。始终检查 Exec 和 Query 返回的错误，并在适当的情况下回滚事务。

if err != nil {
  if rbErr := tx.Rollback(); rbErr != nil {
    log.Fatalf("Failed to rollback transaction: %v", rbErr)
  }
  log.Fatalf("Error executing statement: %v", err)
}

4. 示例代码

下面是一个完整的示例，展示了如何使用 Go 语言与 MySQL 数据库进行交互，包括连接、查询、插入、更新和删除记录。

package main

import (
	"database/sql"
	"log"

	_ "github.com/go-sql-driver/mysql"
)

func main() {
	db, err := sql.Open("mysql", "user:password@tcp(127.0.0.1:3306)/dbname")
	if err != nil {
		log.Fatal(err)
	}
	defer db.Close()

	err = db.Ping()
	if err != nil {
		log.Fatal(err)
	}
	log.Println("Connected to the database successfully.")

	// 查询记录
	rows, err := db.Query("SELECT id, name FROM users")
	if err != nil {
		log.Fatal(err)
	}
	defer rows.Close()

	for rows.Next() {
		var id int
		var name string
		err := rows.Scan(&id, &name)
		if err != nil {
			log.Fatal(err)
		}
		log.Printf("User ID: %d, Name: %s\n", id, name)
	}

	// 插入记录
	result, err := db.Exec("INSERT INTO users (name, email) VALUES (?, ?)", "John Doe", "john.doe@example.com")
	if err != nil {
		log.Fatal(err)
	}

	lastInsertId, err := result.LastInsertId()
	if err != nil {
		log.Fatal(err)
	}

	rowsAffected, err := result.RowsAffected()
	if err != nil {
		log.Fatal(err)
	}

	log.Printf("Last Insert ID: %d, Rows Affected: %d\n", lastInsertId, rowsAffected)

	// 更新记录
	_, err = db.Exec("UPDATE users SET name = ? WHERE id = ?", "Jane Doe", 1)
	if err != nil {
		log.Fatal(err)
	}

	// 删除记录
	_, err = db.Exec("DELETE FROM users WHERE id = ?", 1)
	if err != nil {
		log.Fatal(err)
	}
}

5. 数据库推荐

• SQL 数据库

1. MySQL
1. go-sql-driver/mysql：官方 MySQL 驱动，支持基本的 CRUD 操作。
2. gorm/gorm：一个 ORM 库，支持 MySQL、PostgreSQL、SQLite 等多种数据库。
3. jinzhu/gorm：另一个流行的 ORM 库，同样支持多种数据库。
2. PostgreSQL
1. lib/pq：官方 PostgreSQL 驱动。
2. jmoiron/sqlx：扩展了 database/sql API，提供了结构化的查询结果处理。
3. SQLite
1. mattn/go-sqlite3：Go 语言的 SQLite 驱动。
4. SQL Server
1. microsoft/go-mssqldb：Microsoft 提供的 SQL Server 驱动。

• NoSQL 数据库
• MongoDB

1. mongo-go-driver/mongo：官方 MongoDB 驱动。
2. go.mongodb.org/mongo-driver：官方推荐的新版驱动。
1. Redis
1. go-redis/redis：一个高性能的 Redis 客户端库。
2. goredis/goredis：早期版本的 Redis 客户端库。
2. Elasticsearch
1. elastic/go-elasticsearch：Elasticsearch 官方 Go 客户端。

#数据库##golang##MySQL##SQL#