43 lines
916 B
Go
43 lines
916 B
Go
package utils
|
|
|
|
import (
|
|
"database/sql/driver"
|
|
"fmt"
|
|
"time"
|
|
)
|
|
|
|
// JSONTime format json time field by myself
|
|
type JSONTime struct {
|
|
time.Time
|
|
}
|
|
|
|
// MarshalJSON on JSONTime format Time field with %Y-%m-%d %H:%M:%S
|
|
func (t JSONTime) MarshalJSON() ([]byte, error) {
|
|
if (t == JSONTime{}) {
|
|
formatted := fmt.Sprintf("\"%s\"", "")
|
|
return []byte(formatted), nil
|
|
} else {
|
|
formatted := fmt.Sprintf("\"%s\"", t.Format("2006-01-02 15:04:05"))
|
|
return []byte(formatted), nil
|
|
}
|
|
}
|
|
|
|
// Value insert timestamp into mysql need this function.
|
|
func (t JSONTime) Value() (driver.Value, error) {
|
|
var zeroTime time.Time
|
|
if t.Time.UnixNano() == zeroTime.UnixNano() {
|
|
return nil, nil
|
|
}
|
|
return t.Time, nil
|
|
}
|
|
|
|
// Scan valueof time.Time
|
|
func (t *JSONTime) Scan(v interface{}) error {
|
|
value, ok := v.(time.Time)
|
|
if ok {
|
|
*t = JSONTime{Time: value}
|
|
return nil
|
|
}
|
|
return fmt.Errorf("can not convert %v to timestamp", v)
|
|
}
|