63 lines
979 B
Go
63 lines
979 B
Go
package helpers
|
|
|
|
import (
|
|
"encoding/json"
|
|
"time"
|
|
)
|
|
|
|
type DateOnly struct {
|
|
time.Time
|
|
}
|
|
|
|
func (d DateOnly) MarshalJSON() ([]byte, error) {
|
|
if d.IsZero() {
|
|
return []byte("null"), nil
|
|
}
|
|
return json.Marshal(d.Format("2006-01-02"))
|
|
}
|
|
|
|
func (d *DateOnly) UnmarshalJSON(b []byte) error {
|
|
var s string
|
|
if err := json.Unmarshal(b, &s); err != nil {
|
|
return err
|
|
}
|
|
if s == "" {
|
|
return nil
|
|
}
|
|
t, err := time.Parse("2006-01-02", s)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
d.Time = t
|
|
return nil
|
|
}
|
|
|
|
type KontorTime struct {
|
|
time.Time
|
|
}
|
|
|
|
const layout = "2006-01-02 15:04:05.000000"
|
|
|
|
func (kt KontorTime) MarshalJSON() ([]byte, error) {
|
|
if kt.IsZero() {
|
|
return []byte("null"), nil
|
|
}
|
|
return json.Marshal(kt.Format(layout))
|
|
}
|
|
|
|
func (kt *KontorTime) UnmarshalJSON(b []byte) error {
|
|
var s string
|
|
if err := json.Unmarshal(b, &s); err != nil {
|
|
return err
|
|
}
|
|
if s == "" {
|
|
return nil
|
|
}
|
|
t, err := time.Parse(layout, s)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
kt.Time = t
|
|
return nil
|
|
}
|