97 lines
1.9 KiB
Go
97 lines
1.9 KiB
Go
|
package utils
|
||
|
|
|
||
|
|
import (
|
||
|
|
"database/sql"
|
||
|
|
"database/sql/driver"
|
||
|
|
"encoding/json"
|
||
|
|
"strings"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Date time.Time // 2006-01-02
|
||
|
|
type DateTime time.Time // 2006-01-02 15:04:05
|
||
|
|
|
||
|
|
func (date *Date) UnmarshalJSON(b []byte) error {
|
||
|
|
s := strings.Trim(string(b), "\"")
|
||
|
|
t, err := time.Parse("2006-01-02", s)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
*date = Date(t)
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (date Date) MarshalJSON() ([]byte, error) {
|
||
|
|
return json.Marshal(time.Time(date))
|
||
|
|
}
|
||
|
|
|
||
|
|
func (date Date) Format(s string) string {
|
||
|
|
t := time.Time(date)
|
||
|
|
return t.Format(s)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (date *Date) Scan(value interface{}) (err error) {
|
||
|
|
nullTime := &sql.NullTime{}
|
||
|
|
err = nullTime.Scan(value)
|
||
|
|
*date = Date(nullTime.Time)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func (date Date) Value() (driver.Value, error) {
|
||
|
|
return time.Time(date), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (date Date) GormDataType() string {
|
||
|
|
return "date"
|
||
|
|
}
|
||
|
|
|
||
|
|
func (date Date) GobEncode() ([]byte, error) {
|
||
|
|
return time.Time(date).GobEncode()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (date *Date) GobDecode(b []byte) error {
|
||
|
|
return (*time.Time)(date).GobDecode(b)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (dateTime *DateTime) UnmarshalJSON(b []byte) error {
|
||
|
|
s := strings.Trim(string(b), "\"")
|
||
|
|
t, err := time.Parse("2006-01-02 15:04:05", s)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
*dateTime = DateTime(t)
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (dateTime DateTime) MarshalJSON() ([]byte, error) {
|
||
|
|
return json.Marshal(time.Time(dateTime))
|
||
|
|
}
|
||
|
|
|
||
|
|
func (dateTime DateTime) Format(s string) string {
|
||
|
|
t := time.Time(dateTime)
|
||
|
|
return t.Format(s)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (date *DateTime) Scan(value interface{}) (err error) {
|
||
|
|
nullTime := &sql.NullTime{}
|
||
|
|
err = nullTime.Scan(value)
|
||
|
|
*date = DateTime(nullTime.Time)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func (date DateTime) Value() (driver.Value, error) {
|
||
|
|
return time.Time(date), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (date DateTime) GormDataType() string {
|
||
|
|
return "datetime"
|
||
|
|
}
|
||
|
|
|
||
|
|
func (date DateTime) GobEncode() ([]byte, error) {
|
||
|
|
return time.Time(date).GobEncode()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (date *DateTime) GobDecode(b []byte) error {
|
||
|
|
return (*time.Time)(date).GobDecode(b)
|
||
|
|
}
|