From 21e9f7c79dc78798c212a8749550ff2701afcfe3 Mon Sep 17 00:00:00 2001
From: maguodong <maguodong@gmail.com>
Date: Sat, 28 Mar 2026 15:58:46 +0800
Subject: [PATCH] =?UTF-8?q?refactor(database):=20=E9=87=8D=E6=9E=84?=
 =?UTF-8?q?=E6=95=B0=E6=8D=AE=E5=BA=93=E5=9F=BA=E7=A1=80=E6=A8=A1=E5=9E=8B?=
 =?UTF-8?q?=E6=97=B6=E9=97=B4=E7=B1=BB=E5=9E=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 将 LocalTime 类型重命名为 DateTime 并改进实现
- 添加完整的 JSON 序列化和反序列化支持
- 实现了 MarshalJSON 和 UnmarshalJSON 方法
- 改进 Scan 方法以支持字符串类型转换
- 更新 BaseModel 结构体字段定义和标签
- 移除指针类型改为值类型并优化 GORM 标签配置
---
 database/model/base.go | 55 ++++++++++++++++++++++++++++++++----------
 1 file changed, 42 insertions(+), 13 deletions(-)

diff --git a/database/model/base.go b/database/model/base.go
index 88f9473..9e38b49 100644
--- a/database/model/base.go
+++ b/database/model/base.go
@@ -2,31 +2,53 @@ package model
 
 import (
 	"database/sql/driver"
+	"encoding/json"
 	"fmt"
 	"time"
 )
 
-// LocalTime 自定义时间类型，支持 null 值
-type LocalTime struct {
+// DateTime 自定义时间类型
+type DateTime struct {
 	time.Time
 }
 
-func (t *LocalTime) MarshalJSON() ([]byte, error) {
-	if t == nil || t.IsZero() {
+// MarshalJSON 实现 JSON 序列化
+func (t DateTime) MarshalJSON() ([]byte, error) {
+	if t.IsZero() {
 		return []byte("null"), nil
 	}
-	formatted := fmt.Sprintf("\"%s\"", t.Format("2006-01-02 15:04:05"))
-	return []byte(formatted), nil
+	return json.Marshal(t.Format("2006-01-02 15:04:05"))
 }
 
-func (t *LocalTime) Value() (driver.Value, error) {
-	if t == nil || t.IsZero() {
+// UnmarshalJSON 实现 JSON 反序列化
+func (t *DateTime) UnmarshalJSON(data []byte) error {
+	var s string
+	if err := json.Unmarshal(data, &s); err != nil {
+		return err
+	}
+	if s == "" || s == "null" {
+		t.Time = time.Time{}
+		return nil
+	}
+
+	parsed, err := time.Parse("2006-01-02 15:04:05", s)
+	if err != nil {
+		return err
+	}
+	t.Time = parsed
+	return nil
+}
+
+// Value 实现 driver.Valuer
+func (t DateTime) Value() (driver.Value, error) {
+	if t.IsZero() {
 		return nil, nil
 	}
 	return t.Time, nil
 }
 
-func (t *LocalTime) Scan(v interface{}) error {
+// Scan 实现 sql.Scanner
+func (t *DateTime) Scan(v interface{}) error {
 	if v == nil {
 		t.Time = time.Time{}
 		return nil
@@ -36,14 +58,21 @@ func (t *LocalTime) Scan(v interface{}) error {
 	case time.Time:
 		t.Time = value
 		return nil
+	case string:
+		parsed, err := time.Parse("2006-01-02 15:04:05", value)
+		if err != nil {
+			return err
+		}
+		t.Time = parsed
+		return nil
 	default:
 		return fmt.Errorf("unsupported type: %T", v)
 	}
 }
 
-// BaseModel 基础模型，包含自定义的创建时间和更新时间
+// 基础模型
 type BaseModel struct {
-	Id         int        `json:"id" gorm:"column:id;primaryKey;autoIncrement;common:主键ID"` // 主键 ID
-	CreateTime *LocalTime `json:"create_time" gorm:"column:create_time;type:datetime;autoCreateTime;common:创建时间"`
-	UpdateTime *LocalTime `json:"update_time" gorm:"column:update_time;type:datetime;autoUpdateTime;common:更新时间"`
+	ID         uint     `gorm:"column:id;primaryKey;autoIncrement;common:主键ID" json:"id"`
+	CreateTime DateTime `gorm:"autoCreateTime;column:create_time;type:datetime" json:"create_time"`
+	UpdateTime DateTime `gorm:"autoUpdateTime;column:update_time;type:datetime" json:"update_time"`
 }