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

feat: update

Browse Source
This commit is contained in:
yanghao05
2025-05-23 22:21:28 +08:00
parent c6b1a1664c
commit ab36ea0e5d
2 changed files with 40 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
pkg/ast/model/table.go.tpl
View File

@@ -1,9 +1,5 @@
package model
import (
log "github.com/sirupsen/logrus"
)
func (m *{{.PascalTable}}) log() *log.Entry {
return log.WithField("model", "{{.PascalTable}}Model")
}
var {{.CamelTable}}UpdateExcludeColumns = []Column{
// table.{{.PascalTable}}.CreatedAt,
}

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

@@ -44,8 +44,40 @@ func (m *{{.PascalTable}}) BatchCreate(ctx context.Context, models []*{{.PascalT
return nil
}
// Delete
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)))
m.log().WithField("func", "SoftDelete").Info(stmt.DebugSql())
if err := stmt.QueryContext(ctx, db, m); err != nil {
m.log().WithField("func","SoftDelete").Errorf("error soft deleting {{.PascalTable}} item: %v", err)
return err
}
m.log().WithField("func", "SoftDelete").Infof("{{.PascalTable}} item soft deleted successfully")
return nil
}
// BatchDelete
func (m *{{.PascalTable}}) BatchDelete(ctx context.Context, ids []int64) error {
condIds := lo.Map(ids, func(id int64, _ int) Expression {
return Int64(id)
})
stmt := table.{{.PascalTable}}.UPDATE().SET(table.{{.PascalTable}}.DeletedAt.Set(TimestampzT(time.Now()))).WHERE(table.{{.PascalTable}}.ID.IN(condIds...))
m.log().WithField("func", "BatchSoftDelete").Info(stmt.DebugSql())
if err := stmt.QueryContext(ctx, db, m); err != nil {
m.log().WithField("func","BatchSoftDelete").Errorf("error soft deleting {{.PascalTable}} items: %v", err)
return err
}
m.log().WithField("func", "BatchSoftDelete").Infof("{{.PascalTable}} items soft deleted successfully")
return nil
}
func (m *{{.PascalTable}}) ForceDelete(ctx context.Context) error {
stmt := table.{{.PascalTable}}.DELETE().WHERE(table.{{.PascalTable}}.ID.EQ(Int(m.ID)))
m.log().WithField("func", "Delete").Info(stmt.DebugSql())
@@ -58,7 +90,7 @@ func (m *{{.PascalTable}}) Delete(ctx context.Context) error {
return nil
}
func (m *{{.PascalTable}}) BatchDelete(ctx context.Context, ids []int64) error {
func (m *{{.PascalTable}}) BatchForceDelete(ctx context.Context, ids []int64) error {
condIds := lo.Map(ids, func(id int64, _ int) Expression {
return Int64(id)
})
@@ -76,7 +108,9 @@ func (m *{{.PascalTable}}) BatchDelete(ctx context.Context, ids []int64) error {
}
func (m *{{.PascalTable}}) Update(ctx context.Context) error {
stmt := table.{{.PascalTable}}.UPDATE(table.{{.PascalTable}}.MutableColumns).SET(m).WHERE(table.{{.PascalTable}}.ID.EQ(Int(m.ID))).RETURNING(table.{{.PascalTable}}.AllColumns)
m.UpdatedAt = time.Now()
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())
if err := stmt.QueryContext(ctx, db, m); err != nil {