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:
Rogee
2025-05-23 23:42:18 +08:00
parent b4cc2347e5
commit 906858dbd6
3 changed files with 47 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -106,18 +106,12 @@ func Generate(tables []string, transformer Transformer) error {
}
tableInfo := TableModelParam{
CamelTable: lo.CamelCase(table),
PascalTable: lo.PascalCase(table),
PkgName: gomod.GetModuleName(),
}
if strings.Contains(table, "DeletedAt") {
tableInfo.SoftDelete = true
}
if strings.Contains(table, "UpdatedAt") {
tableInfo.HasUpdatedAt = true
}
if strings.Contains(table, "CreatedAt") {
tableInfo.HasCreatedAt = true
CamelTable: lo.CamelCase(table),
PascalTable: lo.PascalCase(table),
PkgName: gomod.GetModuleName(),
SoftDelete: strings.Contains(modelContent[table], "DeletedAt"),
HasUpdatedAt: strings.Contains(modelContent[table], "UpdatedAt"),
HasCreatedAt: strings.Contains(modelContent[table], "CreatedAt"),
}
items = append(items, tableInfo)

18
pkg/ast/model/provider.gen.go.tpl
View File

@@ -15,8 +15,22 @@ import (
type Cond func(BoolExpression) BoolExpression
func CondDefault() BoolExpression {
return BoolExp(Bool(true))
func ExprCond(expr BoolExpression) Cond {
return func(cond BoolExpression) BoolExpression {
return cond.AND(expr)
}
}
func CondTrue(conds ...Cond) BoolExpression {
cond:= BoolExp(Bool(true))
for _, c := range conds {
cond = c(cond)
}
return cond
}
func CondJoin(cond Cond, conds ...Cond) []Cond {
return append([]Cond{cond}, conds...)
}

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

@@ -12,14 +12,18 @@ import (
)
// conds
{{- if .SoftDelete }}
func (m *{{.PascalTable}}) NotDeleted() Cond {
func (m *{{.PascalTable}}) CondNotDeleted() Cond {
return func(cond BoolExpression) BoolExpression {
return cond.AND(table.Posts.DeletedAt.IS_NULL())
return cond.AND(table.{{.PascalTable}}.DeletedAt.IS_NULL())
}
}
{{- end}}
func (m *{{.PascalTable}}) CondID(id int64) Cond {
return func(cond BoolExpression) BoolExpression {
return cond.AND(table.{{.PascalTable}}.ID.EQ(Int(id)))
}
}
// funcs
func (m *{{.PascalTable}}) log() *log.Entry {
@@ -64,7 +68,7 @@ func (m *{{.PascalTable}}) BatchCreate(ctx context.Context, models []*{{.PascalT
{{- if .SoftDelete }}
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(TimestampT(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 {
@@ -82,7 +86,7 @@ func (m *{{.PascalTable}}) BatchDelete(ctx context.Context, ids []int64) error {
return Int64(id)
})
stmt := table.{{.PascalTable}}.UPDATE().SET(table.{{.PascalTable}}.DeletedAt.Set(TimestampzT(time.Now()))).WHERE(table.{{.PascalTable}}.ID.IN(condIds...))
stmt := table.{{.PascalTable}}.UPDATE().SET(table.{{.PascalTable}}.DeletedAt.SET(TimestampT(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 {
@@ -142,35 +146,32 @@ func (m *{{.PascalTable}}) Update(ctx context.Context) error {
m.log().WithField("func", "Update").Infof("{{.PascalTable}} item updated successfully")
return nil
}
// GetByID
func (m *{{.PascalTable}}) GetByID(ctx context.Context, id int64, conds ...BoolExpression) (*{{.PascalTable}}, error) {
expr := table.{{.PascalTable}}.ID.EQ(Int(m.ID))
if len(conds) > 0 {
for _, c := range conds {
expr = expr.AND(c)
}
}
stmt := table.{{.PascalTable}}.SELECT(table.{{.PascalTable}}.AllColumns).WHERE(expr)
m.log().WithField("func", "GetByID").Info(stmt.DebugSql())
// GetByCond
func (m *{{.PascalTable}}) GetByCond(ctx context.Context, conds ...Cond) (*{{.PascalTable}}, error) {
cond := CondTrue(conds...)
stmt := table.{{.PascalTable}}.SELECT(table.{{.PascalTable}}.AllColumns).WHERE(cond)
m.log().WithField("func", "GetByCond").Info(stmt.DebugSql())
if err := stmt.QueryContext(ctx, db, m); err != nil {
m.log().WithField("func","GetByID").Errorf("error getting {{.PascalTable}} item by ID: %v", err)
m.log().WithField("func","GetByCond").Errorf("error getting {{.PascalTable}} item by ID: %v", err)
return nil, err
}
m.log().WithField("func", "GetByID").Infof("{{.PascalTable}} item retrieved successfully")
m.log().WithField("func", "GetByCond").Infof("{{.PascalTable}} item retrieved successfully")
return m, nil
}
// GetByID
func (m *{{.PascalTable}}) GetByID(ctx context.Context, id int64, conds ...Cond) (*{{.PascalTable}}, error) {
return m.GetByCond(ctx, CondJoin(m.CondID(id), conds...)...)
}
// Count
func (m *{{.PascalTable}}) Count(ctx context.Context, conds ...BoolExpression) (int64, error) {
cond := Bool(true)
if len(conds) > 0 {
for _, c := range conds {
cond = cond.AND(c)
}
}
func (m *{{.PascalTable}}) Count(ctx context.Context, conds ...Cond) (int64, error) {
cond := CondTrue(conds...)
tbl := table.{{.PascalTable}}
stmt := tbl.SELECT(COUNT(tbl.ID).AS("count")).WHERE(cond)