From 69b32fd04319affec261a54557e4aff1a7ceb5f4 Mon Sep 17 00:00:00 2001 From: hhhhhxm <93307037+hhhhhxm@users.noreply.github.com> Date: Wed, 31 May 2023 17:05:20 +0800 Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9E=E5=88=A4=E6=96=AD=E6=98=AF?= =?UTF-8?q?=E5=90=A6=E6=98=AF=E5=91=A8=E6=9C=AB=E7=9A=84=E6=96=B9=E6=B3=95?= =?UTF-8?q?=20(#105)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit * 新增判断是否是周末的方法 * 新增判断是否是周末的方法 --------- Co-authored-by: huangxingming --- datetime/datetime.go | 4 ++++ datetime/datetime_test.go | 16 ++++++++++++++++ 2 files changed, 20 insertions(+) diff --git a/datetime/datetime.go b/datetime/datetime.go index cc54021..b05ba37 100644 --- a/datetime/datetime.go +++ b/datetime/datetime.go @@ -247,3 +247,7 @@ func DayOfYear(t time.Time) int { return int(nowDate.Sub(firstDay).Hours() / 24) } + +func IsWeekend(t1 time.Time) bool { + return time.Saturday == t1.Weekday() || time.Sunday == t1.Weekday() +} diff --git a/datetime/datetime_test.go b/datetime/datetime_test.go index cba4185..3cbacbe 100644 --- a/datetime/datetime_test.go +++ b/datetime/datetime_test.go @@ -286,3 +286,19 @@ func TestDayOfYear(t *testing.T) { result3 := DayOfYear(date3) assert.Equal(0, result3) } + +func TestIsWeekend(t *testing.T) { + assert := internal.NewAssert(t, "TestIsWeekend") + date := time.Date(2023, 06, 03, 0, 0, 0, 0, time.Local) + result := IsWeekend(date) + assert.Equal(true, result) + + date1 := time.Date(2023, 06, 04, 0, 0, 0, 0, time.Local) + result1 := IsWeekend(date1) + assert.Equal(true, result1) + + date2 := time.Date(2023, 06, 02, 0, 0, 0, 0, time.Local) + result2 := IsWeekend(date2) + assert.Equal(false, result2) + +}