快捷導(dǎo)航

Golang拾遺之自定義類型和方法集詳解

更新時間：2023年02月20日 15:27:21 作者：apocelipes

golang拾遺主要是用來記錄一些遺忘了的、平時從沒注意過的golang相關(guān)知識。這篇文章主要整理了一下Golang如何自定義類型和方法集，需要的可以參考一下

golang拾遺主要是用來記錄一些遺忘了的、平時從沒注意過的golang相關(guān)知識。

很久沒更新了，我們先以一個謎題開頭練練手：

package main
 
import (
    "encoding/json"
    "fmt"
    "time"
)
 
type MyTime time.Time
 
func main() {
    myTime := MyTime(time.Now()) // 假設(shè)獲得的時間是 2022年7月20日20：30：00，時區(qū)UTC+8
    res, err := json.Marshal(myTime)
    if err != nil {
        panic(err)
    }
    fmt.Println(string(res))
}

請問上述代碼會輸出什么：

1.編譯錯誤

2.運行時panic

3.{}

4."2022-07-20T20:30:00.135693011+08:00"

很多人一定會選4吧，然而答案是3：

$ go run customize.go

{}

是不是很意外，MyTime就是time.Time，理論上應(yīng)該也實現(xiàn)了json.Marshaler，為什么輸出的是空的呢？

實際上這是最近某個群友遇到的問題，乍一看像是golang的bug，但其實還是沒掌握語言的基本規(guī)則。

在深入下去之前，我們先問自己兩個問題：

MyTime 真的是 Time 類型嗎？
MyTime 真的實現(xiàn)了 json.Marshaler 嗎？

對于問題1，只需要引用spec里的說明即可：

A named type is always different from any other type.

https://go.dev/ref/spec#Type_identity

意思是說，只要是type定義出來的類型，都是不同的（type alias除外），即使他們的underlying type是一樣的，也是兩個不同的類型。

那么問題1的答案就知道了，顯然MyTime不是time.Time。

既然MyTime不是Time，那它是否能用Time類型的method呢？畢竟MyTime的基底類型是Time呀。我們寫段代碼驗證下：

package main
 
import (
    "fmt"
    "time"
)
 
type MyTime time.Time
 
func main() {
    myTime := MyTime(time.Now()) // 假設(shè)獲得的時間是 2022年7月20日20：30：00，時區(qū)UTC+8
    res, err := myTime.MarsharlJSON()
    if err != nil {
            panic(err)
    }
    fmt.Println(string(res))
}

運行結(jié)果：

# command-line-arguments
./checkoutit.go:12:24: myTime.MarsharlJSON undefined (type MyTime has no field or method MarsharlJSON)

現(xiàn)在問題2也有答案了：MyTime沒有實現(xiàn)json.Marshaler。

那么對于一個沒有實現(xiàn)json.Marshaler的類型，json是怎么序列化的呢？這里就不賣關(guān)子了，文檔里有寫，對于沒實現(xiàn)Marshaler的類型，默認(rèn)的流程使用反射獲取所有非export的字段，然后依次序列化，我們再看看time的結(jié)構(gòu)：

type Time struct {
        // wall and ext encode the wall time seconds, wall time nanoseconds,
        // and optional monotonic clock reading in nanoseconds.
        //
        // From high to low bit position, wall encodes a 1-bit flag (hasMonotonic),
        // a 33-bit seconds field, and a 30-bit wall time nanoseconds field.
        // The nanoseconds field is in the range [0, 999999999].
        // If the hasMonotonic bit is 0, then the 33-bit field must be zero
        // and the full signed 64-bit wall seconds since Jan 1 year 1 is stored in ext.
        // If the hasMonotonic bit is 1, then the 33-bit field holds a 33-bit
        // unsigned wall seconds since Jan 1 year 1885, and ext holds a
        // signed 64-bit monotonic clock reading, nanoseconds since process start.
        wall uint64
        ext  int64
 
        // loc specifies the Location that should be used to
        // determine the minute, hour, month, day, and year
        // that correspond to this Time.
        // The nil location means UTC.
        // All UTC times are represented with loc==nil, never loc==&utcLoc.
        loc *Location
}

里面都是非公開字段，所以直接序列化后整個結(jié)果就是{}。當(dāng)然，Time類型自己重新實現(xiàn)了json.Marshaler，所以可以正常序列化成我們期望的值。

而我們的MyTime沒有實現(xiàn)整個接口，所以走了默認(rèn)的序列化流程。

所以我們可以得出一個重要的結(jié)論：從某個類型A派生出的類型B，B并不能獲得A的方法集中的任何一個。

想要B擁有A的所有方法也不是不行，但得和type B A這樣的形式說再見了。

方法一是使用type alias：

- type MyTime time.Time
+ type MyTime = time.Time
 
func main() {
-   myTime := MyTime(time.Now()) // 假設(shè)獲得的時間是 2022年7月20日20：30：00，時區(qū)UTC+8
+   var myTime MyTime = time.Now() // 假設(shè)獲得的時間是 2022年7月20日20：30：00，時區(qū)UTC+8
    res, err := json.Marshal(myTime)
    if err != nil {
        panic(err)
    }
    fmt.Println(string(res))
}

類型別名自如其名，就是創(chuàng)建了一個類型A的別名而沒有定義任何新類型（注意那兩行改動）?，F(xiàn)在MyTime就是Time了，自然也可以直接利用Time的MarshalJSON。

方法二，使用內(nèi)嵌類型：

- type MyTime time.Time
+ type MyTime struct {
+     time.Time
+ }
 
func main() {
-   myTime := MyTime(time.Now()) // 假設(shè)獲得的時間是 2022年7月20日20：30：00，時區(qū)UTC+8
+   myTime := MyTime{time.Now}
    res, err := myTime.MarsharlJSON()
    if err != nil {
            panic(err)
    }
    fmt.Println(string(res))
}

通過將Time嵌入MyTime，MyTime也可以獲得Time類型的方法集。更具體的可以看我之前寫的另一篇文章：golang拾遺：嵌入類型

如果我實在需要派生出一種新的類型呢，通常在我們寫一個通用模塊的時候需要隱藏實現(xiàn)的細(xì)節(jié)，所以想要對原始類型進(jìn)行一定的包裝，這時該怎么辦呢？

實際上我們可以讓MyTime重新實現(xiàn)json.Marshaler：

type MyTime time.Time
 
func (m MyTime) MarshalJSON() ([]byte, error) {
    // 我圖方便就直接復(fù)用Time的了
    return time.Time(m).MarshalJSON()
}
 
func main() {
    myTime := MyTime(time.Now()) // 假設(shè)獲得的時間是 2022年7月20日20：30：00，時區(qū)UTC+8
    res, err := myTime.MarsharlJSON()
    if err != nil {
            panic(err)
    }
    fmt.Println(string(res))
}

這么做看上去違反了DRY原則，其實未必，這里只是示例寫的爛而已，真實場景下往往對派生出來的自定義類型進(jìn)行一些定制，因此序列化函數(shù)里會有額外的一些操作，這樣就和DRY不沖突了。

不管哪一種方案，都可以解決問題，根據(jù)自己的實際需求做選擇即可。

總結(jié)

總結(jié)一下，一個派生自A的自定義類型B，它的方法集中的方法只有兩個來源：