message search (#60)

pkg/util/strings.go

@@ -3,6 +3,7 @@ package util
 import (
 	"fmt"
 	"strconv"
+	"strings"
 	"unicode"
 	"unicode/utf8"
 )
@@ -45,3 +46,26 @@ func IsNumeric(s string) bool {
 func SplitInt64ToTwoInt32(input int64) (int64, int64) {
 	return input & 0xFFFFFFFF, input >> 32
 }
+
+func Str2List(str string, sep string) []string {
+	list := make([]string, 0)
+
+	if str == "" {
+		return list
+	}
+
+	listMap := make(map[string]bool)
+	for _, elem := range strings.Split(str, sep) {
+		elem = strings.TrimSpace(elem)
+		if len(elem) == 0 {
+			continue
+		}
+		if _, ok := listMap[elem]; ok {
+			continue
+		}
+		listMap[elem] = true
+		list = append(list, elem)
+	}
+
+	return list
+}

pkg/util/time.go

@@ -582,8 +582,8 @@ func adjustStartTime(t time.Time, g TimeGranularity) time.Time {
 func adjustEndTime(t time.Time, g TimeGranularity) time.Time {
 	switch g {
 	case GranularitySecond, GranularityMinute, GranularityHour:
-		// 对于精确到秒/分钟/小时的时间，设置为当天结束
-		return time.Date(t.Year(), t.Month(), t.Day(), 23, 59, 59, 999999999, t.Location())
+		// 对于精确到秒/分钟/小时的时间，保持原样
+		return t
 	case GranularityDay:
 		// 精确到天，设置为当天结束
 		return time.Date(t.Year(), t.Month(), t.Day(), 23, 59, 59, 999999999, t.Location())
@@ -634,3 +634,25 @@ func isValidDate(year, month, day int) bool {
 
 	return day <= daysInMonth
 }
+
+func PerfectTimeFormat(start time.Time, end time.Time) string {
+	endTime := end
+
+	// 如果结束时间是某一天的 0 点整，将其减去 1 秒，视为前一天的结束
+	if endTime.Hour() == 0 && endTime.Minute() == 0 && endTime.Second() == 0 && endTime.Nanosecond() == 0 {
+		endTime = endTime.Add(-time.Second) // 减去 1 秒
+	}
+
+	// 判断是否跨年
+	if start.Year() != endTime.Year() {
+		return "2006-01-02 15:04:05" // 完整格式，包含年月日时分秒
+	}
+
+	// 判断是否跨天（但在同一年内）
+	if start.YearDay() != endTime.YearDay() {
+		return "01-02 15:04:05" // 月日时分秒格式
+	}
+
+	// 在同一天内
+	return "15:04:05" // 只显示时分秒
+}