添加时间工具包

timeutils/timeutils.go

@@ -0,0 +1,145 @@
+package timeutils
+
+import (
+	"time"
+)
+
+var (
+	monthMap = map[string]string{
+		"01": "一月",
+		"02": "二月",
+		"03": "三月",
+		"04": "四月",
+		"05": "五月",
+		"06": "六月",
+		"07": "七月",
+		"08": "八月",
+		"09": "九月",
+		"10": "十月",
+		"11": "十一月",
+		"12": "十二月",
+	}
+
+	weekMap = map[string]string{
+		"Monday":    "星期一",
+		"Tuesday":   "星期二",
+		"Wednesday": "星期三",
+		"Thursday":  "星期四",
+		"Friday":    "星期五",
+		"Saturday":  "星期六",
+		"Sunday":    "星期日",
+	}
+)
+
+func ParseLocalDateTime(dateTime string) *time.Time {
+	t, err := time.ParseInLocation(time.DateTime, dateTime, time.Local)
+	if err != nil {
+		panic(err)
+	}
+
+	return &t
+}
+
+func ParseLocalDateOnly(date string) *time.Time {
+	t, err := time.ParseInLocation(time.DateOnly, date, time.Local)
+	if err != nil {
+		panic(err)
+	}
+
+	return &t
+}
+
+func ParseNowDateOnly() *time.Time {
+	return ParseLocalDateOnly(FormatNowDateOnly())
+}
+
+func FormatNowDateTime() string {
+	return time.Now().Format(time.DateTime)
+}
+
+func FormatLocalDateTime(t *time.Time) string {
+	if t == nil {
+		return ""
+	}
+
+	return t.Format(time.DateTime)
+}
+
+func FormatNowDateOnly() string {
+	return time.Now().Format(time.DateOnly)
+}
+
+func FormatLocalDateOnly(t *time.Time) string {
+	if t == nil {
+		return ""
+	}
+
+	return t.Format(time.DateOnly)
+}
+
+func GetYear(t *time.Time) int {
+	if t == nil {
+		return 0
+	}
+
+	year, _ := t.ISOWeek()
+	return year
+}
+
+func GetWeek(t *time.Time) int {
+	if t == nil {
+		return 0
+	}
+
+	_, week := t.ISOWeek()
+	return week
+}
+
+func DiffNatureDays(t1str string, t2str string) int {
+	t1 := ParseLocalDateTime(t1str).Unix()
+	t2 := ParseLocalDateTime(t2str).Unix()
+	var secondsOfDay int64 = 86400
+
+	if t1 == t2 {
+		return -1
+	}
+
+	if t1 > t2 {
+		t1, t2 = t2, t1
+	}
+
+	diffDays := 0
+	secDiff := t2 - t1
+	if secDiff > secondsOfDay {
+		tmpDays := int(secDiff / secondsOfDay)
+		t1 += int64(tmpDays) * secondsOfDay
+		diffDays += tmpDays
+	}
+
+	st := time.Unix(t1, 0)
+	et := time.Unix(t2, 0)
+	dateFormatTpl := "20060102"
+	if st.Format(dateFormatTpl) != et.Format(dateFormatTpl) {
+		diffDays += 1
+	}
+
+	return diffDays
+}
+
+func GetChinaMonth(t *time.Time) string {
+	if t == nil {
+		return ""
+	}
+
+	month := t.Format("01")
+	return monthMap[month]
+}
+
+func GetChinaWeek(t *time.Time) string {
+	if t == nil {
+		return ""
+	}
+
+	week := t.Weekday()
+	return weekMap[week.String()]
+}

timeutils/timeutils_test.go

@@ -0,0 +1,107 @@
+package timeutils
+
+import (
+	"testing"
+	"time"
+)
+
+var testDateTimeStr = "2024-02-27 13:49:50"
+var testDateStr = "2024-02-27"
+
+func TestParseLocalTime(t *testing.T) {
+	timeParsed := ParseLocalDateTime(testDateTimeStr)
+
+	if testDateTimeStr != timeParsed.Format(time.DateTime) {
+		t.Fatal("时间解析失败")
+	}
+}
+
+func TestParseLocalDate(t *testing.T) {
+	timeParsed := ParseLocalDateOnly(testDateStr)
+
+	if testDateStr != timeParsed.Format(time.DateOnly) {
+		t.Fatal("时间解析失败")
+	}
+}
+
+func TestParseNowDateOnly(t *testing.T) {
+	now := time.Now().Local()
+
+	if now.Format(time.DateOnly) != ParseNowDateOnly().Format(time.DateOnly) {
+		t.Fatal("时间解析错误")
+	}
+}
+
+func TestFormatNowDateTime(t *testing.T) {
+	now := time.Now().Local()
+
+	if now.Format(time.DateTime) != FormatNowDateTime() {
+		t.Fatal("时间解析错误")
+	}
+}
+
+func TestFormatLocalDateTime(t *testing.T) {
+	now := time.Now().Local()
+
+	if now.Format(time.DateTime) != FormatLocalDateTime(&now) {
+		t.Fatal("时间解析错误")
+	}
+}
+
+func TestFormatNowDate(t *testing.T) {
+	now := time.Now().Local()
+
+	if now.Format(time.DateOnly) != FormatNowDateOnly() {
+		t.Fatal("时间解析错误")
+	}
+}
+
+func TestFormatLocalDate(t *testing.T) {
+	now := time.Now().Local()
+
+	if now.Format(time.DateOnly) != FormatLocalDateOnly(&now) {
+		t.Fatal("时间解析错误")
+	}
+}
+
+func TestGetYear(t *testing.T) {
+	now := time.Now().Local()
+
+	if now.Year() != GetYear(&now) {
+		t.Fatal("时间解析错误")
+	}
+}
+
+func TestGetWeek(t *testing.T) {
+	now := time.Now().Local()
+	_, week := now.ISOWeek()
+
+	if week != GetWeek(&now) {
+		t.Fatal("时间解析错误")
+	}
+}
+
+func TestDiffNatureDays(t *testing.T) {
+	now := time.Now().Local()
+	afterOneDay := now.Add(24 * time.Hour)
+
+	if DiffNatureDays(now.Format(time.DateTime), afterOneDay.Format(time.DateTime)) != 1 {
+		t.Fatal("时间解析错误")
+	}
+}
+
+func TestGetChinaMonth(t *testing.T) {
+	timeParsed := ParseLocalDateTime(testDateTimeStr)
+
+	if GetChinaMonth(timeParsed) != "二月" {
+		t.Fatal("时间解析错误")
+	}
+}
+
+func TestGetChinaWeek(t *testing.T) {
+	timeParsed := ParseLocalDateTime(testDateTimeStr)
+
+	if GetChinaWeek(timeParsed) != "星期二" {
+		t.Fatal("时间解析错误")
+	}
+}