1 package helper
 2 
 3 import (
 4     "database/sql/driver"
 5     "encoding/json"
 6     "fmt"
 7     "time"
 8 )
 9 
10 const CUS_TIME_FORMAT = "2006-01-02 15:04:05"
11 
12 type CustomTime struct {
13     time.Time
14 }
15 
16 func (ct CustomTime) String() string {
17     return ct.Format(CUS_TIME_FORMAT)
18 }
19 
20 func (ct CustomTime) MarshalJSON() ([]byte, error) {
21     return []byte(`"` + ct.Format(CUS_TIME_FORMAT) + `"`), nil
22 }
23 
24 func (t *CustomTime) UnmarshalJSON(data []byte) error {
25     var timeStr string
26     err := json.Unmarshal(data, &timeStr)
27     if err != nil {
28         return err
29     }
30     parsedTime, err := time.Parse(CUS_TIME_FORMAT, timeStr)
31     if err != nil {
32         return err
33     }
34     *t = CustomTime{
35         parsedTime,
36     }
37     return nil
38 }
39 
40 func (ct *CustomTime) Scan(value interface{}) error {
41     switch v := value.(type) {
42     case time.Time:
43         ct.Time = v
44     case string:
45         ct.Time, _ = time.Parse(CUS_TIME_FORMAT, value.(string))
46     case nil:
47         return nil
48     default:
49         return fmt.Errorf("unsupported Scan, storing driver.Value type %T into type *CustomTime", value)
50     }
51     return nil
52 }
53 
54 func (ct CustomTime) Value() (driver.Value, error) {
55     return ct.Time, nil
56 }