mirror of
https://gitee.com/xiangheng/x_admin.git
synced 2025-10-05 08:07:06 +08:00
112 lines
2.6 KiB
Go
112 lines
2.6 KiB
Go
package core
|
|
|
|
import (
|
|
"database/sql/driver"
|
|
"encoding/json"
|
|
"fmt"
|
|
"time"
|
|
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/schema"
|
|
)
|
|
|
|
const DateFormat = "2006-01-02"
|
|
const TimeFormat = "2006-01-02 15:04:05"
|
|
|
|
// TsTime 自定义时间格式
|
|
type TsTime time.Time
|
|
type OnlyRespTsTime time.Time
|
|
|
|
// func (tst *TsTime) UnmarshalJSON(bs []byte) error {
|
|
// var date string
|
|
// err := json.Unmarshal(bs, &date)
|
|
// if err != nil {
|
|
// return err
|
|
// }
|
|
// tt, _ := time.ParseInLocation(TimeFormat, date, time.Local)
|
|
// *tst = TsTime(tt.Unix())
|
|
// return nil
|
|
// }
|
|
|
|
// // MarshalJSON 将TsTime类型的时间转化为JSON字符串格式
|
|
// // 返回转化后的JSON字符串和错误信息
|
|
// func (tst TsTime) MarshalJSON() ([]byte, error) {
|
|
// tt := time.Unix(int64(tst), 0).Format(TimeFormat)
|
|
// return json.Marshal(tt)
|
|
// }
|
|
|
|
// 通过时间字符串生成时间戳
|
|
func ToUnix(date string) int64 {
|
|
if date == "" {
|
|
return 0
|
|
}
|
|
tt, _ := time.ParseInLocation(TimeFormat, date, time.Local)
|
|
return tt.Unix()
|
|
}
|
|
func NowTime() TsTime {
|
|
return TsTime(time.Now())
|
|
}
|
|
|
|
func (otst OnlyRespTsTime) MarshalJSON() ([]byte, error) {
|
|
tt := time.Time(otst).Format(TimeFormat)
|
|
return json.Marshal(tt)
|
|
}
|
|
|
|
// type TsTime time.Time
|
|
|
|
func (tst *TsTime) UnmarshalJSON(bs []byte) error {
|
|
var date string
|
|
err := json.Unmarshal(bs, &date)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
tt, _ := time.ParseInLocation(TimeFormat, date, time.Local)
|
|
*tst = TsTime(tt)
|
|
return nil
|
|
}
|
|
|
|
// MarshalJSON 将TsTime类型的时间转化为JSON字符串格式
|
|
// 返回转化后的JSON字符串和错误信息
|
|
func (tst TsTime) MarshalJSON() ([]byte, error) {
|
|
tt := time.Time(tst).Format(TimeFormat)
|
|
return json.Marshal(tt)
|
|
}
|
|
|
|
// 写入数据库gorm调用
|
|
func (t TsTime) Value() (driver.Value, error) {
|
|
timeStr := t.String()
|
|
if timeStr == "0001-01-01 00:00:00" {
|
|
return nil, nil
|
|
}
|
|
return timeStr, nil
|
|
}
|
|
|
|
// 读取数据gorm调用
|
|
func (t *TsTime) Scan(v any) error {
|
|
// pt, err := time.ParseInLocation("2006-01-02 15:04:05", v.(time.Time).String(), time.Local)
|
|
if pt, ok := v.(time.Time); ok {
|
|
*t = TsTime(pt)
|
|
return nil
|
|
}
|
|
return fmt.Errorf("cant convert %s to time", v)
|
|
}
|
|
|
|
func (t TsTime) String() string {
|
|
return time.Time(t).Format(TimeFormat)
|
|
}
|
|
|
|
func (TsTime) GormDBDataType(db *gorm.DB, field *schema.Field) string {
|
|
// 使用 field.Tag、field.TagSettings 获取字段的 tag
|
|
// 查看 https://github.com/go-gorm/gorm/blob/master/schema/field.go 获取全部的选项
|
|
|
|
// 根据不同的数据库驱动返回不同的数据类型
|
|
// switch db.Dialector.Name() {
|
|
// case "mysql", "sqlite":
|
|
// return "JSON"
|
|
// case "postgres":
|
|
// return "JSONB"
|
|
// }
|
|
// return ""
|
|
return "DATETIME"
|
|
}
|