当前位置：首页 > article >正文

15分钟学 Go 第 38 天：数据库基础

article 2024/11/6 10:14:04

第38天 - 数据库基础

学习目标

学习如何连接和操作数据库，包括基本的增、删、改、查功能，以及如何使用Go语言中的database/sql包进行数据库交互。

内容概述

在现代应用程序中，数据库是数据持久化的重要部分。Go语言通过database/sql包提供了对SQL数据库的标准接口，使得我们可以用Go语言方便地操作各种关系型数据库，比如MySQL、PostgreSQL和SQLite等。本教程将详细介绍如何在Go语言中连接数据库、执行基本的SQL操作，并通过实例代码帮助理解。

1. 数据库基本概念

在深入Go语言的数据库操作之前，我们先了解一些基本的数据库概念：

关系型数据库：按照表的形式存储数据，并且表与表之间存在关系，如MySQL、PostgreSQL等。
SQL（结构化查询语言）：用于与关系型数据库进行交互的语言，包括查询、更新、插入和删除数据。
Driver：数据库驱动程序，用于在Go应用与数据库之间进行通信。

数据库连接字符串

连接字符串包含了访问数据库所需的信息，包括用户名、密码、数据库地址和数据库名称等。不同类型的数据库连接字符串格式有所不同：

MySQL连接字符串示例：

username:password@tcp(127.0.0.1:3306)/dbname

2. Go环境准备

在Go语言中操作数据库，我们首先需要安装相应的数据库驱动。以MySQL为例，使用以下命令安装驱动：

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

3. 基本的数据库操作

3.1 连接数据库

使用database/sql包连接数据库的基本步骤如下：

package main

import (
    "database/sql"
    "fmt"
    "log"

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

func main() {
    // 数据库连接字符串
    dsn := "username:password@tcp(127.0.0.1:3306)/dbname"
    // 连接数据库
    db, err := sql.Open("mysql", dsn)
    if err != nil {
        log.Fatal(err)
    }
    defer db.Close()

    // 测试数据库连接
    if err := db.Ping(); err != nil {
        log.Fatal(err)
    }
    fmt.Println("成功连接到数据库！")
}

3.2 执行基本操作

3.2.1 创建表

确保在数据库中创建一个示例表格用于后面的操作：

createTable := `CREATE TABLE IF NOT EXISTS users (
    id INT AUTO_INCREMENT PRIMARY KEY,
    name VARCHAR(100),
    age INT
);`
_, err = db.Exec(createTable)
if err != nil {
    log.Fatal(err)
}

3.2.2 插入数据

插入一条用户数据的示例：

insertUser := `INSERT INTO users (name, age) VALUES (?, ?)`
_, err = db.Exec(insertUser, "Alice", 30)
if err != nil {
    log.Fatal(err)
}
fmt.Println("用户插入成功！")

3.2.3 查询数据

查询用户数据的示例：

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

for rows.Next() {
    var id int
    var name string
    var age int
    if err := rows.Scan(&id, &name, &age); err != nil {
        log.Fatal(err)
    }
    fmt.Printf("用户ID: %d, 名称: %s, 年龄: %d\n", id, name, age)
}

3.2.4 更新数据

更新用户年龄的示例：

updateUser := `UPDATE users SET age = ? WHERE name = ?`
_, err = db.Exec(updateUser, 31, "Alice")
if err != nil {
    log.Fatal(err)
}
fmt.Println("用户年龄更新成功！")

3.2.5 删除数据

删除用户的示例：

deleteUser := `DELETE FROM users WHERE name = ?`
_, err = db.Exec(deleteUser, "Alice")
if err != nil {
    log.Fatal(err)
}
fmt.Println("用户删除成功！")

4. 错误处理

在数据库操作中，要始终注意处理错误。对于每一步数据库操作，都需要检查返回的错误，保证程序的稳定性。

5. 总结与最佳实践

连接池：使用sql.DB类型的连接池管理数据库连接，避免频繁建立和关闭连接带来的性能损耗。
参数化查询：使用参数化查询以防止SQL注入攻击。
事务处理：在需要保证原子性和一致性的操作时（如转账），使用数据库事务。

6. 示例代码完整性

以下是完整示例代码，包含所有操作：

package main

import (
    "database/sql"
    "fmt"
    "log"

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

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

    if err := db.Ping(); err != nil {
        log.Fatal(err)
    }
    fmt.Println("成功连接到数据库！")

    // 创建表
    createTable := `CREATE TABLE IF NOT EXISTS users (
        id INT AUTO_INCREMENT PRIMARY KEY,
        name VARCHAR(100),
        age INT
    );`
    _, err = db.Exec(createTable)
    if err != nil {
        log.Fatal(err)
    }

    // 插入数据
    insertUser := `INSERT INTO users (name, age) VALUES (?, ?)`
    _, err = db.Exec(insertUser, "Alice", 30)
    if err != nil {
        log.Fatal(err)
    }
    fmt.Println("用户插入成功！")

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

    for rows.Next() {
        var id int
        var name string
        var age int
        if err := rows.Scan(&id, &name, &age); err != nil {
            log.Fatal(err)
        }
        fmt.Printf("用户ID: %d, 名称: %s, 年龄: %d\n", id, name, age)
    }

    // 更新数据
    updateUser := `UPDATE users SET age = ? WHERE name = ?`
    _, err = db.Exec(updateUser, 31, "Alice")
    if err != nil {
        log.Fatal(err)
    }
    fmt.Println("用户年龄更新成功！")

    // 删除数据
    deleteUser := `DELETE FROM users WHERE name = ?`
    _, err = db.Exec(deleteUser, "Alice")
    if err != nil {
        log.Fatal(err)
    }
    fmt.Println("用户删除成功！")
}

7. 运行流程图

以下是代码运行流程图示例：

+---------------------+
|   初始化数据库        |
|   连接字符串设置      |
+---------------------+
         |
         V
+---------------------+
|   创建数据库连接      |
+---------------------+
         |
         V
+---------------------+
|   测试连接 (Ping)    |
+---------------------+
         |
         V
+---------------------+
|   执行创建表          |
+---------------------+
         |
         V
+---------------------+
|   插入用户数据        |
+---------------------+
         |
         V
+---------------------+
|   查询用户数据        |
+---------------------+
         |
         V
+---------------------+
|   更新用户年龄        |
+---------------------+
         |
         V
+---------------------+
|   删除用户           |
+---------------------+
         |
         V
+---------------------+
|   关闭数据库连接      |
+---------------------+