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

2023-11-12 00:28
文章标签 实战 使用 项目 关系 定义 gorm gen 表间

本文主要是介绍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。

相关文章

一文详解如何在idea中快速搭建一个Spring Boot项目

一文详解如何在idea中快速搭建一个Spring Boot项目

《一文详解如何在idea中快速搭建一个SpringBoot项目》IntelliJIDEA作为Java开发者的‌首选IDE‌,深度集成SpringBoot支持,可一键生成项目骨架、智能配置依赖,这篇文... 目录前言1、创建项目名称2、勾选需要的依赖3、在setting中检查maven4、编写数据源5、开启热
阅读更多...
Python常用命令提示符使用方法详解

Python常用命令提示符使用方法详解

《Python常用命令提示符使用方法详解》在学习python的过程中,我们需要用到命令提示符(CMD)进行环境的配置,：本文主要介绍Python常用命令提示符使用方法的相关资料,文中通过代码介绍的... 目录一、python环境基础命令【Windows】1、检查Python是否安装2、 查看Python的安
阅读更多...
Python并行处理实战之如何使用ProcessPoolExecutor加速计算

Python并行处理实战之如何使用ProcessPoolExecutor加速计算

《Python并行处理实战之如何使用ProcessPoolExecutor加速计算》Python提供了多种并行处理的方式,其中concurrent.futures模块的ProcessPoolExecu... 目录简介完整代码示例代码解释1. 导入必要的模块2. 定义处理函数3. 主函数4. 生成数字列表5.
阅读更多...
Python中help()和dir()函数的使用

Python中help()和dir()函数的使用

《Python中help()和dir()函数的使用》我们经常需要查看某个对象（如模块、类、函数等）的属性和方法,Python提供了两个内置函数help()和dir(),它们可以帮助我们快速了解代... 目录1. 引言2. help() 函数2.1 作用2.2 使用方法2.3 示例(1) 查看内置函数的帮助(
阅读更多...
Linux脚本(shell)的使用方式

Linux脚本(shell)的使用方式

《Linux脚本(shell)的使用方式》：本文主要介绍Linux脚本(shell)的使用方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录概述语法详解数学运算表达式Shell变量变量分类环境变量Shell内部变量自定义变量：定义、赋值自定义变量：引用、修改、删
阅读更多...
SpringBoot项目配置logback-spring.xml屏蔽特定路径的日志

SpringBoot项目配置logback-spring.xml屏蔽特定路径的日志

《SpringBoot项目配置logback-spring.xml屏蔽特定路径的日志》在SpringBoot项目中,使用logback-spring.xml配置屏蔽特定路径的日志有两种常用方式,文中的... 目录方案一：基础配置（直接关闭目标路径日志）方案二：结合 Spring Profile 按环境屏蔽关
阅读更多...
Java使用HttpClient实现图片下载与本地保存功能

Java使用HttpClient实现图片下载与本地保存功能

《Java使用HttpClient实现图片下载与本地保存功能》在当今数字化时代,网络资源的获取与处理已成为软件开发中的常见需求,其中,图片作为网络上最常见的资源之一,其下载与保存功能在许多应用场景中都... 目录引言一、Apache HttpClient简介二、技术栈与环境准备三、实现图片下载与保存功能1.
阅读更多...
Python中使用uv创建环境及原理举例详解

Python中使用uv创建环境及原理举例详解

《Python中使用uv创建环境及原理举例详解》uv是Astral团队开发的高性能Python工具,整合包管理、虚拟环境、Python版本控制等功能,：本文主要介绍Python中使用uv创建环境及... 目录一、uv工具简介核心特点：二、安装uv1. 通过pip安装2. 通过脚本安装验证安装：配置镜像源（可
阅读更多...
LiteFlow轻量级工作流引擎使用示例详解

LiteFlow轻量级工作流引擎使用示例详解

《LiteFlow轻量级工作流引擎使用示例详解》：本文主要介绍LiteFlow是一个灵活、简洁且轻量的工作流引擎,适合用于中小型项目和微服务架构中的流程编排,本文给大家介绍LiteFlow轻量级工... 目录1. LiteFlow 主要特点2. 工作流定义方式3. LiteFlow 流程示例4. LiteF
阅读更多...
使用Python开发一个现代化屏幕取色器

使用Python开发一个现代化屏幕取色器

《使用Python开发一个现代化屏幕取色器》在UI设计、网页开发等场景中,颜色拾取是高频需求,：本文主要介绍如何使用Python开发一个现代化屏幕取色器,有需要的小伙伴可以参考一下... 目录一、项目概述二、核心功能解析2.1 实时颜色追踪2.2 智能颜色显示三、效果展示四、实现步骤详解4.1 环境配置4.
阅读更多...

Copyright China编程 23002807@qq.com

沪ICP备2023033072号-2