Go (Golang) 已成为构建健壮、高性能后端服务的流行选择。 Go 的主要优势之一是它对数据库操作的出色支持，无论您使用传统 SQL 数据库还是现代 NoSQL 解决方案。在本指南中，我们将探索如何在 Go 中与数据库交互，涵盖 SQL 和 NoSQL 方法。
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 数据库的基础知识：
database/sql 包提供了数据库的低级接口，让您可以很好地控制查询和操作。
虽然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数据库，我们来看看如何使用官方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 中使用数据库时，请记住以下最佳实践：
使用连接池：数据库/sql和大多数NoSQL驱动程序都实现连接池。确保重用数据库连接，而不是为每个操作打开新的连接。
正确处理错误：始终检查数据库操作返回的错误并进行适当处理。
使用prepared statements：对于SQL数据库，使用prepared statements可以提高性能并防止SQL注入攻击。
关闭资源：使用完结果集、语句和数据库连接后，始终关闭它们。 defer 关键字对此很有用。
必要时使用事务：对于需要多个步骤的操作，使用事务来保证数据的一致性。
注意 N 1 查询问题：使用 ORM 时，请注意 N 1 查询问题，并在适当的时候使用急切加载。
使用上下文进行超时：使用上下文设置数据库操作的超时，特别是对于长时间运行的查询。
需要避免的常见陷阱：
Go 为 SQL 和 NoSQL 数据库交互提供强大的支持。无论您喜欢数据库/sql 的低级控制、像 GORM 这样的 ORM 的便利性，还是使用像 MongoDB 这样的 NoSQL 数据库，Go 都能满足您的需求。
以上是Go 中的数据库交互：从 SQL 到 NoSQL的详细内容。更多信息请关注PHP中文网其他相关文章！
