在 Go 中，自定义时间类型是非常常见的需求，尤其是在处理特殊格式的时间时，或者需要为某些时间类型添加额外的逻辑。Go 的 time 包允许我们通过创建自定义类型来实现这种需求，并且可以通过 time.Time 类型来扩展自定义类型。
Go 的 time 包提供了 time.Time 类型表示时间，但有时我们需要为时间类型添加额外的逻辑或者不同的格式化方式。通过自定义时间类型，可以实现自己的时间逻辑。
假设我们想要定义一个自定义的时间类型 MyTime，并为它提供额外的格式化方法。
Life is like a journey
package main
import (
"fmt"
"time"
)
// 自定义时间类型
type MyTime time.Time
// 为 MyTime 类型实现一个方法，返回自定义格式化的时间
func (mt MyTime) FormatCustom() string {
// 将 MyTime 转换为 time.Time 类型以便调用 Format
t := time.Time(mt)
// 格式化为 "YYYY-MM-DD"
return t.Format("2006-01-02")
}
// 为 MyTime 类型提供一个工厂函数，用于从 string 创建时间
func NewMyTimeFromString(s string) (MyTime, error) {
t, err := time.Parse("2006-01-02", s)
if err != nil {
return MyTime{}, err
}
return MyTime(t), nil
}
func main() {
// 创建自定义时间类型实例
t1 := MyTime(time.Now())
fmt.Println("Custom formatted time:", t1.FormatCustom())
// 使用工厂函数从字符串创建自定义时间
t2, err := NewMyTimeFromString("2025-01-07")
if err != nil {
fmt.Println("Error creating MyTime:", err)
return
}
fmt.Println("Custom formatted time from string:", t2.FormatCustom())
}
自定义时间类型：我们创建了一个 MyTime 类型，它实际上是一个 time.Time 类型的别名。
type MyTime time.Time
通过这种方式，我们可以使用 MyTime 类型来代表时间。
添加方法：为了给 MyTime 类型添加自定义功能，我们定义了一个 FormatCustom 方法。该方法将 MyTime 转换为 time.Time 类型并返回一个自定义格式的字符串。
func (mt MyTime) FormatCustom() string {
t := time.Time(mt)
return t.Format("2006-01-02")
}
工厂函数：我们定义了一个 NewMyTimeFromString 函数，它可以从一个日期字符串（例如 "2025-01-07"） 创建一个 MyTime 类型。
func NewMyTimeFromString(s string) (MyTime, error) {
t, err := time.Parse("2006-01-02", s)
if err != nil {
return MyTime{}, err
}
return MyTime(t), nil
}
MarshalJSON 和 UnmarshalJSON 方法如果你需要让自定义的时间类型支持 JSON 编解码，可以通过实现 MarshalJSON 和 UnmarshalJSON 方法来实现自定义的 JSON 编码和解码方式。
package main
import (
"encoding/json"
"fmt"
"time"
)
// 自定义时间类型
type MyTime time.Time
// 为 MyTime 类型实现 JSON 编码方法
func (mt MyTime) MarshalJSON() ([]byte, error) {
// 将 MyTime 转换为 time.Time
t := time.Time(mt)
// 使用自定义格式化
return json.Marshal(t.Format("2006-01-02"))
}
// 为 MyTime 类型实现 JSON 解码方法
func (mt *MyTime) UnmarshalJSON(b []byte) error {
// 解析自定义格式
str := string(b)
t, err := time.Parse(`"2006-01-02"`, str)
if err != nil {
return err
}
*mt = MyTime(t)
return nil
}
func main() {
// 创建一个 MyTime 实例
t1 := MyTime(time.Now())
// 将 MyTime 转换为 JSON
data, err := json.Marshal(t1)
if err != nil {
fmt.Println("Error marshalling MyTime:", err)
return
}
fmt.Println("MyTime as JSON:", string(data))
// 将 JSON 数据解析为 MyTime
var t2 MyTime
err = json.Unmarshal(data, &t2)
if err != nil {
fmt.Println("Error unmarshalling MyTime:", err)
return
}
fmt.Println("MyTime after unmarshalling:", t2)
}
MyTime 类型实现了 MarshalJSON 方法，方法内部将时间转为 string 类型，然后返回 JSON 编码的字节切片。
func (mt MyTime) MarshalJSON() ([]byte, error) {
t := time.Time(mt)
return json.Marshal(t.Format("2006-01-02"))
}
MyTime 类型实现了 UnmarshalJSON 方法，将从 JSON 数据中读取的时间字符串解析为 MyTime 类型。
func (mt *MyTime) UnmarshalJSON(b []byte) error {
str := string(b)
t, err := time.Parse(`"2006-01-02"`, str)
if err != nil {
return err
}
*mt = MyTime(t)
return nil
}
main 函数中，我们演示了如何将 MyTime 类型实例化、转换为 JSON 字符串，以及从 JSON 字符串解析回 MyTime。time.Time 类型的内存布局和行为。自定义时间类型是 Go 中非常强大的功能，可以帮助你封装时间相关的业务逻辑，同时保持对标准 time.Time 类型的灵活性。通过创建自定义类型并为其添加方法，或者实现 JSON 编解码接口，你可以轻松地处理复杂的时间格式和业务需求。