GOM ORM在GO中 - 多态关联 - 如何惯用亚型访问

Question

我正在使用 Gorm ORM，并在物品和子类型武器/盔甲/珠宝之间设置了多态关联。

type Item struct {
    models.Model
    // see https://gorm.io/docs/has_many.html#Polymorphism-Association
    SubID     string
    SubType   string
    CraftedBy string
}

type ItemWeaponSubtype struct {
    models.Model
    Items           []Item `gorm:"polymorphic:Sub;polymorphicValue:weapon"`
    Name            string
    Quality         string `gorm:"type:varchar(20)""`
    Material        string `gorm:"type:varchar(20)""`
    EquipmentSlotId string
    DamageBonuses
}

我希望能够有一个项目名称列表（例如库存列表）。最终，我希望能够找出所有子类型之间共享的任何其他公共属性（例如重量、成本等）。

我对现有的“解决方案”并不满意，我认为必须有更好的方法来做到这一点。有比我更有经验的人可以向我展示一种实现这一目标的模式吗？

我的想法是拥有一个嵌套函数，能够构建具有公共属性的 DTO。

但对于我想要支持的每种项目类型，我都需要一个 switch 语句。

// ItemCommonDetails contains fields that all subtypes have and is useful for displaying inventory lists etc
type ItemCommonDetails struct {
    Name string
}

func (ir *ItemRepository) GetItemCommonDetailsFromId(itemId string) (ItemCommonDetails, error) {
    var item models.Item
    result := ir.db.First(&item, "id = ?", itemId)
    if 0 == result.RowsAffected {
        return ItemCommonDetails{Name: "Err!"}, &common_dto.StatusError{Code: http.StatusNotFound, Message: "Item [" + itemId + "] not found"}
    }

    defineReturn := func(result *gorm.DB, name string) (ItemCommonDetails, error) {
        if result.RowsAffected == 0 {
            return ItemCommonDetails{Name: "Err!"}, &common_dto.StatusError{Code: http.StatusNotFound, Message: "Item [" + itemId + "] not found"}
        }
        return ItemCommonDetails{Name: name}, nil
    }

    switch item.SubType {
    case "weapon":
        var weapon models.ItemWeaponSubtype
        result := ir.db.First(&weapon, "id = ?", item.SubID)
        return defineReturn(result, weapon.Name)
    case "armor":
        var armor models.ItemArmorSubtype
        result := ir.db.First(&armor, "id = ?", item.SubID)
        return defineReturn(result, armor.Name)
    case "jewelry":
        var jewelry models.ItemJewelrySubtype
        result := ir.db.First(&jewelry, "id = ?", item.SubID)
        return defineReturn(result, jewelry.Name)
    default:
        return ItemCommonDetails{Name: "Err!"}, &common_dto.StatusError{Code: http.StatusNotFound, Message: "Item [" + itemId + "] not found"}
    }
}

有没有更通用的方法来做到这一点？我在 Gorm 文档中找不到任何可以让您神奇地从 Item 中提取子类型的内容。我认为这很难正确输入提示，但也许存在某种反射方法可以让我提取公共属性？

Answer 1

这是一个相当晚的回复，但我建议您也许应该再看看您的表关系。从数据库设计的角度来看，如果您在多个表上共享属性，那么可能值得将这些属性提取到另一个表并让每个原始表引用该表。简单的例子：

// originally, we have Car and Bike sharing the attributes NumWheels and Owner
type Car struct {
  NumWheels int
  Owner string
  Mileage int
  Make string
  Model string
}

type Bike struct {
  NumWheels int
  Owner string
  Color string
  Brand string
}

// one refactoring later...
type Vehicle struct {
  NumWheels int
  Owner string
}

type Car struct {
  Vehicle
  Mileage int
  Make string
  Model string
}

type Bike struct {
  Vehicle
  Color string
  Brand string
}

好的，这是一个简单的多态示例，其中汽车和自行车都是车辆，所以我们可以只使用 go 的结构嵌​​入语法。但如果您考虑项目，也许它们总是具有重量、成本等。您可以简单地将这些属性移至您的 Item 模型中，而不是将它们包含在子类型中。或者，您可以在结构上定义方法（不是 100% Go 用来描述这个概念的语言），以便您可以利用接口：

type Itemlike interface {
  GetWeight() int
  GetCost() int
  // etc
}

func GetProperties(i *Itemlike) map[string]int {
  res := map[string]int{"weight": i.GetWeight(), "cost": i.GetCost()}
  return res
}

显然，后面的示例对查询没有帮助，只有查询后才能使用这个GetProperties函数。因此，您可以查询库存中所有内容的完整列表，然后对该列表中的每个项目调用 GetProperties （或类似）函数