Go 中的数据库交互：从 SQL 到 NoSQL

介绍

go (golang) 已成为构建健壮、高性能后端服务的流行选择。 go 的主要优势之一是它对数据库操作的出色支持，无论您使用传统 sql 数据库还是现代 nosql 解决方案。在本指南中，我们将探索如何在 go 中与数据库交互，涵盖 sql 和 nosql 方法。

目录

1. sql 数据库交互
   • 使用database/sql包
   • 使用 orm：gorm
2. nosql 数据库交互
   • mongodb 与官方 go 驱动程序
3. 最佳实践和常见陷阱
4. 结论

sql 数据库交互

使用database/sql包

go 的标准库提供了 database/sql 包，它提供了围绕 sql（或类似 sql）数据库的通用接口。该软件包旨在与特定于数据库的驱动程序结合使用。

让我们从一个使用 sqlite 的简单示例开始：

package main

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

func main() {
    // open the database
    db, err := sql.open("sqlite3", "./test.db")
    if err != nil {
        log.fatal(err)
    }
    defer db.close()

    // create table
    _, err = db.exec(`create table if not exists users (
        id integer primary key autoincrement,
        name text,
        age integer
    )`)
    if err != nil {
        log.fatal(err)
    }

    // insert a user
    result, err := db.exec("insert into users (name, age) values (?, ?)", "alice", 30)
    if err != nil {
        log.fatal(err)
    }

    // get the id of the inserted user
    id, err := result.lastinsertid()
    if err != nil {
        log.fatal(err)
    }
    fmt.printf("inserted user with id: %d\n", id)

    // query for the user
    var name string
    var age int
    err = db.queryrow("select name, age from users where id = ?", id).scan(&name, &age)
    if err != nil {
        log.fatal(err)
    }
    fmt.printf("user: %s, age: %d\n", name, age)
}

此示例演示了在 go 中使用 sql 数据库的基础知识：

1. 打开数据库连接
2. 创建表格
3. 插入数据
4. 查询数据

database/sql 包提供了数据库的低级接口，让您可以很好地控制查询和操作。

使用 orm：gorm

虽然database/sql包很强大，但许多开发人员更喜欢使用对象关系映射（orm）工具来更方便地进行数据库操作。 gorm 是 go 中最受欢迎的 orm 之一。

这是一个将 gorm 与 sqlite 结合使用的示例：

package main

import (
    "fmt"
    "log"

    "gorm.io/driver/sqlite"
    "gorm.io/gorm"
)

type user struct {
    id   uint
    name string
    age  int
}

func main() {
    // open the database
    db, err := gorm.open(sqlite.open("test.db"), &gorm.config{})
    if err != nil {
        log.fatal(err)
    }

    // auto migrate the schema
    db.automigrate(&user{})

    // create a user
    user := user{name: "bob", age: 25}
    result := db.create(&user)
    if result.error != nil {
        log.fatal(result.error)
    }
    fmt.printf("inserted user with id: %d\n", user.id)

    // query for the user
    var fetcheduser user
    db.first(&fetcheduser, user.id)
    fmt.printf("user: %s, age: %d\n", fetcheduser.name, fetcheduser.age)

    // update the user
    db.model(&fetcheduser).update("age", 26)

    // delete the user
    db.delete(&fetcheduser)
}

gorm 提供了对数据库操作的更高级别的抽象，允许您直接使用 go 结构，而不是编写原始 sql 查询。它还提供自动迁移、挂钩和关联等功能。

nosql 数据库交互

mongodb 与官方 go 驱动程序

对于nosql数据库，我们来看看如何使用官方go驱动与mongodb交互：

package main

import (
    "context"
    "fmt"
    "log"
    "time"

    "go.mongodb.org/mongo-driver/bson"
    "go.mongodb.org/mongo-driver/mongo"
    "go.mongodb.org/mongo-driver/mongo/options"
)

type User struct {
    Name string `bson:"name"`
    Age  int    `bson:"age"`
}

func main() {
    // Set client options
    clientOptions := options.Client().ApplyURI("mongodb://localhost:27017")

    // Connect to MongoDB
    ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
    defer cancel()
    client, err := mongo.Connect(ctx, clientOptions)
    if err != nil {
        log.Fatal(err)
    }

    // Check the connection
    err = client.Ping(context.TODO(), nil)
    if err != nil {
        log.Fatal(err)
    }

    fmt.Println("Connected to MongoDB!")

    // Get a handle for your collection
    collection := client.Database("test").Collection("users")

    // Insert a user
    user := User{Name: "Charlie", Age: 35}
    insertResult, err := collection.InsertOne(context.TODO(), user)
    if err != nil {
        log.Fatal(err)
    }
    fmt.Printf("Inserted user with ID: %v\n", insertResult.InsertedID)

    // Find a user
    var result User
    filter := bson.M{"name": "Charlie"}
    err = collection.FindOne(context.TODO(), filter).Decode(&result)
    if err != nil {
        log.Fatal(err)
    }
    fmt.Printf("Found user: %+v\n", result)

    // Update a user
    update := bson.M{
        "$set": bson.M{
            "age": 36,
        },
    }
    updateResult, err := collection.UpdateOne(context.TODO(), filter, update)
    if err != nil {
        log.Fatal(err)
    }
    fmt.Printf("Updated %v user(s)\n", updateResult.ModifiedCount)

    // Delete a user
    deleteResult, err := collection.DeleteOne(context.TODO(), filter)
    if err != nil {
        log.Fatal(err)
    }
    fmt.Printf("Deleted %v user(s)\n", deleteResult.DeletedCount)

    // Disconnect
    err = client.Disconnect(context.TODO())
    if err != nil {
        log.Fatal(err)
    }
    fmt.Println("Connection to MongoDB closed.")
}

此示例演示了使用官方 go 驱动程序对 mongodb 进行基本的 crud（创建、读取、更新、删除）操作。

最佳实践和常见陷阱

在 go 中使用数据库时，请记住以下最佳实践：

MvMmall 网店系统

免费的开源程序长期以来，为中国的网上交易提供免费开源的网上商店系统一直是我们的初衷和努力奋斗的目标，希望大家一起把MvMmall网上商店系统的免费开源进行到底。2高效的执行效率由资深的开发团队设计，从系统架构,数据库优化，配以通过W3C验证的面页模板，全面提升页面显示速度和提高程序负载能力。3灵活的模板系统MvMmall网店系统程序代码与网页界面分离，灵活的模板方案，完全自定义模板，官方提供免费模

下载

1. 使用连接池：数据库/sql和大多数nosql驱动程序都实现连接池。确保重用数据库连接，而不是为每个操作打开新的连接。

2. 正确处理错误：始终检查数据库操作返回的错误并进行适当处理。

3. 使用prepared statements：对于sql数据库，使用prepared statements可以提高性能并防止sql注入攻击。

4. 关闭资源：使用完结果集、语句和数据库连接后，始终关闭它们。 defer 关键字对此很有用。

5. 必要时使用事务：对于需要多个步骤的操作，使用事务来保证数据的一致性。

6. 注意 n+1 查询问题：使用 orm 时，请注意 n+1 查询问题，并在适当的时候使用急切加载。

7. 使用上下文进行超时：使用上下文设置数据库操作的超时，特别是对于长时间运行的查询。

需要避免的常见陷阱：

1. 忽略 sql 注入漏洞：始终使用参数化查询或准备好的语句。
2. 不处理连接错误：检查并处理连接错误，必要时实现重试逻辑。
3. 过度使用 gorm hooks：虽然方便，但过度使用 gorm hooks 可能会导致难以调试的问题。
4. 未正确建立索引：确保您的数据库架构已针对您的查询模式正确建立索引。
5. 以明文形式存储敏感数据：在将敏感数据存储到数据库之前始终对其进行加密。

结论

go 为 sql 和 nosql 数据库交互提供强大的支持。无论您喜欢数据库/sql 的低级控制、像 gorm 这样的 orm 的便利性，还是使用像 mongodb 这样的 nosql 数据库，go 都能满足您的需求。