Merge pull request #1092 from tiansin/master

feat: 🎨 some time features.
2024-11-30 03:07:45 +08:00 · 2021-01-14 00:15:44 +08:00 · 2021-01-14 00:15:44 +08:00 · 6a67dceca7
commit 6a67dceca7
parent 9c8a68f742 55a7c01f73
2 changed files with 230 additions and 0 deletions
--- a/os/gtime/gtime_time.go
+++ b/os/gtime/gtime_time.go
@ -318,6 +318,108 @@ func (t *Time) Sub(u *Time) time.Duration {
 	return t.Time.Sub(u.Time)
 }

+// StartOfMinute Modify to start of current minute, seconds become 0
+func (t *Time) StartOfMinute() *Time {
+	newTime := t.Clone()
+	newTime.Time = newTime.Time.Truncate(time.Minute)
+	return newTime
+}
+
+// StartOfHour Modify to start of current hour, minutes and seconds become 0
+func (t *Time) StartOfHour() *Time {
+	y, m, d := t.Date()
+	newTime := t.Clone()
+	newTime.Time = time.Date(y, m, d, newTime.Time.Hour(), 0, 0, 0, newTime.Time.Location())
+	return newTime
+}
+
+// StartOfDay Resets the time to 00:00:00 start of day
+func (t *Time) StartOfDay() *Time {
+	y, m, d := t.Date()
+	newTime := t.Clone()
+	newTime.Time = time.Date(y, m, d, 0, 0, 0, 0, newTime.Time.Location())
+	return newTime
+}
+
+// StartOfWeek Resets the date to the first day of week and the time to 00:00:00
+func (t *Time) StartOfWeek() *Time {
+	weekday := int(t.Weekday())
+	return t.StartOfDay().AddDate(0, 0, -weekday)
+}
+
+// StartOfMonth Resets the date to the first day of the month and the time to 00:00:00
+func (t *Time) StartOfMonth() *Time {
+	y, m, _ := t.Date()
+	newTime := t.Clone()
+	newTime.Time = time.Date(y, m, 1, 0, 0, 0, 0, newTime.Time.Location())
+	return newTime
+}
+
+// StartOfQuarter Resets the date to the first day of the quarter and the time to 00:00:00
+func (t *Time) StartOfQuarter() *Time {
+	month := t.StartOfMonth()
+	offset := (int(month.Month()) - 1) % 3
+	return month.AddDate(0, -offset, 0)
+}
+
+// StartOfHalf Resets the date to the first day of the half year and the time to 00:00:00
+func (t *Time) StartOfHalf() *Time {
+	month := t.StartOfMonth()
+	offset := (int(month.Month()) - 1) % 6
+	return month.AddDate(0, -offset, 0)
+}
+
+// StartOfYear Resets the date to the first day of the year and the time to 00:00:00
+func (t *Time) StartOfYear() *Time {
+	y, _, _ := t.Date()
+	newTime := t.Clone()
+	newTime.Time = time.Date(y, time.January, 1, 0, 0, 0, 0, newTime.Time.Location())
+	return newTime
+}
+
+// EndOfMinute Modify to end of current minute, seconds become 59
+func (t *Time) EndOfMinute() *Time {
+	return t.StartOfMinute().Add(time.Minute - time.Nanosecond)
+}
+
+// EndOfHour Modify to end of current hour, minutes and seconds become 59
+func (t *Time) EndOfHour() *Time {
+	return t.StartOfHour().Add(time.Hour - time.Nanosecond)
+}
+
+// EndOfDay Resets the time to 23:59:59 end of day
+func (t *Time) EndOfDay() *Time {
+	y, m, d := t.Date()
+	newTime := t.Clone()
+	newTime.Time = time.Date(y, m, d, 23, 59, 59, int(time.Second-time.Nanosecond), newTime.Time.Location())
+	return newTime
+}
+
+// EndOfWeek Resets the date to end of week and time to 23:59:59
+func (t *Time) EndOfWeek() *Time {
+	return t.StartOfWeek().AddDate(0, 0, 7).Add(-time.Nanosecond)
+}
+
+// EndOfMonth Resets the date to end of the month and time to 23:59:59
+func (t *Time) EndOfMonth() *Time {
+	return t.StartOfMonth().AddDate(0, 1, 0).Add(-time.Nanosecond)
+}
+
+// EndOfQuarter Resets the date to end of the quarter and time to 23:59:59
+func (t *Time) EndOfQuarter() *Time {
+	return t.StartOfQuarter().AddDate(0, 3, 0).Add(-time.Nanosecond)
+}
+
+// EndOfHalf Resets the date to the end of the half year and the time to 23:59:59
+func (t *Time) EndOfHalf() *Time {
+	return t.StartOfHalf().AddDate(0, 6, 0).Add(-time.Nanosecond)
+}
+
+// EndOfYear Resets the date to end of the year and time to 23:59:59
+func (t *Time) EndOfYear() *Time {
+	return t.StartOfYear().AddDate(1, 0, 0).Add(-time.Nanosecond)
+}
+
 // MarshalJSON implements the interface MarshalJSON for json.Marshal.
 func (t *Time) MarshalJSON() ([]byte, error) {
 	return []byte(`"` + t.String() + `"`), nil
--- a/os/gtime/gtime_z_unit_time_test.go
+++ b/os/gtime/gtime_z_unit_time_test.go
@ -256,3 +256,131 @@ func Test_Truncate(t *testing.T) {
 		t.Assert(timeTemp.UnixNano(), timeTemp1.Truncate(time.Hour).UnixNano())
 	})
 }
+
+func Test_StartOfMinute(t *testing.T) {
+	gtest.C(t, func(t *gtest.T) {
+		timeTemp := gtime.NewFromStr("2020-12-12 18:24:06")
+		timeTemp1 := timeTemp.StartOfMinute()
+		t.Assert(timeTemp1.Format("Y-m-d H:i:s"), "2020-12-12 18:24:00")
+	})
+}
+
+func Test_EndOfMinute(t *testing.T) {
+	gtest.C(t, func(t *gtest.T) {
+		timeTemp := gtime.NewFromStr("2020-12-12 18:24:06")
+		timeTemp1 := timeTemp.EndOfMinute()
+		t.Assert(timeTemp1.Format("Y-m-d H:i:s"), "2020-12-12 18:24:59")
+	})
+}
+
+func Test_StartOfHour(t *testing.T) {
+	gtest.C(t, func(t *gtest.T) {
+		timeTemp := gtime.NewFromStr("2020-12-12 18:24:06")
+		timeTemp1 := timeTemp.StartOfHour()
+		t.Assert(timeTemp1.Format("Y-m-d H:i:s"), "2020-12-12 18:00:00")
+	})
+}
+
+func Test_EndOfHour(t *testing.T) {
+	gtest.C(t, func(t *gtest.T) {
+		timeTemp := gtime.NewFromStr("2020-12-12 18:24:06")
+		timeTemp1 := timeTemp.EndOfHour()
+		t.Assert(timeTemp1.Format("Y-m-d H:i:s"), "2020-12-12 18:59:59")
+	})
+}
+
+func Test_StartOfDay(t *testing.T) {
+	gtest.C(t, func(t *gtest.T) {
+		timeTemp := gtime.NewFromStr("2020-12-12 18:24:06")
+		timeTemp1 := timeTemp.StartOfDay()
+		t.Assert(timeTemp1.Format("Y-m-d H:i:s"), "2020-12-12 00:00:00")
+	})
+}
+
+func Test_EndOfDay(t *testing.T) {
+	gtest.C(t, func(t *gtest.T) {
+		timeTemp := gtime.NewFromStr("2020-12-12 18:24:06")
+		timeTemp1 := timeTemp.EndOfDay()
+		t.Assert(timeTemp1.Format("Y-m-d H:i:s"), "2020-12-12 23:59:59")
+	})
+}
+
+func Test_StartOfWeek(t *testing.T) {
+	gtest.C(t, func(t *gtest.T) {
+		timeTemp := gtime.NewFromStr("2020-12-12 18:24:06")
+		timeTemp1 := timeTemp.StartOfWeek()
+		t.Assert(timeTemp1.Format("Y-m-d H:i:s"), "2020-12-06 00:00:00")
+	})
+}
+
+func Test_EndOfWeek(t *testing.T) {
+	gtest.C(t, func(t *gtest.T) {
+		timeTemp := gtime.NewFromStr("2020-12-12 18:24:06")
+		timeTemp1 := timeTemp.EndOfWeek()
+		t.Assert(timeTemp1.Format("Y-m-d H:i:s"), "2020-12-12 23:59:59")
+	})
+}
+
+func Test_StartOfMonth(t *testing.T) {
+	gtest.C(t, func(t *gtest.T) {
+		timeTemp := gtime.NewFromStr("2020-12-12 18:24:06")
+		timeTemp1 := timeTemp.StartOfMonth()
+		t.Assert(timeTemp1.Format("Y-m-d H:i:s"), "2020-12-01 00:00:00")
+	})
+}
+
+func Test_EndOfMonth(t *testing.T) {
+	gtest.C(t, func(t *gtest.T) {
+		timeTemp := gtime.NewFromStr("2020-12-12 18:24:06")
+		timeTemp1 := timeTemp.EndOfMonth()
+		t.Assert(timeTemp1.Format("Y-m-d H:i:s"), "2020-12-31 23:59:59")
+	})
+}
+
+func Test_StartOfQuarter(t *testing.T) {
+	gtest.C(t, func(t *gtest.T) {
+		timeTemp := gtime.NewFromStr("2020-12-06 18:24:06")
+		timeTemp1 := timeTemp.StartOfQuarter()
+		t.Assert(timeTemp1.Format("Y-m-d H:i:s"), "2020-10-01 00:00:00")
+	})
+}
+
+func Test_EndOfQuarter(t *testing.T) {
+	gtest.C(t, func(t *gtest.T) {
+		timeTemp := gtime.NewFromStr("2020-12-06 18:24:06")
+		timeTemp1 := timeTemp.EndOfQuarter()
+		t.Assert(timeTemp1.Format("Y-m-d H:i:s"), "2020-12-31 23:59:59")
+	})
+}
+
+func Test_StartOfHalf(t *testing.T) {
+	gtest.C(t, func(t *gtest.T) {
+		timeTemp := gtime.NewFromStr("2020-12-06 18:24:06")
+		timeTemp1 := timeTemp.StartOfHalf()
+		t.Assert(timeTemp1.Format("Y-m-d H:i:s"), "2020-07-01 00:00:00")
+	})
+}
+
+func Test_EndOfHalf(t *testing.T) {
+	gtest.C(t, func(t *gtest.T) {
+		timeTemp := gtime.NewFromStr("2020-12-06 18:24:06")
+		timeTemp1 := timeTemp.EndOfHalf()
+		t.Assert(timeTemp1.Format("Y-m-d H:i:s"), "2020-12-31 23:59:59")
+	})
+}
+
+func Test_StartOfYear(t *testing.T) {
+	gtest.C(t, func(t *gtest.T) {
+		timeTemp := gtime.NewFromStr("2020-12-06 18:24:06")
+		timeTemp1 := timeTemp.StartOfYear()
+		t.Assert(timeTemp1.Format("Y-m-d H:i:s"), "2020-01-01 00:00:00")
+	})
+}
+
+func Test_EndOfYear(t *testing.T) {
+	gtest.C(t, func(t *gtest.T) {
+		timeTemp := gtime.NewFromStr("2020-12-06 18:24:06")
+		timeTemp1 := timeTemp.EndOfYear()
+		t.Assert(timeTemp1.Format("Y-m-d H:i:s"), "2020-12-31 23:59:59")
+	})
+}