GORM 设计原理 | 青训营笔记

BrianINTJ
79 阅读2分钟

在企业开发中，我们经常会使用到 GORM 这个 ORM 库。本篇文章将从 SQL、插件扩展、ConnlPool 以及 Dialector 分别展开，帮助大家了解 GORM 设计原理，并且思考如何从多维度了解 GORM 来助力企业开发。

  1. SQL

GORM 是一个 ORM 库，它的核心功能就是将数据库中的数据映射到 Go 语言的结构体中。在 GORM 中，我们可以使用链式 API 来构建 SQL 语句，这样可以让我们的代码更加简洁易读。例如：

db.Where("name = ?", "jinzhu").Find(&users)

上面的代码会生成如下的 SQL 语句：

SELECT * FROM users WHERE name = 'jinzhu';

在 GORM 中，我们可以使用各种条件来构建 SQL 语句，例如 AND、OR、NOT 等。同时，GORM 还支持使用原生 SQL 语句，这样我们可以更加灵活地操作数据库。

  1. 插件扩展

GORM 支持插件扩展，这使得我们可以根据自己的需求来定制 GORM。例如，我们可以为 GORM 添加一个日志插件，用于记录所有的数据库操作。这样，在出现问题时，我们可以通过查看日志来定位问题。

要创建一个 GORM 插件，我们需要实现 gorm.Plugin 接口。这个接口只有一个方法：Name() string，用于返回插件的名称。然后，我们可以使用 db.Use() 方法来注册插件。

例如，我们可以创建一个简单的日志插件：

type LoggerPlugin struct{}

func (p LoggerPlugin) Name() string {
    return "logger"
}

func (p LoggerPlugin) Initialize(db *gorm.DB) (err error) {
    db.Callback().Create().Before("gorm:create").Register("logger:before_create", func(scope *gorm.Scope) {
        fmt.Println("Before create:", scope.SQL)
    })
    db.Callback().Create().After("gorm:create").Register("logger:after_create", func(scope *gorm.Scope) {
        fmt.Println("After create:", scope.SQL)
    })
    return
}

然后，我们可以在初始化 GORM 时注册这个插件：

db, err := gorm.Open(sqlite.Open("test.db"), &gorm.Config{})
if err != nil {
    panic("failed to connect database")
}

db.Use(LoggerPlugin{})
  1. ConnlPool

GORM 支持连接池，这可以提高数据库操作的性能。在 GORM 中，我们可以通过 db.SetMaxIdleConns() 和 db.SetMaxOpenConns() 方法来设置连接池的大小。例如：

db, err := gorm.Open(sqlite.Open("test.db"), &gorm.Config{})
if err != nil {
    panic("failed to connect database")
}

sqlDB, err := db.DB()
if err != nil {
    panic(err)
}

sqlDB.SetMaxIdleConns(10)
sqlDB.SetMaxOpenConns(100)
  1. Dialector

GORM 支持多种数据库，例如 MySQL、PostgreSQL、SQLite 等。要使用不同的数据库，我们只需要更换相应的 Dialector 即可。例如，要使用 MySQL，我们可以这样做：

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

func main() {
    dsn := "user:password@tcp(localhost:3306)/dbname?charset=utf8mb4&parseTime=True&loc=Local"
    db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
    if err != nil {
        panic("failed to connect database")
    }
}

总结

通过本篇文章，我们了解了 GORM 的设计原理，包括 SQL、插件扩展、ConnlPool 以及 Dialector。在企业开发中，我们可以根据自己的需求来定制 GORM，提高开发效率。同时，了解 GORM 的设计原理，有助于我们在遇到问题时快速定位问题，提高开发效率。

avatar
文章
阅读
粉丝