如何将地图转换为 GORM 的关联关系？

在 GORM 中，将地图转换为关联关系需要自定义操作。本文提供了一种方法，通过使用未导出字段（对 GORM 隐藏）和钩子来模拟 hasmany 关系。该策略涉及将地图保留在未导出字段中，并在 afterfind 和 beforesave 钩子中对其进行管理。这样，地图就成为关系的真实来源，并且对它的更改会在保存时反映在关联切片中。通过添加访问器方法，可以从其他包中访问未导出的字段，从而提供与关系类似的接口。

所以我确实有一个结构体 a，它有一个 b 的映射，其中映射的键是 b 的 id，其中 b 还包含它自身的 id。

type A struct {
    gorm.Model
    bValues map[string]*B
}

type B struct {
    gorm.Model
    SomeValue string
}

我怎样才能让grom将这些地图视为关系，就像它们只是切片一样？ gorm 是否从结构体或映射键中获取外键并不重要。

解决方案

gorm 无法开箱即用地执行此操作，但您可以尝试使用一些钩子来实现类似的操作。

策略是按照标准表示 hasmany 关系，然后保留一些未导出（因此对 gorm 隐藏）的字段，即地图：

type a struct {
    gorm.model
    bs      []*b
    bsasmap map[uint]*b  
}

type b struct {
    gorm.model
    aid       uint   
    somevalue string
}

注意 bsasmany 中的小写 b 表示该字段未导出，因此对 gorm 不可见。我使用 uint 类型键，因为这就是 gorm.model 使用的。 aid是拥有这个b的a的id。

afterfind 挂钩在从数据库查询实体的任何操作之后运行。

func (a *a) afterfind(tx *gorm.db) error {
    a.bsasmap = make(map[uint]*b, len(a.bs))
    for _, b := range a.bs {
        a.bsasmap[b.id] = b
    }
    return nil
}

您还需要使用 beforesave 挂钩，以便将 bsasmap 字段中的任何更改保存到 bs 切片：

func (a *a) beforesave(tx *gorm.db) error {
    a.bs = make([]*b, 0, len(a.bsasmap))
    for _, b := range a.bsasmap {
        a.bs = append(a.bs, b)
    }
    return nil
}

这意味着从应用程序代码的角度来看，bsasmap 成为关系的真实来源，并且对 bs 的任何更改都将在保存时撤消。由于 bsasmap 是未导出的字段，因此您无法在其他包中访问它，但您可以通过添加 getb(id uint) *b 和 setb(*b) 等访问器方法来解决这个问题。

import (
    "gorm.io/datatypes"
)

type Alert struct {
    Model
    PrometheusRuleID string `gorm:"comment:关联PrometheusRuleID" example:"123456789"`
    // 一下都是Alert的字段
    Alert       string         `gorm:"alert" example:"etcd"`
    For         string         `gorm:"for" example:"15m"`
    Expr        string         `gorm:"expr" example:"up{job=\"apiserver\"} !=1"`
    Labels      datatypes.JSON `gorm:"labels" example:"{\"severity\": \"warning\"}"`
    Annotations datatypes.JSON `gorm:"annotations" example:"{\"message\": \"Reloading configuration has failed\"}"`
}

尝试datatypes。

以上就是《如何将地图转换为 GORM 的关联关系？》的详细内容，更多关于的资料请关注golang学习网公众号！