112 lines
2.8 KiB
Go
112 lines
2.8 KiB
Go
package util
|
||
|
||
import (
|
||
"fmt"
|
||
"strconv"
|
||
"time"
|
||
)
|
||
|
||
func ZeroTimestamp() int64 {
|
||
now := time.Now()
|
||
midnight := time.Date(now.Year(), now.Month(), now.Day(), 0, 0, 0, 0, now.Location())
|
||
return midnight.Unix()
|
||
}
|
||
|
||
func GetHour(ts int64, tz string) int {
|
||
loc, _ := time.LoadLocation(tz)
|
||
return time.Unix(ts, 0).In(loc).Hour()
|
||
}
|
||
|
||
// 将ISO 8601格式的时间字符串转换为指定时区的时间戳
|
||
// timeStr: ISO 8601格式的时间字符串，如："2025-03-17T16:00:00.000Z"
|
||
// timezone: 时区字符串，如："Asia/Shanghai"，"America/New_York"，留空则使用UTC
|
||
func ParseTimeToTimestamp(timeStr string, timezone string) (int64, error) {
|
||
// 如果 timeStr 是纯数字，则直接解析为时间戳返回
|
||
if ts, err := strconv.ParseInt(timeStr, 10, 64); err == nil {
|
||
return ts, nil
|
||
}
|
||
// 先解析为UTC时间
|
||
t, err := time.Parse(time.RFC3339, timeStr)
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
|
||
// 如果没有指定时区，直接返回UTC时间戳
|
||
if timezone == "" {
|
||
return t.Unix(), nil
|
||
}
|
||
|
||
// 加载指定时区
|
||
loc, err := time.LoadLocation(timezone)
|
||
if err != nil {
|
||
return 0, fmt.Errorf("invalid timezone: %v", err)
|
||
}
|
||
|
||
// 将时间转换到指定时区
|
||
t = t.In(loc)
|
||
return t.Unix(), nil
|
||
}
|
||
|
||
// GetZeroTimestamp 获取指定时区的当天0点时间戳
|
||
// timezone: 时区字符串，如："Asia/Shanghai"，"America/New_York"，留空则使用UTC
|
||
// offset: 日期偏移，0表示今天，-1表示昨天，1表示明天，以此类推
|
||
func GetZeroTimestamp(timezone string, offset int) (int64, error) {
|
||
var loc *time.Location
|
||
var err error
|
||
|
||
// 如果没有指定时区，使用UTC
|
||
if timezone == "" {
|
||
loc = time.UTC
|
||
} else {
|
||
// 加载指定时区
|
||
loc, err = time.LoadLocation(timezone)
|
||
if err != nil {
|
||
return 0, fmt.Errorf("invalid timezone: %v", err)
|
||
}
|
||
}
|
||
|
||
// 获取当前时间并转换到指定时区
|
||
now := time.Now().In(loc)
|
||
|
||
// 应用日期偏移
|
||
if offset != 0 {
|
||
now = now.AddDate(0, 0, offset)
|
||
}
|
||
|
||
// 获取当天0点时间
|
||
zero := time.Date(now.Year(), now.Month(), now.Day(), 0, 0, 0, 0, loc)
|
||
|
||
return zero.Unix(), nil
|
||
}
|
||
|
||
// GetDateStr 获取指定时区的日期字符串
|
||
// timezone: 时区字符串，如："Asia/Shanghai"，"America/New_York"，留空则使用UTC
|
||
// offset: 日期偏移，0表示今天，-1表示昨天，1表示明天，以此类推
|
||
// 返回格式："2006-01-02"
|
||
func GetDateStr(timezone string, offset int) (string, error) {
|
||
var loc *time.Location
|
||
var err error
|
||
|
||
// 如果没有指定时区，使用UTC
|
||
if timezone == "" {
|
||
loc = time.UTC
|
||
} else {
|
||
// 加载指定时区
|
||
loc, err = time.LoadLocation(timezone)
|
||
if err != nil {
|
||
return "", fmt.Errorf("invalid timezone: %v", err)
|
||
}
|
||
}
|
||
|
||
// 获取当前时间并转换到指定时区
|
||
now := time.Now().In(loc)
|
||
|
||
// 应用日期偏移
|
||
if offset != 0 {
|
||
now = now.AddDate(0, 0, offset)
|
||
}
|
||
|
||
// 格式化日期
|
||
return now.Format("2006-01-02"), nil
|
||
}
|