gorm之项目实战-使用gen以及定义表间关系

2023-11-12 00:28

本文主要是介绍gorm之项目实战-使用gen以及定义表间关系,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

gorm之项目实战

ER图

Image1

关系整理

  1. 一对一关系:

    • User 和 UserLog: 一个用户对应一个用户日志,通过 User 模型的主键与 UserLog 模型的外键建立一对一关系。
  2. 一对多关系:

    • User 和 Teacher: 一个用户可以对应多个老师,通过 Teacher 模型的外键(UserID)与 User 模型的主键建立一对多关系。
    • User 和 Student: 一个用户可以对应多个学生,通过 Student 模型的外键(UserID)与 User 模型的主键建立一对多关系。
    • Teacher 和 Class: 一个老师可以有多个班级,通过 Class 模型的外键(TeacherID)与 Teacher 模型的主键建立一对多关系。
    • Student 和 StudentClass: 一个学生可以有多个班级,通过 StudentClass 模型的外键(StudentID)与 Student 模型的主键建立一对多关系。
    • Student 和 Attendance: 一个学生可以有多条考勤记录,通过 Attendance 模型的外键(StudentID)与 Student 模型的主键建立一对多关系。
  3. 多对多关系:

    • Student 和 Class: 一个学生可以属于多个班级,一个班级可以有多个学生,通过 StudentClass 模型作为中间表,建立多对多关系。
    • Teacher 和 Class: 一个老师可以教授多个班级,一个班级可以有多个老师,通过 Class 模型的外键(TeacherID)与 Teacher 模型的主键,建立多对多关系。
  4. 一对多逆向关系:

    • Parent 和 Student: 一个家长可以有多个子女,通过 Student 模型的外键(StudentID)与 Parent 模型的主键建立一对多逆向关系。

使用gen自动化生成代码

首先使用gen工具生成代码

package main// gorm gen configureimport ("fmt""gorm.io/driver/mysql""gorm.io/gorm""gorm.io/gen"
)const MySQLDSN = "root:root@tcp(127.0.0.1:3306)/school?charset=utf8mb4&parseTime=True"func connectDB(dsn string) *gorm.DB {db, err := gorm.Open(mysql.Open(dsn))if err != nil {panic(fmt.Errorf("connect db fail: %w", err))}return db
}func main() {// 指定生成代码的具体相对目录(相对当前文件),默认为:./query// 默认生成需要使用WithContext之后才可以查询的代码,但可以通过设置gen.WithoutContext禁用该模式g := gen.NewGenerator(gen.Config{// 默认会在 OutPath 目录生成CRUD代码,并且同目录下生成 model 包// 所以OutPath最终package不能设置为model,在有数据库表同步的情况下会产生冲突// 若一定要使用可以通过ModelPkgPath单独指定model package的名称OutPath: "dao/query",/* ModelPkgPath: "dal/model"*/// gen.WithoutContext:禁用WithContext模式// gen.WithDefaultQuery:生成一个全局Query对象Q// gen.WithQueryInterface:生成Query接口Mode: gen.WithDefaultQuery | gen.WithQueryInterface,})// 通常复用项目中已有的SQL连接配置db(*gorm.DB)// 非必需,但如果需要复用连接时的gorm.Config或需要连接数据库同步表信息则必须设置g.UseDB(connectDB(MySQLDSN))// 从连接的数据库为所有表生成Model结构体和CRUD代码// 也可以手动指定需要生成代码的数据表g.ApplyBasic(g.GenerateAllTable()...)// 执行并生成代码g.Execute()
}

在gen生成的model中定义外键联系

生成的model代码在dao/model

Image2

我们需要在这些Model中定义外键关系,先把User表and Teacher表与Student表关系定义

Image3

一对一关系

首先先确定主表和附表,主表

  • 主表:User,逐渐:UserID

  • 附表:Student,外键:UserID

  • 附表:Teacher,外键:UserID

  • 在model.User中添加两个

Image4

  • 把generate改下,我们要用改过的模型生成query

  •     g.ApplyBasic(model.Student{},model.Teacher{},model.User{},model.UserLog{},model.Class{},model.Course{},model.Attendance{},model.StudentClass{},model.Parent{},)// 执行并生成代码g.Execute()
    

增加用户业务逻辑

这边按UserType新建对应的学生和教师

func CreateUser(c *gin.Context) {var req request.CreateUserRequestif err := c.ShouldBindJSON(&req); err != nil {c.JSON(http.StatusBadRequest, gin.H{"code": http.StatusBadRequest, "error": "创建用户请求失败,无效的请求参数"})return}var user model.Userswitch req.UserType {case "Student":user = model.User{Username:  req.Username,Password:  req.Password,OpenID:    req.OpenID,Avatar:    req.Avatar,LastLogin: time.Now(),UserType:  req.UserType,IsValid:   req.IsValid,Student: model.Student{StudentName: req.Username,},}case "Teacher":user = model.User{Username:  req.Username,Password:  req.Password,OpenID:    req.OpenID,Avatar:    req.Avatar,LastLogin: time.Now(),UserType:  req.UserType,IsValid:   req.IsValid,Teacher: model.Teacher{TeacherName: req.Username,},}default:c.JSON(http.StatusBadRequest, gin.H{"code": http.StatusBadRequest, "error": "无效的用户类型"})return}err := query.User.WithContext(context.Background()).Create(&user)if err != nil {c.JSON(http.StatusInternalServerError, gin.H{"code": http.StatusInternalServerError, "error": fmt.Sprintf("创建用户请求失败,无法创建用户: %v", err)})return}c.JSON(http.StatusOK, gin.H{"code": http.StatusOK, "message": "用户创建成功"})
}

Image5

删除用户业务逻辑

// DeleteUser 处理删除用户请求的函数
func DeleteUser(c *gin.Context) {userIDStr := c.Param("id") // Assuming the route has "id" parametervar User model.UseruserID, err := strconv.ParseInt(userIDStr, 10, 64)if err != nil {c.JSON(http.StatusBadRequest, gin.H{"code": http.StatusBadRequest, "error": "Failed to delete user, invalid user ID"})return}//var user model.Userconfig.GVA_DB.Take(&User, userID)ret := config.GVA_DB.Select("Student").Delete(&User)if err != nil {c.JSON(http.StatusInternalServerError, gin.H{"code": http.StatusInternalServerError, "error": fmt.Sprintf("Failed to delete user, unable to delete user: %v", err)})return}c.JSON(http.StatusOK, gin.H{"code": http.StatusOK, "message": fmt.Sprintf("User deleted successfully, RowsAffected: %v", ret.RowsAffected)})
}

Image6

更新

这里因为是主表,逻辑上之更新注表就好,后面改相应表的在编写逻辑

主要是改改头像什么的


查找

// GetUser 处理获取单个用户请求的函数
func GetUser(c *gin.Context) {userIDStr := c.Param("id") // Assuming the route has "id" parameteru := query.UseruserID, err := strconv.ParseInt(userIDStr, 10, 64)if err != nil {c.JSON(http.StatusBadRequest, gin.H{"code": http.StatusBadRequest, "error": "Failed to get user, invalid user ID"})return}user, err := query.User.WithContext(context.Background()).Where(query.User.UserID.Eq(int32(userID))).Preload(u.Student, u.Teacher).First()if err != nil {c.JSON(http.StatusInternalServerError, gin.H{"code": http.StatusInternalServerError, "error": fmt.Sprintf("Failed to get user, unable to get user: %v", err)})return}c.JSON(http.StatusOK, gin.H{"code": http.StatusOK, "user": user})
}// GetAllUsers 处理获取所有用户请求的函数
func GetAllUsers(c *gin.Context) {u := query.Userusers, err := query.User.WithContext(context.Background()).Preload(u.Student, u.Teacher).Find()if err != nil {c.JSON(http.StatusInternalServerError, gin.H{"code": http.StatusInternalServerError, "error": fmt.Sprintf("Failed to get all users, unable to get user list: %v", err)})return}c.JSON(http.StatusOK, gin.H{"code": http.StatusOK, "users": users})
}

这篇关于gorm之项目实战-使用gen以及定义表间关系的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/393839

相关文章

Java使用ANTLR4对Lua脚本语法校验详解

《Java使用ANTLR4对Lua脚本语法校验详解》ANTLR是一个强大的解析器生成器,用于读取、处理、执行或翻译结构化文本或二进制文件,下面就跟随小编一起看看Java如何使用ANTLR4对Lua脚本... 目录什么是ANTLR?第一个例子ANTLR4 的工作流程Lua脚本语法校验准备一个Lua Gramm

Java Optional的使用技巧与最佳实践

《JavaOptional的使用技巧与最佳实践》在Java中,Optional是用于优雅处理null的容器类,其核心目标是显式提醒开发者处理空值场景,避免NullPointerExce... 目录一、Optional 的核心用途二、使用技巧与最佳实践三、常见误区与反模式四、替代方案与扩展五、总结在 Java

使用Java将DOCX文档解析为Markdown文档的代码实现

《使用Java将DOCX文档解析为Markdown文档的代码实现》在现代文档处理中,Markdown(MD)因其简洁的语法和良好的可读性,逐渐成为开发者、技术写作者和内容创作者的首选格式,然而,许多文... 目录引言1. 工具和库介绍2. 安装依赖库3. 使用Apache POI解析DOCX文档4. 将解析

Qt中QUndoView控件的具体使用

《Qt中QUndoView控件的具体使用》QUndoView是Qt框架中用于可视化显示QUndoStack内容的控件,本文主要介绍了Qt中QUndoView控件的具体使用,具有一定的参考价值,感兴趣的... 目录引言一、QUndoView 的用途二、工作原理三、 如何与 QUnDOStack 配合使用四、自

C++使用printf语句实现进制转换的示例代码

《C++使用printf语句实现进制转换的示例代码》在C语言中,printf函数可以直接实现部分进制转换功能,通过格式说明符(formatspecifier)快速输出不同进制的数值,下面给大家分享C+... 目录一、printf 原生支持的进制转换1. 十进制、八进制、十六进制转换2. 显示进制前缀3. 指

Python列表去重的4种核心方法与实战指南详解

《Python列表去重的4种核心方法与实战指南详解》在Python开发中,处理列表数据时经常需要去除重复元素,本文将详细介绍4种最实用的列表去重方法,有需要的小伙伴可以根据自己的需要进行选择... 目录方法1:集合(set)去重法(最快速)方法2:顺序遍历法(保持顺序)方法3:副本删除法(原地修改)方法4:

使用Python构建一个Hexo博客发布工具

《使用Python构建一个Hexo博客发布工具》虽然Hexo的命令行工具非常强大,但对于日常的博客撰写和发布过程,我总觉得缺少一个直观的图形界面来简化操作,下面我们就来看看如何使用Python构建一个... 目录引言Hexo博客系统简介设计需求技术选择代码实现主框架界面设计核心功能实现1. 发布文章2. 加

在Spring Boot中浅尝内存泄漏的实战记录

《在SpringBoot中浅尝内存泄漏的实战记录》本文给大家分享在SpringBoot中浅尝内存泄漏的实战记录,结合实例代码给大家介绍的非常详细,感兴趣的朋友一起看看吧... 目录使用静态集合持有对象引用,阻止GC回收关键点:可执行代码:验证:1,运行程序(启动时添加JVM参数限制堆大小):2,访问 htt

Node.js 数据库 CRUD 项目示例详解(完美解决方案)

《Node.js数据库CRUD项目示例详解(完美解决方案)》:本文主要介绍Node.js数据库CRUD项目示例详解(完美解决方案),本文给大家介绍的非常详细,对大家的学习或工作具有一定的参考... 目录项目结构1. 初始化项目2. 配置数据库连接 (config/db.js)3. 创建模型 (models/

SQL表间关联查询实例详解

《SQL表间关联查询实例详解》本文主要讲解SQL语句中常用的表间关联查询方式,包括:左连接(leftjoin)、右连接(rightjoin)、全连接(fulljoin)、内连接(innerjoin)、... 目录简介样例准备左外连接右外连接全外连接内连接交叉连接自然连接简介本文主要讲解SQL语句中常用的表