如何在GORM中处理多态查询及关联记录删除？

你的模型定义是典型的GORM多态关联结构，但官方文档对多态查询、删除的示例较少。直接用switch判断类型确实会破坏多态性的简洁性，这里提供两种更优雅的实现方式：

方式一：类型映射表+通用处理函数 #

先定义一个全局映射表，将QuestionType字符串与对应的结构体实例绑定：

import "fmt"
import "gorm.io/gorm"

var questionTypeMap = map[string]interface{}{
    "text_question":   &TextQuestion{},
    "choice_question": &ChoiceQuestion{},
}

func DeleteAssociatedQuestion(db *gorm.DB, prompt *Prompt) error {
    // 根据类型获取对应Question模型实例
    questionModel, exists := questionTypeMap[prompt.QuestionType]
    if !exists {
        return fmt.Errorf("不支持的问题类型: %s", prompt.QuestionType)
    }

    // 查询关联的Question记录
    if err := db.Where("id = ?", prompt.QuestionID).First(questionModel).Error; err != nil {
        return err
    }

    // 删除该Question
    return db.Delete(questionModel).Error
}

方式二：接口抽象+类型断言 #

先定义一个统一的Question接口，让所有问题类型实现该接口：

type Question interface {
    GetModelID() uint
}

func (t *TextQuestion) GetModelID() uint {
    return t.ID
}

func (c *ChoiceQuestion) GetModelID() uint {
    return c.ID
}

再结合映射表实现通用的获取与删除逻辑：

func GetAndDeleteQuestion(db *gorm.DB, prompt *Prompt) (Question, error) {
    questionModel, exists := questionTypeMap[prompt.QuestionType]
    if !exists {
        return nil, fmt.Errorf("不支持的问题类型: %s", prompt.QuestionType)
    }

    // 查询关联记录
    if err := db.First(questionModel, prompt.QuestionID).Error; err != nil {
        return nil, err
    }

    // 删除记录
    if err := db.Delete(questionModel).Error; err != nil {
        return nil, err
    }

    return questionModel.(Question), nil
}

注意事项 #

• GORM默认会将结构体名转为蛇形字符串作为多态类型值（比如TextQuestion对应text_question），如果自定义了QuestionType的值，需要同步更新questionTypeMap中的键名。
• 上述函数支持单条Prompt的关联Question处理，批量场景只需遍历Prompt列表调用即可。

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