使用GORM连接数据库和实现增删改查操作 | 青训营

85 阅读4分钟

简介

在现代的应用开发中,与数据库进行交互是非常常见的任务。GORM(Go Object Relational Mapping)是一个流行的 Go 语言 ORM 库,它大大简化了与数据库的交互,允许开发者通过 Go 语言的结构体来进行数据库操作,而无需编写繁琐的 SQL 语句。本文将介绍连接数据库并演示如何使用 GORM 实现常见的增删改查操作。

步骤 1:安装 GORM 和数据库驱动

首先,确保你的项目中已经导入了 GORM 库及所需的数据库驱动。你可以使用以下命令安装 GORM 和 SQLite3 驱动(示例中使用 SQLite3 作为数据库):

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

步骤 2:连接数据库

在开始进行任何数据库操作之前,我们需要建立与数据库的连接。创建一个 main.go 文件,导入 GORM 和 SQLite3 驱动,并连接到数据库:

goCopy code
package main

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

func main() {
    // 连接到 SQLite3 数据库
    db, err := gorm.Open(sqlite.Open("test.db"), &gorm.Config{})
    if err != nil {
        log.Fatal(err)
    }
    defer db.Close()
    
    // 数据库自动迁移(创建表)
    db.AutoMigrate(&Product{})
}

步骤 3:定义数据模型

在 GORM 中,我们使用结构体来定义数据模型,结构体字段与数据库表的列相对应。定义一个名为 Product 的数据模型,用于表示数据库中的表。在 main.go 文件下方添加以下代码:

goCopy code
type Product struct {
    gorm.Model
    Code  string
    Price uint
}

步骤 4:实现增删改查操作

接下来,我们将实现基本的增删改查操作。这些操作可以通过 GORM 提供的方法来完成。在 main.go 文件中的 main 函数内,实现增删改查操作:

goCopy code
func main() {
    // ...

    // 创建记录
    db.Create(&Product{Code: "Laptop", Price: 1000})
    
    // 查询单个记录
    var laptop Product
    db.First(&laptop, "code = ?", "Laptop")
    fmt.Println("Product:", laptop)
    
    // 更新记录
    db.Model(&laptop).Update("Price", 1200)
    
    // 删除记录
    db.Delete(&laptop)

    // ...

    // 查询所有记录
    var products []Product
    db.Find(&products)
    fmt.Println("All Products:", products)
}

GORM的特性

  • 全功能 ORM
  • 关联 (Has One,Has Many,Belongs To,Many To Many,多态,单表继承)
  • Create,Save,Update,Delete,Find 中钩子方法
  • 支持 Preload、Joins 的预加载
  • 事务,嵌套事务,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的作用

  • 简化数据库操作 GORM使数据库操作更加易于管理和执行。开发人员可以使用Go的结构体来定义数据库中的表,然后使用GORM提供的方法来执行插入、查询、更新和删除等操作,而无需编写复杂的SQL语句。
  • 提高开发效率 GORM减少了编写大量重复性的SQL代码的需要。这可以显著减少开发时间,并降低了出错的机会。
  • 跨数据库兼容性 GORM抽象了底层数据库的细节,使得可以在不同的数据库系统之间进行切换而不需要大量的修改。只需更改GORM的连接配置即可适应不同的数据库。
  • 支持数据关系映射 GORM支持多对一、一对多、多对多等数据库关系的映射。这意味着你可以通过定义结构体之间的关系来建立数据库表之间的关联,而无需手动管理外键等。
  • 数据验证和校验 GORM提供了数据验证和校验的功能,可以在插入和更新数据之前对数据进行验证,从而保证数据的完整性和一致性。
  • 钩子和回调 GORM支持在不同数据库操作的不同阶段插入自定义逻辑,例如在创建、更新、删除等操作之前或之后执行某些代码。
  • 自动迁移 GORM支持自动根据定义的结构体来创建数据库表,这称为自动迁移。这使得数据库的版本管理和升级变得更加容易。
  • 事务支持 GORM支持数据库事务,可以在一系列操作中保持数据的一致性,确保要么所有操作都成功,要么所有操作都失败。

总结

总之,GORM在Go语言中提供了一个强大的工具,用于管理和操作关系型数据库。它使开发人员能够更专注于业务逻辑而不必过多关注数据库操作的细节,从而提高了开发效率并降低了出错的风险。