alai-server/backend/utils/datatype.go

97 lines
1.9 KiB
Go
Raw Normal View History

2022-06-29 21:26:05 -04:00
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)
}