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

fivyi3re  于 2023-02-27  发布在  Go
关注(0)|答案(2)|浏览(264)

我在一个小的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是否可能直接返回一个空对象,以避免这种手动检查?

xn1cxnb4

xn1cxnb41#

GORM is 返回空对象;当谈到Go语言的值时,“empty”和“default”是一样的,实际上都被称为零值。在您的情况下,您需要控制JSON输出,而不是GORM返回值。
如果字段的类型包含零值,则可以向字段添加omitempty标记,以便从JSON输出中排除这些字段:

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

对于具有此标记的每个字段,当调用EncodeMarshal时,如果字段包含其零值(例如,如果Name等于"",则Namestring),该字段将不包含在输出中。如果您以这种方式标记所有导出的字段,并且它们都包含零值,输出将是一个空JSON对象{}
还请注意:

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

等同于,但效率明显低于:

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

你的另一个选择是定制marshal函数,类似于:

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

ix0qys7i

ix0qys7i2#

omitempty不适用于类型为time的结构字段。

相关问题