Yunlot/models/GTime.go

package models

import (
	"encoding/json"
	"fmt"
	orm2 "github.com/beego/beego/v2/client/orm"
	"time"
)

type Time struct {
	time.Time
}

// MarshalJSON 序列化为JSON
func (t Time) MarshalJSON() ([]byte, error) {
	if t.IsZero() {
		return []byte("\"\""), nil
	}
	stamp := fmt.Sprintf("\"%s\"", t.Format("2006-01-02 15:04:05"))
	return []byte(stamp), nil
}

// UnmarshalJSON 反序列化为JSON
func (t *Time) UnmarshalJSON(data []byte) error {
	var err error
	t.Time, err = time.Parse("2006-01-02 15:04:05", string(data)[1:20])
	return err
}

func (t *Time) String() string {
	data, _ := json.Marshal(t)
	return string(data)
}

func (t *Time) FieldType() int {
	return orm2.TypeDateTimeField
}

func (t *Time) SetRaw(value interface{}) error {
	switch value.(type) {
	case time.Time:
		t.Time = value.(time.Time)
	}
	return nil
}

func (t *Time) RawValue() interface{} {
	str := t.Format("2006-01-02 15:04:05")
	if str == "0001-01-01 00:00:00" {
		return nil
	}
	return str
}

func (t *Time) NowDbTime() Time {
	dbTime := Time{}
	dbTime.Time = time.Now()
	return dbTime
}
func (t *Time) AddDates(years int, months int, days int) Time {
	dbTime := Time{}
	dbTime.Time = t.Time.AddDate(years, months, days)
	return dbTime
}
func TimeParse(value string) (Time, error) {
	dbTime := Time{}
	times, err := time.Parse("2006-01-02 15:04:05", value)
	dbTime.Time = times
	return dbTime, err
}