实践记录以及工具使用5 | 使用 GORM连接数据库

Rookie173
84 阅读4分钟

学习心得：使用 GORM 实现数据库连接及增删改查

在学习Go语言的过程中，数据库操作是必不可少的环节。为了简化数据库操作，提高开发效率，我选择了GORM（Go的对象关系映射库）。GORM为开发者提供了强大的ORM功能，使得我们可以以面向对象的方式操作数据库，极大地减少了重复代码量。本篇心得将整理我使用GORM连接数据库并实现增删改查操作的过程及学习收获。

GORM 简介

GORM 是一个功能丰富且直观的 Go ORM 库，具有以下特点：

  1. 支持多种数据库：包括 MySQL、PostgreSQL、SQLite 等。
  2. 自动迁移：可以根据结构体生成数据库表。
  3. 链式调用：查询操作可以通过链式调用构建，代码更加简洁。
  4. 钩子函数：支持增删改查前后的自定义逻辑。

实现步骤

1. 准备工作

安装 GORM 和数据库驱动

在开始前，需要确保 Go 环境已安装，并安装 GORM 及对应的数据库驱动。例如，这里以 MySQL 为例：

go get -u gorm.io/gorm
go get -u gorm.io/driver/mysql
数据库配置

首先，需要准备一个 MySQL 数据库，创建一个名为 gorm_demo 的数据库：

CREATE DATABASE gorm_demo;
USE gorm_demo;

2. 编写代码实现增删改查

创建项目结构

项目结构如下：

gorm-demo/
├── main.go
├── models/
│   └── user.go
定义数据库模型

在 models/ 目录下，创建 user.go 文件，定义 User 模型：

package models

import "gorm.io/gorm"

// User 表示用户表结构
type User struct {
	ID       uint   `gorm:"primaryKey"`
	Name     string `gorm:"size:100;not null"`
	Email    string `gorm:"uniqueIndex;not null"`
	Password string `gorm:"not null"`
}

GORM 使用结构体表示数据库表，通过标签定义字段属性：

  • gorm:"primaryKey" 表示主键。
  • gorm:"size:100;not null" 限制字符串长度且不能为空。
  • gorm:"uniqueIndex" 表示唯一索引。
初始化数据库连接

在 main.go 中初始化数据库连接：

package main

import (
	"fmt"
	"log"

	"gorm-demo/models"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
)

var db *gorm.DB

func initDB() {
	// 数据库连接配置
	dsn := "root:password@tcp(127.0.0.1:3306)/gorm_demo?charset=utf8mb4&parseTime=True&loc=Local"
	var err error
	db, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
	if err != nil {
		log.Fatalf("连接数据库失败: %v", err)
	}
	fmt.Println("数据库连接成功")

	// 自动迁移
	err = db.AutoMigrate(&models.User{})
	if err != nil {
		log.Fatalf("数据库迁移失败: %v", err)
	}
	fmt.Println("数据库迁移成功")
}
  • gorm.Open 用于连接数据库。
  • AutoMigrate 会根据模型结构自动创建或更新数据库表。
实现增删改查功能

在 main.go 中实现以下功能：

新增数据
func createUser(name, email, password string) {
	user := models.User{Name: name, Email: email, Password: password}
	if err := db.Create(&user).Error; err != nil {
		log.Fatalf("创建用户失败: %v", err)
	}
	fmt.Printf("创建用户成功: %+v\n", user)
}
查询数据
func queryUsers() {
	var users []models.User
	if err := db.Find(&users).Error; err != nil {
		log.Fatalf("查询用户失败: %v", err)
	}
	fmt.Println("用户列表:")
	for _, user := range users {
		fmt.Printf("%+v\n", user)
	}
}
更新数据
func updateUser(id uint, name string) {
	var user models.User
	if err := db.First(&user, id).Error; err != nil {
		log.Fatalf("未找到用户: %v", err)
	}
	user.Name = name
	if err := db.Save(&user).Error; err != nil {
		log.Fatalf("更新用户失败: %v", err)
	}
	fmt.Printf("用户更新成功: %+v\n", user)
}
删除数据
func deleteUser(id uint) {
	if err := db.Delete(&models.User{}, id).Error; err != nil {
		log.Fatalf("删除用户失败: %v", err)
	}
	fmt.Printf("用户删除成功: ID=%d\n", id)
}
主函数调用

将上述功能集成到主函数中：

func main() {
	initDB()

	// 测试功能
	createUser("Alice", "alice@example.com", "password123")
	createUser("Bob", "bob@example.com", "password456")

	queryUsers()

	updateUser(1, "Alice Updated")
	queryUsers()

	deleteUser(2)
	queryUsers()
}

学习体会

1. GORM 的强大功能

GORM 极大简化了数据库操作流程，从表结构设计到增删改查实现，都能通过结构体和链式调用轻松完成。例如，通过 AutoMigrate 可以动态生成数据库表，省去了编写SQL脚本的麻烦。

此外，GORM 支持事务操作、钩子函数、关联查询等高级功能，在复杂项目中可以进一步提升开发效率。

2. Go 语言类型安全的优势

在与数据库交互时，Go语言的类型系统保证了数据的安全性。例如，GORM 会根据结构体字段的类型自动映射到数据库字段，避免了数据类型不匹配的问题。

3. 遇到的挑战与解决

数据库驱动问题

在初始配置中，我遇到了驱动报错问题。通过查阅文档发现，MySQL 驱动需要设置 charset=utf8mb4 来支持 Emoji 等特殊字符。

调试 SQL

GORM 默认屏蔽了生成的SQL语句。为了调试，可以启用日志模式：

db = db.Debug()

总结

通过本次实践，我掌握了使用 GORM 进行数据库操作的基本流程，包括模型定义、数据库连接、自动迁移以及增删改查功能实现。GORM 的易用性和功能丰富性，使得开发过程更加高效且可维护。

在未来的开发中，我将进一步探索 GORM 的高级特性，如事务管理、关联查询等，以应对更复杂的业务需求。

avatar
文章
阅读
粉丝