mirror of
https://gitee.com/xiangheng/x_admin.git
synced 2025-10-03 07:22:12 +08:00
98 lines
2.3 KiB
Go
98 lines
2.3 KiB
Go
package core
|
||
|
||
import (
|
||
"database/sql/driver"
|
||
"encoding/json"
|
||
"time"
|
||
|
||
"gorm.io/gorm"
|
||
"gorm.io/gorm/schema"
|
||
)
|
||
|
||
const DateFormat = "2006-01-02"
|
||
const TimeFormat = "2006-01-02 15:04:05"
|
||
|
||
// 注解:时间类型从time.Time改为string的原因是struct转interface{}时，丢弃了部分信息，导致导出excel时，时间格式不对
|
||
// TsTime 自定义时间格式
|
||
type TsTime string
|
||
|
||
// 通过时间字符串生成时间戳
|
||
//
|
||
// func ToUnix(date string) int64 {
|
||
// if date == "" {
|
||
// return 0
|
||
// }
|
||
// tt, _ := time.ParseInLocation(TimeFormat, date, time.Local)
|
||
// return tt.Unix()
|
||
// }
|
||
func ParseTimeToTsTime(date time.Time) TsTime {
|
||
return TsTime(date.Format(TimeFormat))
|
||
}
|
||
|
||
func NowTime() TsTime {
|
||
return TsTime(time.Now().Format(TimeFormat))
|
||
}
|
||
|
||
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.Format(TimeFormat),
|
||
)
|
||
return nil
|
||
}
|
||
|
||
// MarshalJSON 将TsTime类型的时间转化为JSON字符串格式
|
||
// 返回转化后的JSON字符串和错误信息
|
||
func (tst TsTime) MarshalJSON() ([]byte, error) {
|
||
// tt := time.Time(tst.Str).Format(TimeFormat)
|
||
tt, _ := time.Parse(TimeFormat, tst.String())
|
||
str := tt.Format(TimeFormat)
|
||
|
||
return json.Marshal(str)
|
||
}
|
||
|
||
// 写入数据库gorm调用
|
||
func (t TsTime) Value() (driver.Value, error) {
|
||
// timeStr := t.String()
|
||
// if timeStr == "0001-01-01 00:00:00" {
|
||
// return nil, nil
|
||
// }
|
||
return t.String(), 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 _, ok := v.(time.Time); ok {
|
||
*t = TsTime(v.(time.Time).Format(TimeFormat))
|
||
return nil
|
||
} else {
|
||
*t = "0001-01-01 00:00:00"
|
||
return nil
|
||
}
|
||
}
|
||
|
||
func (t TsTime) String() string {
|
||
return string(t)
|
||
}
|
||
|
||
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"
|
||
}
|