package model
|
|
import (
|
"database/sql/driver"
|
"encoding/json"
|
"fmt"
|
"time"
|
)
|
|
type CustomTime time.Time
|
|
const ctLayout = "2006-01-02 15:04:05" // 想要的时间格式
|
|
func (ct *CustomTime) MarshalJSON() ([]byte, error) {
|
return json.Marshal(time.Time(*ct).Format(ctLayout))
|
}
|
|
func (ct *CustomTime) UnmarshalJSON(b []byte) error {
|
var s string
|
if err := json.Unmarshal(b, &s); err != nil {
|
return err
|
}
|
|
t, err := time.Parse(ctLayout, s)
|
if err != nil {
|
return err
|
}
|
|
*ct = CustomTime(t)
|
return nil
|
}
|
|
// Scan 将数据库值扫描到Go中的CustomTime
|
func (ct *CustomTime) Scan(value interface{}) error {
|
if value == nil {
|
*ct = CustomTime(time.Time{})
|
return nil
|
}
|
if t, ok := value.(time.Time); ok {
|
*ct = CustomTime(t)
|
return nil
|
}
|
return fmt.Errorf("can't scan %T into CustomTime", value)
|
}
|
|
// Value 将CustomTime的值转换为数据库值
|
func (ct CustomTime) Value() (driver.Value, error) {
|
if time.Time(ct).IsZero() {
|
return nil, nil
|
}
|
return time.Time(ct), nil
|
}
|
