使用GROM操作数据库 | 青训营

76 阅读2分钟

什么是GORM

GORM(Go Object Relational Mapping)是一个用于 Go 语言的开源 ORM 库,它提供了一个简洁且功能强大的方式来处理数据库操作。GORM 能够帮助开发者在应用程序和数据库之间建立映射关系,从而避免了繁琐的 SQL 查询和数据处理过程。

特性

  • 全功能 ORM
  • 关联 (Has One,Has Many,Belongs To,Many To Many,多态,单表继承)
  • Create,Save,Update,Delete,Find 中钩子方法
  • 支持 PreloadJoins 的预加载
  • 事务,嵌套事务,Save Point,Rollback To Saved Point
  • Context、预编译模式、DryRun 模式
  • 批量插入,FindInBatches,Find/Create with Map,使用 SQL 表达式、Context Valuer 进行 CRUD
  • SQL 构建器,Upsert,数据库锁,Optimizer/Index/Comment Hint,命名参数,子查询
  • 复合主键,索引,约束
  • Auto Migration
  • 自定义 Logger
  • 灵活的可扩展插件 API:Database Resolver(多数据库,读写分离)、Prometheus…
  • 每个特性都经过了测试的重重考验
  • 开发者友好

示例

一个使用 GORM 进行数据库操作的简单示例:

package main

import (
	"gorm.io/driver/sqlite"
	"gorm.io/gorm"
	"log"
)

type User struct {
	ID   uint   `gorm:"primaryKey"`
	Name string `gorm:"not null"`
	Age  int
}

func main() {
	dsn := "test.db"
	db, err := gorm.Open(sqlite.Open(dsn), &gorm.Config{})
	if err != nil {
		log.Fatal("无法连接数据库")
	}

	// 自动迁移
	db.AutoMigrate(&User{})

	// 创建记录
	newUser := User{Name: "Alice", Age: 25}
	db.Create(&newUser)

	// 查询记录
	var user User
	db.First(&user, 1)
	log.Printf("查询结果:%v", user)

	// 更新记录
	db.Model(&user).Update("Age", 26
)

	// 删除记录
	db.Delete(&user)
}

实践

步骤 1:安装 GORM 首先,我们需要通过 go get 命令安装 GORM 和适用于你选择数据库的驱动。在本例中使用 SQLite 数据库作为示范:

go get -u gorm.io/gorm
go get -u gorm.io/driver/sqlite

步骤 2:连接数据库 在代码中导入 GORM 包,并创建一个数据库连接。以下是一个示例代码,演示如何连接到 SQLite 数据库:

package main

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

func main() {
	dsn := "test.db" // 数据库文件路径
	db, err := gorm.Open(sqlite.Open(dsn), &gorm.Config{})
	if err != nil {
		panic("无法连接数据库")
	}
	
	// 在这里可以使用 db 进行数据库操作
}

步骤 3:定义模型 在 GORM 中,模型是与数据库表对应的结构体。我们将创建一个 User 模型作为示例:

type User struct {
	ID   uint   `gorm:"primaryKey"`
	Name string `gorm:"not null"`
	Age  int
}

步骤 4:创建表 一旦模型定义完成,你可以通过 GORM 的自动迁移功能轻松创建数据库表。在 main 函数中加入以下代码:

db.AutoMigrate(&User{})

步骤 5:实现增删改查操作 接下来,我们将展示如何使用 GORM 实现常见的增删改查操作。

  1. 插入数据
newUser := User{Name: "Alice", Age: 25}
db.Create(&newUser)
  1. 查询数据
var user User
db.First(&user, 1) // 通过 ID 查询第一条记录
db.Where("name = ?", "Alice").First(&user) // 通过条件查询第一条记录
  1. 更新数据
db.Model(&user).Update("Age", 26)
  1. 删除数据
db.Delete(&user)