当找不到值时 Gorm 返回空对象而不是默认对象

Question

我在一个小 go 应用程序中使用 GORM 进行了 MySQL 查询。

我已经声明了我的domain结构

type Domain struct {
    gorm.Model
    Name string
    ...
}

然后当我用这种方法用 GORM 向 MySQL 发送查询时。

func DomainInfos(w http.ResponseWriter, r *http.Request) {
    vars := mux.Vars(r)
    w.WriteHeader(http.StatusOK)

    var d Domain

    config.DbConnection.Where("name = ?", vars["domain"]).Find(&d)

    json.NewEncoder(w).Encode(d)
}

当找不到域时，它会从我的结构定义中返回默认对象

{
    {0 0001-01-01 00:00:00 +0000 UTC 0001-01-01 00:00:00 +0000 UTC <nil>} 
    0
    0
    0
}

所以我写了一些条件来手动返回一个空对象

if d.ID == 0 {
    json.NewEncoder(w).Encode(make(map[string]string))
    return
}

当查询不返回任何内容时，GORM 是否可以直接返回一个空对象，以避免这种手动检查？

Answer 1

GORM返回一个空对象；在 Go 的值中，“空”和“默认”是相同的，实际上都称为零值。在您的情况下，您需要控制 JSON 输出，而不是 GORM 返回值。

您可以将omitempty标记添加到字段中，如果它们的类型包含零值，则将它们从 JSON 输出中排除：

type Domain struct {
    gorm.Model
    Name string `json:",omitempty"`
    ...
}

对于带有此标记的每个字段，当您调用Encodeor时Marshal，如果该字段包含其零值（例如Name，对于 ，它是 a string，如果它等于""），则该字段将不会包含在输出中。如果您以这种方式标记所有导出的字段，并且它们都包含零值，则输出将是一个空的 JSON 对象{}。

另请注意：

json.NewEncoder(w).Encode(make(map[string]string))

相当于，但效率明显低于：

w.Write([]byte("{}"))

您的另一个选择是自定义元帅函数，如下所示：

func (d Domain) MarshalJSON() ([]byte, error) {
    if t.ID == 0 {
        return []byte("{}"), nil
    }

    // Wrap the type to avoid infinite recursion on MarshalJSON
    type dd Domain
    return json.Marshal(dd(d))
}

您可以在此处查看工作示例：https ://play.golang.org/p/mIRfRKXeyyW