first commit

schemas/base_util.go

@@ -0,0 +1,71 @@
+package schemas
+
+import (
+	"database/sql/driver"
+	"encoding/json"
+	"time"
+)
+
+type CustomTime struct {
+	Time *time.Time
+}
+
+// MarshalJSON 实现 JSON 序列化
+func (ct CustomTime) MarshalJSON() ([]byte, error) {
+	if ct.Time == nil {
+		return []byte(`""`), nil
+	}
+	return json.Marshal(ct.Time.Format("2006-01-02 15:04:05"))
+}
+
+// UnmarshalJSON 实现 JSON 反序列化
+func (ct *CustomTime) UnmarshalJSON(data []byte) error {
+	var timeStr string
+	if err := json.Unmarshal(data, &timeStr); err != nil {
+		return err
+	}
+
+	if timeStr == "" {
+		ct.Time = nil
+		return nil
+	}
+
+	parsedTime, err := time.Parse("2006-01-02 15:04:05", timeStr)
+	if err != nil {
+		return err
+	}
+
+	ct.Time = &parsedTime
+	return nil
+}
+
+// Scan 实现 sql.Scanner 接口，用于数据库扫描
+func (ct *CustomTime) Scan(value interface{}) error {
+	if value == nil {
+		ct.Time = nil
+		return nil
+	}
+
+	switch v := value.(type) {
+	case time.Time:
+		ct.Time = &v
+	case *time.Time:
+		ct.Time = v
+	default:
+		ct.Time = nil
+	}
+	return nil
+}
+
+// Value 实现 driver.Valuer 接口，用于数据库存储
+func (ct CustomTime) Value() (driver.Value, error) {
+	if ct.Time == nil {
+		return nil, nil
+	}
+	return *ct.Time, nil
+}
+
+// NewCustomTime 创建新的 CustomTime 实例
+func NewCustomTime(t *time.Time) CustomTime {
+	return CustomTime{Time: t}
+}