Go?time包AddDate使用解惑實例詳解

更新時間：2022年09月20日 16:17:25 作者：Mr_houzi

這篇文章主要為大家介紹了Go?time包AddDate使用解惑實例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪

引例

假設，今天是10月31日，是10月的最后一天，我們想通過 AddDate()計算下個月的最后一天。

today := time.Date(2022, 10, 31, 0, 0, 0, 0, time.Local)
nextDay := today.AddDate(0, 1, 0)
fmt.Println(nextDay.Format("20060102"))
// 輸出：20221201

結(jié)果輸出：20221201，而非我們預期的下個月最后一天11月30日。

Go Time 包中是這么處理的

AddDate() 對月份+1，即變成了11-31，換算成對應的天數(shù)、最終換算成對應的納秒數(shù)存儲在 Time 對象中；
輸出時，Format()將輸出標準的日期，Time 中的納秒會轉(zhuǎn)為 12-01，而不是 11-31，因為這天并不存在；

只要是涉及到大小月的最后一天都會出現(xiàn)這個問題。

today := time.Date(2022, 3, 31, 0, 0, 0, 0, time.Local)
d := today.AddDate(0, -1, 0)
fmt.Println(d.Format("20060102"))
// 20220303
today := time.Date(2022, 3, 31, 0, 0, 0, 0, time.Local)
d := today.AddDate(0, 1, 0)
fmt.Println(d.Format("20060102"))
// 20220501
today := time.Date(2022, 10, 31, 0, 0, 0, 0, time.Local)
d := today.AddDate(0, -1, 0)
fmt.Println(d.Format("20060102"))
// 20221001
today := time.Date(2022, 10, 31, 0, 0, 0, 0, time.Local)
d := today.AddDate(0, 1, 0)
fmt.Println(d.Format("20060102"))
// 20221201

源碼分析

看一下 Go Time 包具體源碼，仍以開頭10-31 + 1 month的例子為用例。
AddDate()，首先對 month+1，然后調(diào)用Date()處理。

// time/time.go
func (t Time) AddDate(years int, months int, days int) Time {
    year, month, day := t.Date() // 獲取當前年月日
    hour, min, sec := t.Clock() // 獲取當前時分秒
    return Date(year+years, month+Month(months), day+days, hour, min, sec, int(t.nsec()), t.Location())
}

Date()中此時傳入的參數(shù)是

year 2020
month 11
day 31
hour、min、sec、nsec 為運行時的時分秒納秒

d 計算的是絕對紀元到今天之前的天數(shù)：

**d = 今年之前的天數(shù) + 年初到當月之前的天數(shù) + 月初到當天之前的天數(shù)；**

最終，將 d 轉(zhuǎn)換成納秒 + 當天經(jīng)過的納秒存儲在 Time 對象中。

// time/time.go
func Date(year int, month Month, day, hour, min, sec, nsec int, loc *Location) Time {
    ……
    // Compute days since the absolute epoch.
    d := daysSinceEpoch(year)
    // Add in days before this month.
    d += uint64(daysBefore[month-1])
    if isLeap(year) && month >= March {
        d++ // February 29
    }
    // Add in days before today.
    d += uint64(day - 1)
    // Add in time elapsed today.
    abs := d * secondsPerDay
    abs += uint64(hour*secondsPerHour + min*secondsPerMinute + sec)
    ……
    return t
}

對 Date() 輸入2022-11-31和輸入2022-12-01，將得到同樣的 d（天數(shù)）。兩者底層存儲的時候都是一樣的數(shù)據(jù)，F(xiàn)ormat() 時將2022-11-31的Time 格式化成 2022-12-01也就不例外了，輸出當然要顯示讓人看得懂的常規(guī)標準日期嘛。

// 2022-11-31
d = 2022年之前的天數(shù) + 1月到10月的總天數(shù) + 30天
// 2022-12-01
d = 2022年之前的天數(shù) + 1月到11月的總天數(shù) + 0天
  = 2022年之前的天數(shù) + 1月到10月的總天數(shù) + 30天 + 0天

你甚至可以往 Date() 輸入非標準日期2022-11-35，它和標準日期 2022-12-05，將得到同樣的 d （天數(shù)）。
“非標準日期”和“標準日期”就像天平的兩邊，雖然形式不一樣，但他們實際的質(zhì)量（d 天數(shù)）是一樣的。記住這句話，后面有用。

預期偏差

我們弄清楚了原理，但仍然不能接受這個結(jié)果。這樣的結(jié)果是 Go 的 bug 嗎？還是 Go Time 包偷懶了？

然而并不是，恰恰是我們的“預期”出現(xiàn)了問題。

正常來說，我們預期 10-30 + 1 month是 11-30 日，這很合理。那我們?yōu)槭裁催€期待 10-31 + 1 month 也是 11-30 日？僅僅因為 10-31是當前月的最后一天，我們也期待 +1 month 后是下個月的最后一天嗎？

10-30 和 10-31 兩個日期相差一天，進行同樣的 +1 month 操作后，就變成為了同一天。這就像 1 + 10 = 2 + 10 一樣的結(jié)果，這顯然不合理。

Go 目前的處理結(jié)果是正確的，并且他在 AddDate() 注釋中也注明了會處理“溢出”的情況。況且，不止 Go 語言這么處理，PHP 也是這么處理的，見文章令人困惑的strtotime

怎么解決

道理我都懂，但我就是想獲取上/下一個月的最后一天怎么辦？

利用前面源碼分析階段，提到的“天平原理”，就能拿到我們想要的結(jié)果。

today := time.Date(2022, 10, 31, 0, 0, 0, 0, time.Local)
d := today.Day()
// 上個月最后一天
// 10-00 日 等于 9-30 日
day1 := today.AddDate(0, 0, -d)
fmt.Println(day1.Format("20060102"))
// 下個月最后一天
// 12-00 日 等于 11-30 日
day2 := today.AddDate(0, 2, -d)
fmt.Println(day2.Format("20060102"))
// 20220930
// 20221130