【Go语言快速上手】第二部分:Go语言进阶之数据库操作

文章目录

  • 前言:数据库操作
  • [一、使用 `database/sql` 包操作关系型数据库](#一、使用 database/sql 包操作关系型数据库)
    • [1.1 连接数据库](#1.1 连接数据库)
      • [1.1.1 安装 MySQL 驱动](#1.1.1 安装 MySQL 驱动)
      • [1.1.2 连接到 MySQL 数据库](#1.1.2 连接到 MySQL 数据库)
    • [1.2 执行 SQL 查询](#1.2 执行 SQL 查询)
      • [1.2.1 查询单行数据](#1.2.1 查询单行数据)
      • [1.2.2 查询多行数据](#1.2.2 查询多行数据)
    • [1.3 执行 SQL 插入、更新和删除](#1.3 执行 SQL 插入、更新和删除)
      • 插入数据
      • [1.3.1 更新数据](#1.3.1 更新数据)
      • [1.3.2 删除数据](#1.3.2 删除数据)
  • [二、使用 NoSQL 数据库](#二、使用 NoSQL 数据库)
    • [2.1 使用 MongoDB](#2.1 使用 MongoDB)
      • [2.1.1 安装 MongoDB 驱动](#2.1.1 安装 MongoDB 驱动)
      • [2.1.2 连接到 MongoDB 并执行查询](#2.1.2 连接到 MongoDB 并执行查询)
    • [2.2 使用 Redis](#2.2 使用 Redis)
      • [2.2.1 安装 Redis 驱动](#2.2.1 安装 Redis 驱动)
      • [2.2.2 连接到 Redis 并执行操作](#2.2.2 连接到 Redis 并执行操作)

前言:数据库操作

Go 语言通过内置的 database/sql 包支持关系型数据库的操作,同时也支持通过第三方库与 NoSQL 数据库进行交互。你可以通过标准的 SQL 操作与关系型数据库(如 MySQL、PostgreSQL)进行交互,也可以使用专门的库来连接 NoSQL 数据库(如 MongoDB、Redis)。本篇博客将介绍如何在 Go 中操作关系型数据库与 NoSQL 数据库。


一、使用 database/sql 包操作关系型数据库

Go 的 database/sql 包提供了一个统一的接口,允许与多种关系型数据库(如 MySQL、PostgreSQL 等)进行交互。你可以通过 database/sql 包提供的 API 执行 SQL 查询、插入、更新和删除等操作。

1.1 连接数据库

首先,你需要安装并导入适用于数据库的驱动,例如对于 MySQL,你可以使用 github.com/go-sql-driver/mysql 驱动,针对 PostgreSQL,可以使用 github.com/lib/pq

1.1.1 安装 MySQL 驱动

sh 复制代码
go get -u github.com/go-sql-driver/mysql

1.1.2 连接到 MySQL 数据库

go 复制代码
package main

import (
    "fmt"
    "log"
    "database/sql"
    _ "github.com/go-sql-driver/mysql"
)

func main() {
    // 连接数据库
    dsn := "root:password@tcp(127.0.0.1:3306)/testdb"
    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("Successfully connected to MySQL database")
}

1.2 执行 SQL 查询

执行 SQL 查询时,使用 QueryQueryRow 方法获取数据。

1.2.1 查询单行数据

go 复制代码
package main

import (
    "fmt"
    "log"
    "database/sql"
    _ "github.com/go-sql-driver/mysql"
)

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

    // 查询单行数据
    var name string
    err = db.QueryRow("SELECT name FROM users WHERE id = ?", 1).Scan(&name)
    if err != nil {
        log.Fatal(err)
    }
    fmt.Println("Name:", name)
}

1.2.2 查询多行数据

go 复制代码
package main

import (
    "fmt"
    "log"
    "database/sql"
    _ "github.com/go-sql-driver/mysql"
)

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

    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
        if err := rows.Scan(&id, &name); err != nil {
            log.Fatal(err)
        }
        fmt.Println(id, name)
    }

    if err := rows.Err(); err != nil {
        log.Fatal(err)
    }
}

1.3 执行 SQL 插入、更新和删除

插入数据

go 复制代码
package main

import (
    "fmt"
    "log"
    "database/sql"
    _ "github.com/go-sql-driver/mysql"
)

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

    // 插入数据
    result, err := db.Exec("INSERT INTO users(name) VALUES(?)", "Alice")
    if err != nil {
        log.Fatal(err)
    }

    lastInsertID, err := result.LastInsertId()
    if err != nil {
        log.Fatal(err)
    }
    fmt.Println("Inserted record with ID:", lastInsertID)
}

1.3.1 更新数据

go 复制代码
package main

import (
    "fmt"
    "log"
    "database/sql"
    _ "github.com/go-sql-driver/mysql"
)

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

    // 更新数据
    result, err := db.Exec("UPDATE users SET name = ? WHERE id = ?", "Bob", 1)
    if err != nil {
        log.Fatal(err)
    }

    affectedRows, err := result.RowsAffected()
    if err != nil {
        log.Fatal(err)
    }
    fmt.Println("Affected rows:", affectedRows)
}

1.3.2 删除数据

go 复制代码
package main

import (
    "fmt"
    "log"
    "database/sql"
    _ "github.com/go-sql-driver/mysql"
)

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

    // 删除数据
    result, err := db.Exec("DELETE FROM users WHERE id = ?", 1)
    if err != nil {
        log.Fatal(err)
    }

    affectedRows, err := result.RowsAffected()
    if err != nil {
        log.Fatal(err)
    }
    fmt.Println("Affected rows:", affectedRows)
}

二、使用 NoSQL 数据库

Go 语言也支持与 NoSQL 数据库进行交互,如 MongoDB 和 Redis。我们将介绍如何使用 Go 操作这两种数据库。

2.1 使用 MongoDB

MongoDB 是一个文档型 NoSQL 数据库,可以通过 go.mongodb.org/mongo-driver 驱动与 MongoDB 进行交互。

2.1.1 安装 MongoDB 驱动

sh 复制代码
go get go.mongodb.org/mongo-driver/mongo

2.1.2 连接到 MongoDB 并执行查询

go 复制代码
package main

import (
    "fmt"
    "log"
    "context"
    "go.mongodb.org/mongo-driver/mongo"
    "go.mongodb.org/mongo-driver/mongo/options"
)

func main() {
    client, err := mongo.Connect(context.TODO(), options.Client().ApplyURI("mongodb://localhost:27017"))
    if err != nil {
        log.Fatal(err)
    }
    defer client.Disconnect(context.TODO())

    collection := client.Database("testdb").Collection("users")

    var result map[string]interface{}
    err = collection.FindOne(context.TODO(), map[string]interface{}{"name": "Alice"}).Decode(&result)
    if err != nil {
        log.Fatal(err)
    }

    fmt.Println("Found user:", result)
}

2.2 使用 Redis

Redis 是一个键值存储数据库,可以使用 github.com/go-redis/redis/v8 库与 Redis 进行交互。

2.2.1 安装 Redis 驱动

sh 复制代码
go get github.com/go-redis/redis/v8

2.2.2 连接到 Redis 并执行操作

go 复制代码
package main

import (
    "fmt"
    "log"
    "github.com/go-redis/redis/v8"
    "context"
)

func main() {
    rdb := redis.NewClient(&redis.Options{
        Addr: "localhost:6379", // Redis 地址
    })
    ctx := context.Background()

    // 设置键值对
    err := rdb.Set(ctx, "name", "Alice", 0).Err()
    if err != nil {
        log.Fatal(err)
    }

    // 获取键值对
    val, err := rdb.Get(ctx, "name").Result()
    if err != nil {
        log.Fatal(err)
    }

    fmt.Println("name:", val)
}

相关推荐
s153355 分钟前
数据结构之顺序表,链表,栈,队列
数据结构·数据库
Naiva20 分钟前
【小技巧】Python + PyCharm 小智AI配置MCP接入点使用说明(内测)( PyInstaller打包成 .exe 可执行文件)
开发语言·python·pycharm
梦子要转行29 分钟前
matlab/Simulink-全套50个汽车性能建模与仿真源码模型9
开发语言·matlab·汽车
北方有星辰zz1 小时前
数据结构:栈
java·开发语言·数据结构
我是唐青枫1 小时前
C#.NET NLog 详解
开发语言·c#·.net
Mr_Xuhhh2 小时前
网络基础(1)
c语言·开发语言·网络·c++·qt·算法
旺旺大力包2 小时前
【JS笔记】JS 和 noodjs 的常见操作(十)
开发语言·javascript·node.js·ecmascript
混乱意志2 小时前
dgraph example数据导入
数据库·后端
Web极客码2 小时前
WordPress 站点漏洞利用:数据库恶意注入与多重感染的案例分析
数据库·wordpress·网站安全·数据库注入·wordpress漏洞·wordpress安全插件
刺客xs2 小时前
MySQL数据库----DML语句
数据库·mysql