Ent框架多字段联合主键关联边报错及多次考试记录实现咨询

我定义了User、Exam、ExamResult三个Ent实体，User与Exam通过ExamResult建立多对多关联以记录用户多次考试结果。我希望将user_id、exam_id、times设为ExamResult的联合主键，但执行generate命令时出现错误：“edge schema primary key can only be defined on "id" or ("user_id", "exam_id") in the same order”。请问遗漏了哪些配置？如何在Ent框架中实现用户每次考试结果的存储需求？

报错的核心原因是：你最初可能用了Ent框架自动生成多对多关联的方式（比如通过Through关键字），这种情况下Ent会把中间表ExamResult视为内部的「关联边缘表（edge table）」，默认限制了它的主键只能是id或者固定组合(user_id, exam_id)，不允许添加额外字段（比如times）到主键中。

要实现「记录用户多次考试结果」的需求，不能依赖Ent自动生成的多对多关联，需要把ExamResult作为独立实体来定义，手动配置关联关系和联合主键。具体步骤如下：

1. 调整实体关联配置 #

删除User、Exam实体中自动生成多对多的Through配置，改为手动建立与ExamResult的一对多关联：

• User实体：添加To类型的Edge，关联到ExamResult（一个用户有多个考试结果）
• Exam实体：添加To类型的Edge，关联到ExamResult（一场考试有多个用户的结果）
• ExamResult实体：添加两个From类型的Edge，分别关联到User和Exam，并指定关联字段user_id、exam_id

2. 配置ExamResult的联合主键 #

在ExamResult的Schema中做以下关键配置：

• 关闭默认的自增id字段（因为我们用自定义联合主键）
• 显式指定联合主键为user_id、exam_id、times
• 为联合主键添加唯一索引（确保数据唯一性）

代码示例 #

ExamResult实体Schema #

package ent

import (
	"time"

	"entgo.io/ent"
	"entgo.io/ent/schema/edge"
	"entgo.io/ent/schema/field"
	"entgo.io/ent/schema/index"
)

type ExamResult struct {
	ent.Schema
}

// Fields 定义考试结果的字段，包含times（考试次数）及其他业务字段
func (ExamResult) Fields() []ent.Field {
	return []ent.Field{
		field.Int("times").Comment("用户参加该考试的次数"),
		field.Float("score").Comment("考试分数"),
		field.String("remark").Optional().Comment("考试备注"),
		field.Time("created_at").Default(time.Now).Comment("考试完成时间"),
	}
}

// Edges 定义与User、Exam的关联
func (ExamResult) Edges() []ent.Edge {
	return []ent.Edge{
		edge.From("user", User.Type).
			Ref("exam_results").
			Field("user_id").  // 指定关联字段名
			Unique().
			Required(),
		edge.From("exam", Exam.Type).
			Ref("exam_results").
			Field("exam_id").  // 指定关联字段名
			Unique().
			Required(),
	}
}

// Indexes 添加联合唯一索引，确保同一用户同一考试的同一次数不会重复
func (ExamResult) Indexes() []ent.Index {
	return []ent.Index{
		index.Fields("user_id", "exam_id", "times").Unique(),
	}
}

// PrimaryKey 自定义联合主键
func (ExamResult) PrimaryKey() []string {
	return []string{"user_id", "exam_id", "times"}
}

// DisableID 关闭默认的自增id字段
func (ExamResult) DisableID() bool {
	return true
}

User实体Schema的Edges部分 #

func (User) Edges() []ent.Edge {
	return []ent.Edge{
		// 替换原有的多对多Through配置，改为一对多关联到ExamResult
		edge.To("exam_results", ExamResult.Type),
	}
}

Exam实体Schema的Edges部分 #

func (Exam) Edges() []ent.Edge {
	return []ent.Edge{
		// 替换原有的多对多Through配置，改为一对多关联到ExamResult
		edge.To("exam_results", ExamResult.Type),
	}
}

验证配置 #

完成上述配置后，重新执行Ent的generate命令：

go generate ./ent

此时就能成功生成代码，且ExamResult的联合主键符合需求，后续可以通过该实体存储用户的多次考试结果。

内容的提问来源于stack exchange，提问作者Marshall Yang