feat: add generate auto time tag logic

Author: sliveryou

util/db.go

@@ -131,6 +131,14 @@ func getColumnInfos(c *CmdConfig) ([]*ColumnInfo, error) {
 			IsUnsigned: strings.Contains(ct, "unsigned") && !c.DisableUnsigned, IsNullable: in == "YES",
 		}
 
+		if isTimeType(ci.DataType) {
+			switch ci.Name {
+			case createAt:
+				ci.Another = autoCreateTime
+			case updateAt:
+				ci.Another = autoUpdateTime
+			}
+		}
 		ci.Indexes, ci.UniqueIndexes = getColumnIndexInfos(indexInfos, ci.Name)
 		columnInfos = append(columnInfos, &ci)
 	}

util/define.go

@@ -155,6 +155,7 @@ type ColumnInfo struct {
 	IsAutoIncrement bool         `mysql:"EXTRA"`
 	IsUnsigned      bool         `mysql:"COLUMN_TYPE"`
 	IsNullable      bool         `mysql:"IS_NULLABLE"`
+	Another         string       `mysql:"-"`
 	Indexes         []*IndexInfo `mysql:"-"`
 	UniqueIndexes   []*IndexInfo `mysql:"-"`
 }

util/template.go

@@ -12,6 +12,13 @@ import (
 	"github.com/pkg/errors"
 )
 
+const (
+	createAt       = "create_at"
+	updateAt       = "update_at"
+	autoCreateTime = "autoCreateTime:milli"
+	autoUpdateTime = "autoUpdateTime:milli"
+)
+
 var (
 	generator *template.Template
 
@@ -139,3 +146,15 @@ func uniqueStrings(slice []string) []string {
 
 	return result
 }
+
+// isTimeType reports whether the data type is time type.
+func isTimeType(dataType string) bool {
+	timeTypes := []string{"year", "date", "datetime", "time", "timestamp"}
+	for _, timeType := range timeTypes {
+		if dataType == timeType {
+			return true
+		}
+	}
+
+	return false
+}

util/tpl/gormv2.tpl

@@ -7,5 +7,6 @@ gorm:"
 {{- range $i, $v := .UniqueIndexes }}
     {{- if eq $i 0 }};uniqueIndex:{{ $v.Name }}{{ else }},{{ $v.Name }}{{ end }}{{ end -}}
 {{- if .Default }};default:{{ .Default }}{{ end -}}
+{{- if .Another }};{{ .Another }}{{ end -}}
 {{- if .Comment }};comment:{{ .Comment }}{{ end -}}
 "