如何解决gorm中的外键约束
我想创建一个 Booking-Class 关系,其中每个 Booking 只能分配一个 Class。
type Class struct {
Id int `json:"id"`
Name string `json:"name"`
}
type Booking struct {
Id int `json:"id"`
User string `json:"user"`
Members int `json:"members"`
ClassId int `json:"classid"`
}
我知道它类似于此处解释的gorm的“属于”关系https://gorm.io/docs/belongs_to.html,但我想知道是否有可能实现“外键约束”而不在Booking模型中定义Class类型的字段(只有 ClassId int)。为了确保使用不存在的 ClassId-s,我定义了函数:
func Find(slice []int,val int) bool {
for _,item := range slice {
if item == val {
return true
}
}
return false
}
func GetClassKeys(d *gorm.DB) []int {
var keys []int
rows,_ := d.Raw(`SELECT id
FROM classes`).Rows()
defer rows.Close()
for rows.Next() {
var key int
rows.Scan(&key)
keys = append(keys,key)
}
return keys
}
并在创建/更新预订之前执行此检查:
if !Find(GetClassKeys(db),booking.ClassId) {
log.Println("Wrong class id value")
return
}
但这不能处理删除类 id 的情况(常规数据库会自动执行)。我想知道有没有办法通过简单地引用 User 模型中 Class 的主键来使用 gorm 实现正常的数据库外键功能?提前致谢
解决方法
我认为 Migrator 工具不会帮助您,因为它假设您将使用默认的 Gorm 模式来定义关系,而您已明确决定不使用这些模式。
剩下的唯一选择是自己管理约束,通过 SQL 脚本或更简单的方式,您在 AutoMigrate
调用中运行的一些自定义查询:
import "gorm.io/gorm/clause"
// somewhere you must be calling
db.AutoMigrate(&Class{},&Booking{})
// then you'd have:
constraint := "fk_booking_classid"
var count int64
err := db.Raw(
"SELECT count(*) FROM INFORMATION_SCHEMA.table_constraints WHERE constraint_schema = ? AND table_name = ? AND constraint_name = ?",db.Migrator().CurrentDatabase(),"bookings",constraint,).Scan(&count).Error
// handle error
if (count == 0) {
err := db.Exec(
"ALTER TABLE bookings ADD CONSTRAINT ? FOREIGN KEY class_id REFERENCES classes(id)",clause.Table{Name: constraint},).Error
// handle error
}
当然,这否定了自动迁移的优势(这意味着当诸如字段名称之类的事情发生变化时,约束将在不更改迁移代码的情况下更新)。
我会看看为什么你不想像 gorm 期望的那样定义外键,因为这闻到你试图直接使用数据库模型作为你的 JSON API 请求/响应模型,而这通常不会' t导致一个好的结局:)
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。