ZiHangQin/electricity_bill_calc_service
1
0
Fork 0
forked from free-lancers/electricity_bill_calc_service
Code Pull Requests Activity

refactor(utils):改进对于月份差值的计算。

Browse Source
This commit is contained in:
徐涛 2022-08-20 10:35:43 +08:00
parent 8283eca993
commit a2c39263d3
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
utils/time.go
View File

@ -1,6 +1,8 @@
package utils package utils
import "time" import (
"time"
)
func VeryBeginOfDate(date time.Time) time.Time { func VeryBeginOfDate(date time.Time) time.Time {
return time.Date(date.Year(), date.Month(), date.Day(), 0, 0, 0, 0, date.Location()) return time.Date(date.Year(), date.Month(), date.Day(), 0, 0, 0, 0, date.Location())
@ -10,7 +12,7 @@ func VeryEndOfDate(date time.Time) time.Time {
return time.Date(date.Year(), date.Month(), date.Day(), 23, 59, 59, 999999999, date.Location()) return time.Date(date.Year(), date.Month(), date.Day(), 23, 59, 59, 999999999, date.Location())
} }
func IsNextMonth(t1, t2 time.Time) bool { func DifferenceInMonth(t1, t2 time.Time) int {
var differYear, differMonth int var differYear, differMonth int
if t1.After(t2) { if t1.After(t2) {
differYear = t1.Year() - t2.Year() differYear = t1.Year() - t2.Year()
@ -19,5 +21,9 @@ func IsNextMonth(t1, t2 time.Time) bool {
differYear = t2.Year() - t1.Year() differYear = t2.Year() - t1.Year()
differMonth = int(t2.Month() - t1.Month()) differMonth = int(t2.Month() - t1.Month())
} }
return (differMonth == -11 && differYear == 1) || (differMonth == 1 && differYear == 0) return differYear*12 + differMonth
}
func IsNextMonth(t1, t2 time.Time) bool {
return DifferenceInMonth(t1, t2) == 1
} }