Gorm多对多关联创建后无法返回关联角色数据问题求助

在使用GORM框架创建用户数据时，期望返回包含多对多关联Roles的UserResponse结构体，但实际返回结果中Roles字段始终为nil。以下是相关代码及结果：

结构体定义 #

UserResponse结构体 #

type UserRespone struct {
    ID        int           `json:"id,omitempty"`
    Email     string        `json:"email"`
    FirstName string        `json:"first_name"`
    LastName  string        `json:"last_name"`
    Address   string        `json:"address"`
    Roles     []roles.Roles `json:"roles" gorm:"many2many:users_roles;foreignKey:ID;joinForeignKey:UserID;references:ID;joinReferences:RolesID"`
}

Roles结构体 #

type Roles struct {
    ID   int `json:"id" gorm:"<-:false;primaryKey"`
    Name string
}

创建函数代码 #

func (r *UserRepositoryImpl) AuthSignUp(ctx context.Context, req user.AuthSignUp) (user.UserRespone, error) {
    createdUser := req.ToUser()
    
    hashPassword, _ := bcrypt.GenerateFromPassword([]byte(createdUser.Password), bcrypt.DefaultCost)

    createdUser.Password = string(hashPassword[:])

    result := r.Db.
        WithContext(ctx).
        Create(&createdUser)

    for _, rolesID := range req.RolesID {
        userRole := new(user.UsersRoles)

        userRole.UserID = createdUser.ID
        userRole.RolesID = rolesID
        result = r.Db.WithContext(ctx).Create(&userRole)
    }

    return createdUser.ToResponse(), result.Error
}

结果对比 #

期望返回 #

user.UserResponse{
        ID:        4,
        Email:     "putri4@gmail.com",
        FirstName: "putri",
        LastName:  "cantik",
        Address:   "bonang",
        Roles: []roles.Roles{
            {
                ID: 1,
                Name: "Admin",
            },
            {
                ID: 2,
                Name: "Consumer",
            },
        },
    }

实际返回 #

user.UserRespones{
        ID:        4,
        Email:     "putri4@gmail.com",
        FirstName: "putri",
        LastName:  "cantik",
        Address:   "bonang",
        Roles:     []roles.Roles(nil),
    }

原因分析 #

GORM不会自动加载关联数据，即使手动创建了中间表的关联记录，createdUser对象中也不会存储Roles的信息，需要主动触发关联查询才能填充Roles字段。

解决方案 #

方案1：创建关联后预加载Roles数据 #

修改创建函数，在返回前通过Preload方法查询并加载关联的Roles：

func (r *UserRepositoryImpl) AuthSignUp(ctx context.Context, req user.AuthSignUp) (user.UserRespone, error) {
    createdUser := req.ToUser()
    
    hashPassword, err := bcrypt.GenerateFromPassword([]byte(createdUser.Password), bcrypt.DefaultCost)
    if err != nil {
        return user.UserRespone{}, err
    }
    createdUser.Password = string(hashPassword)

    // 创建用户
    result := r.Db.WithContext(ctx).Create(&createdUser)
    if result.Error != nil {
        return user.UserRespone{}, result.Error
    }

    // 创建中间表关联
    for _, rolesID := range req.RolesID {
        userRole := new(user.UsersRoles)
        userRole.UserID = createdUser.ID
        userRole.RolesID = rolesID
        result = r.Db.WithContext(ctx).Create(&userRole)
        if result.Error != nil {
            return user.UserRespone{}, result.Error
        }
    }

    // 预加载Roles关联数据
    err = r.Db.WithContext(ctx).Preload("Roles").First(&createdUser, createdUser.ID).Error
    if err != nil {
        return user.UserRespone{}, err
    }

    return createdUser.ToResponse(), nil
}

注意：确保createdUser对应的原始User结构体也正确配置了多对多关联（和UserResponse的关联标签一致），否则Preload无法识别关联关系。

方案2：使用GORM的Association方法关联Roles #

跳过手动创建中间表，直接用GORM的关联方法批量添加角色，同时自动加载关联：

func (r *UserRepositoryImpl) AuthSignUp(ctx context.Context, req user.AuthSignUp) (user.UserRespone, error) {
    createdUser := req.ToUser()
    
    hashPassword, err := bcrypt.GenerateFromPassword([]byte(createdUser.Password), bcrypt.DefaultCost)
    if err != nil {
        return user.UserRespone{}, err
    }
    createdUser.Password = string(hashPassword)

    // 创建用户
    result := r.Db.WithContext(ctx).Create(&createdUser)
    if result.Error != nil {
        return user.UserRespone{}, result.Error
    }

    // 根据RolesID查询对应的Roles对象
    var rolesList []roles.Roles
    err = r.Db.WithContext(ctx).Where("id IN ?", req.RolesID).Find(&rolesList).Error
    if err != nil {
        return user.UserRespone{}, err
    }

    // 关联Roles到用户
    err = r.Db.WithContext(ctx).Model(&createdUser).Association("Roles").Append(rolesList)
    if err != nil {
        return user.UserRespone{}, err
    }

    // 加载关联的Roles
    err = r.Db.WithContext(ctx).Preload("Roles").First(&createdUser, createdUser.ID).Error
    if err != nil {
        return user.UserRespone{}, err
    }

    return createdUser.ToResponse(), nil
}

这种方法更符合GORM的使用习惯，无需手动操作中间表，减少出错概率。

内容的提问来源于stack exchange，提问作者Jo Cerdikiawan