rogee/atomctl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add table functions

Browse Source
This commit is contained in:
yanghao05
2025-05-23 22:27:15 +08:00
parent e83332ea6a
commit 9ddea39084
2 changed files with 22 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
pkg/ast/model/generage.go
View File

@@ -30,6 +30,8 @@ type TableModelParam struct {
CamelTable string // user CamelTable string // user
PascalTable string // User PascalTable string // User
SoftDelete bool SoftDelete bool
HasUpdatedAt bool
HasCreatedAt bool
} }
func Generate(tables []string, transformer Transformer) error { func Generate(tables []string, transformer Transformer) error {
@@ -111,6 +113,12 @@ func Generate(tables []string, transformer Transformer) error {
if strings.Contains(table, "DeletedAt") { if strings.Contains(table, "DeletedAt") {
tableInfo.SoftDelete = true tableInfo.SoftDelete = true
} }
if strings.Contains(table, "UpdatedAt") {
tableInfo.HasUpdatedAt = true
}
if strings.Contains(table, "CreatedAt") {
tableInfo.HasCreatedAt = true
}
items = append(items, tableInfo) items = append(items, tableInfo)

11
pkg/ast/model/table_funcs.go.tpl
View File

@@ -17,7 +17,15 @@ func (m *{{.PascalTable}}) log() *log.Entry {
} }
func (m *{{.PascalTable}}) Create(ctx context.Context) error { func (m *{{.PascalTable}}) Create(ctx context.Context) error {
{{- if .HasCreatedAt}}
m.CreatedAt = time.Now() m.CreatedAt = time.Now()
{{- end}}
{{- if .HasUpdatedAt}}
m.UpdatedAt = time.Now()
{{- end}}
stmt := table.Medias.INSERT(table.{{.PascalTable}}.MutableColumns).MODEL(m).RETURNING(table.Medias.AllColumns) stmt := table.Medias.INSERT(table.{{.PascalTable}}.MutableColumns).MODEL(m).RETURNING(table.Medias.AllColumns)
m.log().WithField("func","Create").Info( stmt.DebugSql()) m.log().WithField("func","Create").Info( stmt.DebugSql())
@@ -46,7 +54,6 @@ func (m *{{.PascalTable}}) BatchCreate(ctx context.Context, models []*{{.PascalT
// if SoftDelete // if SoftDelete
{{- if .SoftDelete }} {{- if .SoftDelete }}
func (m *{{.PascalTable}}) Delete(ctx context.Context) error { func (m *{{.PascalTable}}) Delete(ctx context.Context) error {
stmt := table.{{.PascalTable}}.UPDATE().SET(table.{{.PascalTable}}.DeletedAt.Set(TimestampzT(time.Now()))).WHERE(table.{{.PascalTable}}.ID.EQ(Int(m.ID))) stmt := table.{{.PascalTable}}.UPDATE().SET(table.{{.PascalTable}}.DeletedAt.Set(TimestampzT(time.Now()))).WHERE(table.{{.PascalTable}}.ID.EQ(Int(m.ID)))
m.log().WithField("func", "SoftDelete").Info(stmt.DebugSql()) m.log().WithField("func", "SoftDelete").Info(stmt.DebugSql())
@@ -111,7 +118,9 @@ func (m *{{.PascalTable}}) BatchForceDelete(ctx context.Context, ids []int64) er
} }
func (m *{{.PascalTable}}) Update(ctx context.Context) error { func (m *{{.PascalTable}}) Update(ctx context.Context) error {
{{- if .HasUpdatedAt}}
m.UpdatedAt = time.Now() m.UpdatedAt = time.Now()
{{- end}}
stmt := table.{{.PascalTable}}.UPDATE(table.{{.PascalTable}}.MutableColumns.Except({{.CamelTable}}UpdateExcludeColumns...)).SET(m).WHERE(table.{{.PascalTable}}.ID.EQ(Int(m.ID))).RETURNING(table.{{.PascalTable}}.AllColumns) stmt := table.{{.PascalTable}}.UPDATE(table.{{.PascalTable}}.MutableColumns.Except({{.CamelTable}}UpdateExcludeColumns...)).SET(m).WHERE(table.{{.PascalTable}}.ID.EQ(Int(m.ID))).RETURNING(table.{{.PascalTable}}.AllColumns)
m.log().WithField("func", "Update").Info(stmt.DebugSql()) m.log().WithField("func", "Update").Info(stmt.DebugSql())