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值，“空”和“默认”是相同的，实际上称为zero value。根据您的情况，您需要控制JSON输出，而不是GORM返回值。

您可以将omitempty标记添加到字段中，以使其在类型中包含零值时从JSON输出中排除：

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

对于带有此标记的每个字段，当您调用Encode或Marshal时，如果该字段包含其零值（例如，对于Name，即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

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

1 个答案: