Skip to content

Commit fd09b48

Browse files
chore(template): update .template go-zero/model
1 parent 67ee884 commit fd09b48

8 files changed

Lines changed: 20 additions & 16 deletions

File tree

.template/go-zero/model/delete.tpl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
func (m *default{{.upperStartCamelObject}}Model) Delete(ctx context.Context, session sqlx.Session, {{.lowerStartCamelPrimaryKey}} {{.dataType}}) error {
22
sb := sqlbuilder.DeleteFrom(m.table)
3-
sb.Where(sb.EQ("{{.originalPrimaryKey}}", {{.lowerStartCamelPrimaryKey}}))
3+
sb.Where(sb.EQ(condition.Field("{{.originalPrimaryKey}}"), {{.lowerStartCamelPrimaryKey}}))
44
statement, args := sb.Build()
55
var err error
66
if session != nil {
@@ -20,7 +20,7 @@ func (m *default{{.upperStartCamelObject}}Model) DeleteWithCache(ctx context.Con
2020
{{end}} {{.keys}}
2121
_, err {{if .containsIndexCache}}={{else}}:={{end}} m.cachedConn.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) {
2222
sb := sqlbuilder.DeleteFrom(m.table)
23-
sb.Where(sb.EQ("{{.originalPrimaryKey}}", {{.lowerStartCamelPrimaryKey}}))
23+
sb.Where(sb.EQ(condition.Field("{{.originalPrimaryKey}}"), {{.lowerStartCamelPrimaryKey}}))
2424
statement, args := sb.Build()
2525
if session != nil {
2626
return session.ExecCtx(ctx, statement, args...)

.template/go-zero/model/find-one-by-field-extra-method.tpl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ func (m *default{{.upperStartCamelObject}}Model) formatPrimary(primary any) stri
44

55
func (m *default{{.upperStartCamelObject}}Model) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary any) error {
66
sb := sqlbuilder.Select({{.lowerStartCamelObject}}Rows).From(m.table)
7-
sb.Where(sb.EQ("{{.originalPrimaryField}}", primary))
7+
sb.Where(sb.EQ(condition.Field("{{.originalPrimaryField}}"), primary))
88
sql, args := sb.Build()
99
return conn.QueryRowCtx(ctx, v, sql, args...)
1010
}

.template/go-zero/model/find-one.tpl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
func (m *default{{.upperStartCamelObject}}Model) FindOne(ctx context.Context, session sqlx.Session, {{.lowerStartCamelPrimaryKey}} {{.dataType}}) (*{{.upperStartCamelObject}}, error) {
22
sb := sqlbuilder.Select({{.lowerStartCamelObject}}Rows).From(m.table)
3-
sb.Where(sb.EQ("{{.originalPrimaryKey}}", {{.lowerStartCamelPrimaryKey}}))
3+
sb.Where(sb.EQ(condition.Field("{{.originalPrimaryKey}}"), {{.lowerStartCamelPrimaryKey}}))
44
sb.Limit(1)
55
sql, args := sb.Build()
66
var resp {{.upperStartCamelObject}}
@@ -25,7 +25,7 @@ func (m *default{{.upperStartCamelObject}}Model) FindOneWithCache(ctx context.Co
2525
var resp {{.upperStartCamelObject}}
2626
err := m.cachedConn.QueryRowCtx(ctx, &resp, {{.cacheKeyVariable}}, func(ctx context.Context, conn sqlx.SqlConn, v any) error {
2727
sb := sqlbuilder.Select({{.lowerStartCamelObject}}Rows).From(m.table)
28-
sb.Where(sb.EQ("{{.originalPrimaryKey}}", {{.lowerStartCamelPrimaryKey}}))
28+
sb.Where(sb.EQ(condition.Field("{{.originalPrimaryKey}}"), {{.lowerStartCamelPrimaryKey}}))
2929
sql, args := sb.Build()
3030
if session != nil {
3131
return session.QueryRowCtx(ctx, v, sql, args...)

.template/go-zero/model/import-no-cache.tpl

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,10 +4,8 @@ import (
44
"strings"
55
{{if .time}}"time"{{end}}
66

7-
"github.com/zeromicro/go-zero/core/stores/builder"
87
"github.com/zeromicro/go-zero/core/stores/sqlc"
98
"github.com/zeromicro/go-zero/core/stores/sqlx"
10-
"github.com/zeromicro/go-zero/core/stringx"
119
"github.com/huandu/go-sqlbuilder"
1210
"github.com/jzero-io/jzero-contrib/condition"
1311
"github.com/jzero-io/jzero-contrib/modelx"

.template/go-zero/model/import.tpl

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,10 +5,8 @@ import (
55
"strings"
66
{{if .time}}"time"{{end}}
77

8-
"github.com/zeromicro/go-zero/core/stores/builder"
98
"github.com/zeromicro/go-zero/core/stores/sqlc"
109
"github.com/zeromicro/go-zero/core/stores/sqlx"
11-
"github.com/zeromicro/go-zero/core/stringx"
1210
"github.com/huandu/go-sqlbuilder"
1311
"github.com/jzero-io/jzero-contrib/condition"
1412
"github.com/jzero-io/jzero-contrib/modelx"

.template/go-zero/model/model-new.tpl

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,9 +7,12 @@ func new{{.upperStartCamelObject}}Model(conn sqlx.SqlConn, op ...opts.Opt[modelx
77
if o.CachedConn != nil {
88
cachedConn = *o.CachedConn
99
}
10+
11+
initVars()
12+
1013
return &default{{.upperStartCamelObject}}Model{
1114
cachedConn: cachedConn,
1215
conn: conn,
13-
table: {{.table}},
16+
table: condition.Table({{.table}}),
1417
}
1518
}

.template/go-zero/model/update.tpl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ func (m *default{{.upperStartCamelObject}}Model) Update(ctx context.Context, ses
66
assigns = append(assigns, sb.Assign(s, nil))
77
}
88
sb.Set(assigns...)
9-
sb.Where(sb.EQ("{{.originalPrimaryKey}}", nil))
9+
sb.Where(sb.EQ(condition.Field("{{.originalPrimaryKey}}"), nil))
1010
statement, _ := sb.Build()
1111

1212
var err error
@@ -32,7 +32,7 @@ func (m *default{{.upperStartCamelObject}}Model) UpdateWithCache(ctx context.Con
3232
assigns = append(assigns, sb.Assign(s, nil))
3333
}
3434
sb.Set(assigns...)
35-
sb.Where(sb.EQ("{{.originalPrimaryKey}}", nil))
35+
sb.Where(sb.EQ(condition.Field("{{.originalPrimaryKey}}"), nil))
3636
statement, _ := sb.Build()
3737
if session != nil{
3838
return session.ExecCtx(ctx, statement, {{.expressionValues}})

.template/go-zero/model/var.tpl

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,13 @@
11
var (
2-
{{.lowerStartCamelObject}}FieldNames = builder.RawFieldNames(&{{.upperStartCamelObject}}{}{{if .postgreSql}}, true{{end}})
3-
{{.lowerStartCamelObject}}Rows = strings.Join({{.lowerStartCamelObject}}FieldNames, ",")
4-
{{.lowerStartCamelObject}}RowsExpectAutoSet = {{if .postgreSql}}strings.Join(stringx.Remove({{.lowerStartCamelObject}}FieldNames, {{if .autoIncrement}}"{{.originalPrimaryKey}}", {{end}} {{.ignoreColumns}}), ","){{else}}strings.Join(stringx.Remove({{.lowerStartCamelObject}}FieldNames, {{if .autoIncrement}}"{{.originalPrimaryKey}}", {{end}} {{.ignoreColumns}}), ","){{end}}
5-
{{.lowerStartCamelObject}}RowsWithPlaceHolder = {{if .postgreSql}}builder.PostgreSqlJoin(stringx.Remove({{.lowerStartCamelObject}}FieldNames, "{{.originalPrimaryKey}}", {{.ignoreColumns}})){{else}}strings.Join(stringx.Remove({{.lowerStartCamelObject}}FieldNames, "{{.originalPrimaryKey}}", {{.ignoreColumns}}), "=?,") + "=?"{{end}}
2+
{{.lowerStartCamelObject}}FieldNames []string
3+
{{.lowerStartCamelObject}}Rows string
4+
{{.lowerStartCamelObject}}RowsExpectAutoSet string
65

76
{{if .withCache}}{{.cacheKeys}}{{end}}
87
)
8+
9+
func initVars() {
10+
{{.lowerStartCamelObject}}FieldNames = condition.RawFieldNames(&{{.upperStartCamelObject}}{})
11+
{{.lowerStartCamelObject}}Rows = strings.Join({{.lowerStartCamelObject}}FieldNames, ",")
12+
{{.lowerStartCamelObject}}RowsExpectAutoSet = strings.Join(condition.RemoveIgnoreColumns({{.lowerStartCamelObject}}FieldNames, {{if .autoIncrement}}"{{.originalPrimaryKey}}", {{end}} {{.ignoreColumns}}), ",")
13+
}

0 commit comments

Comments
 (0)