后端重构
This commit is contained in:
Yvan
@@ -0,0 +1,322 @@
|
||||
{{- $db := "r.data.db" }}
|
||||
|
||||
{{- if .IsAdd}}
|
||||
|
||||
// Get{{.StructName}}List 新增搜索语句 (GORM Gen 风格)
|
||||
// 在 List 方法中添加以下条件:
|
||||
{{ GenerateGormGenSearchConditions .Fields .StructName }}
|
||||
|
||||
// Get{{.StructName}}List 新增排序语句
|
||||
// 在 List 方法中添加以下排序字段到 orderMap:
|
||||
{{- range .Fields}}
|
||||
{{- if .Sort}}
|
||||
// orderMap["{{.ColumnName}}"] = t.{{.FieldName}}
|
||||
{{- end}}
|
||||
{{- end}}
|
||||
|
||||
{{- if .HasDataSource }}
|
||||
// GetDataSource 方法新增关联语句
|
||||
{{range $key, $value := .DataSourceMap}}
|
||||
{{$key}} := make([]map[string]any, 0)
|
||||
{{ $dataDB := "r.data.db" }}
|
||||
{{$dataDB}}.Table("{{$value.Table}}"){{- if $value.HasDeletedAt}}.Where("deleted_at IS NULL"){{ end }}.Select("{{$value.Label}} as label,{{$value.Value}} as value").Scan(&{{$key}})
|
||||
res["{{$key}}"] = {{$key}}
|
||||
{{- end }}
|
||||
{{- end }}
|
||||
|
||||
{{- else}}
|
||||
package {{.Package}}
|
||||
|
||||
import (
|
||||
{{- if not .OnlyTemplate }}
|
||||
"context"
|
||||
"{{.Module}}/internal/biz/{{.Package}}"
|
||||
"{{.Module}}/internal/data/model"
|
||||
"{{.Module}}/internal/data/query"
|
||||
{{- if not .IsTree}}
|
||||
{{.Package}}Req "{{.Module}}/internal/service/types/{{.Package}}/request"
|
||||
{{- else }}
|
||||
"{{.Module}}/pkg/utils"
|
||||
{{- end }}
|
||||
{{- if .AutoCreateResource }}
|
||||
"gorm.io/gorm"
|
||||
{{- end}}
|
||||
{{- if .NeedSort}}
|
||||
"gorm.io/gen/field"
|
||||
{{- end}}
|
||||
"github.com/go-kratos/kratos/v2/log"
|
||||
{{- end }}
|
||||
)
|
||||
|
||||
type {{.Abbreviation}}Repo struct {
|
||||
data *Data
|
||||
log *log.Helper
|
||||
}
|
||||
|
||||
|
||||
// New{{.StructName}}Repo 创建{{.Description}}仓储
|
||||
// 实现 {{.Package}}.{{.StructName}}Repo 接口
|
||||
func New{{.StructName}}Repo(data *Data, logger log.Logger) {{.Package}}.{{.StructName}}Repo {
|
||||
return &{{.Abbreviation}}Repo{
|
||||
data: data,
|
||||
log: log.NewHelper(logger),
|
||||
}
|
||||
}
|
||||
|
||||
// Create 创建{{.Description}}
|
||||
func (r *{{.Abbreviation}}Repo) Create(ctx context.Context, {{.Abbreviation}} *{{.Package}}.{{.StructName}}) error {
|
||||
m := toModel{{.StructName}}({{.Abbreviation}})
|
||||
if err := query.{{.StructName}}.WithContext(ctx).Create(m); err != nil {
|
||||
return err
|
||||
}
|
||||
// 回写生成的ID
|
||||
{{.Abbreviation}}.{{.PrimaryField.FieldName}} = {{- if eq .PrimaryField.FieldType "uint"}} uint(m.{{.PrimaryField.FieldName}}){{- else if eq .PrimaryField.FieldType "int"}} int(m.{{.PrimaryField.FieldName}}){{- else if eq .PrimaryField.FieldType "int64"}} m.{{.PrimaryField.FieldName}}{{- else}} m.{{.PrimaryField.FieldName}}{{- end}}
|
||||
return nil
|
||||
}
|
||||
|
||||
// Delete 删除{{.Description}}
|
||||
func (r *{{.Abbreviation}}Repo) Delete(ctx context.Context, {{.PrimaryField.FieldJson}} {{.PrimaryField.FieldType}}{{- if .AutoCreateResource -}}, userID uint{{- end -}}) error {
|
||||
t := query.{{.StructName}}
|
||||
{{- if .AutoCreateResource }}
|
||||
return {{$db}}.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
|
||||
q := query.Use(tx).{{.StructName}}
|
||||
if _, err := q.WithContext(ctx).Where(t.{{.PrimaryField.FieldName}}.Eq({{- if eq .PrimaryField.FieldType "uint"}}int64({{.PrimaryField.FieldJson}}){{- else if eq .PrimaryField.FieldType "int"}}int64({{.PrimaryField.FieldJson}}){{- else}}{{.PrimaryField.FieldJson}}{{- end}})).Update(t.DeletedBy, userID); err != nil {
|
||||
return err
|
||||
}
|
||||
if _, err := q.WithContext(ctx).Where(t.{{.PrimaryField.FieldName}}.Eq({{- if eq .PrimaryField.FieldType "uint"}}int64({{.PrimaryField.FieldJson}}){{- else if eq .PrimaryField.FieldType "int"}}int64({{.PrimaryField.FieldJson}}){{- else}}{{.PrimaryField.FieldJson}}{{- end}})).Delete(); err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
})
|
||||
{{- else }}
|
||||
_, err := t.WithContext(ctx).Where(t.{{.PrimaryField.FieldName}}.Eq({{- if eq .PrimaryField.FieldType "uint"}}int64({{.PrimaryField.FieldJson}}){{- else if eq .PrimaryField.FieldType "int"}}int64({{.PrimaryField.FieldJson}}){{- else}}{{.PrimaryField.FieldJson}}{{- end}})).Delete()
|
||||
return err
|
||||
{{- end }}
|
||||
}
|
||||
|
||||
|
||||
// DeleteByIds 批量删除{{.Description}}
|
||||
func (r *{{.Abbreviation}}Repo) DeleteByIds(ctx context.Context, {{.PrimaryField.FieldJson}}s []{{.PrimaryField.FieldType}}{{- if .AutoCreateResource -}}, deletedBy uint{{- end -}}) error {
|
||||
t := query.{{.StructName}}
|
||||
// 转换 ID 类型
|
||||
int64Ids := make([]int64, len({{.PrimaryField.FieldJson}}s))
|
||||
for i, id := range {{.PrimaryField.FieldJson}}s {
|
||||
int64Ids[i] = int64(id)
|
||||
}
|
||||
{{- if .AutoCreateResource }}
|
||||
return {{$db}}.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
|
||||
q := query.Use(tx).{{.StructName}}
|
||||
if _, err := q.WithContext(ctx).Where(t.{{.PrimaryField.FieldName}}.In(int64Ids...)).Update(t.DeletedBy, deletedBy); err != nil {
|
||||
return err
|
||||
}
|
||||
if _, err := q.WithContext(ctx).Where(t.{{.PrimaryField.FieldName}}.In(int64Ids...)).Delete(); err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
})
|
||||
{{- else}}
|
||||
_, err := t.WithContext(ctx).Where(t.{{.PrimaryField.FieldName}}.In(int64Ids...)).Delete()
|
||||
return err
|
||||
{{- end}}
|
||||
}
|
||||
|
||||
{{- if .AutoCreateResource }}
|
||||
// DeleteWithTx 在事务中删除{{.Description}}（用于 Service 层事务管理）
|
||||
// @param ctx 上下文
|
||||
// @param tx 事务对象（由 Service 层传入）
|
||||
// @param {{.PrimaryField.FieldJson}} 主键ID
|
||||
// @return error 删除失败时返回错误
|
||||
func (r *{{.Abbreviation}}Repo) DeleteWithTx(ctx context.Context, tx *gorm.DB, {{.PrimaryField.FieldJson}} {{.PrimaryField.FieldType}}) error {
|
||||
t := query.{{.StructName}}
|
||||
q := query.Use(tx).{{.StructName}}
|
||||
_, err := q.WithContext(ctx).Where(t.{{.PrimaryField.FieldName}}.Eq({{- if eq .PrimaryField.FieldType "uint"}}int64({{.PrimaryField.FieldJson}}){{- else if eq .PrimaryField.FieldType "int"}}int64({{.PrimaryField.FieldJson}}){{- else}}{{.PrimaryField.FieldJson}}{{- end}})).Delete()
|
||||
return err
|
||||
}
|
||||
|
||||
// DeleteByIdsWithTx 在事务中批量删除{{.Description}}（用于 Service 层事务管理）
|
||||
// @param ctx 上下文
|
||||
// @param tx 事务对象（由 Service 层传入）
|
||||
// @param {{.PrimaryField.FieldJson}}s 主键ID列表
|
||||
// @return error 删除失败时返回错误
|
||||
func (r *{{.Abbreviation}}Repo) DeleteByIdsWithTx(ctx context.Context, tx *gorm.DB, {{.PrimaryField.FieldJson}}s []{{.PrimaryField.FieldType}}) error {
|
||||
t := query.{{.StructName}}
|
||||
// 转换 ID 类型
|
||||
int64Ids := make([]int64, len({{.PrimaryField.FieldJson}}s))
|
||||
for i, id := range {{.PrimaryField.FieldJson}}s {
|
||||
int64Ids[i] = int64(id)
|
||||
}
|
||||
q := query.Use(tx).{{.StructName}}
|
||||
_, err := q.WithContext(ctx).Where(t.{{.PrimaryField.FieldName}}.In(int64Ids...)).Delete()
|
||||
return err
|
||||
}
|
||||
{{- end }}
|
||||
|
||||
// Update 更新{{.Description}}
|
||||
func (r *{{.Abbreviation}}Repo) Update(ctx context.Context, {{.Abbreviation}} *{{.Package}}.{{.StructName}}) error {
|
||||
t := query.{{.StructName}}
|
||||
m := toModel{{.StructName}}({{.Abbreviation}})
|
||||
_, err := t.WithContext(ctx).Where(t.{{.PrimaryField.FieldName}}.Eq(m.{{.PrimaryField.FieldName}})).Updates(m)
|
||||
return err
|
||||
}
|
||||
|
||||
// FindByID 根据ID获取{{.Description}}
|
||||
func (r *{{.Abbreviation}}Repo) FindByID(ctx context.Context, {{.PrimaryField.FieldJson}} {{.PrimaryField.FieldType}}) (*{{.Package}}.{{.StructName}}, error) {
|
||||
t := query.{{.StructName}}
|
||||
m, err := t.WithContext(ctx).Where(t.{{.PrimaryField.FieldName}}.Eq({{- if eq .PrimaryField.FieldType "uint"}}int64({{.PrimaryField.FieldJson}}){{- else if eq .PrimaryField.FieldType "int"}}int64({{.PrimaryField.FieldJson}}){{- else}}{{.PrimaryField.FieldJson}}{{- end}})).First()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return toBiz{{.StructName}}(m), nil
|
||||
}
|
||||
|
||||
|
||||
{{- if .IsTree }}
|
||||
// List 获取{{.Description}}列表（树形）
|
||||
func (r *{{.Abbreviation}}Repo) List(ctx context.Context) ([]*{{.Package}}.{{.StructName}}, error) {
|
||||
t := query.{{.StructName}}
|
||||
models, err := t.WithContext(ctx).Find()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
// 转换为 biz 类型
|
||||
result := make([]*{{.Package}}.{{.StructName}}, len(models))
|
||||
for i, m := range models {
|
||||
result[i] = toBiz{{.StructName}}(m)
|
||||
}
|
||||
return utils.BuildTree(result), nil
|
||||
}
|
||||
|
||||
// HasChildren 检查是否有子节点
|
||||
func (r *{{.Abbreviation}}Repo) HasChildren(ctx context.Context, {{.PrimaryField.FieldJson}} {{.PrimaryField.FieldType}}) (bool, error) {
|
||||
t := query.{{.StructName}}
|
||||
count, err := t.WithContext(ctx).Where(t.ParentID.Eq({{- if eq .PrimaryField.FieldType "uint"}}int64({{.PrimaryField.FieldJson}}){{- else if eq .PrimaryField.FieldType "int"}}int64({{.PrimaryField.FieldJson}}){{- else}}{{.PrimaryField.FieldJson}}{{- end}})).Count()
|
||||
return count > 0, err
|
||||
}
|
||||
{{- else }}
|
||||
// List 分页获取{{.Description}}列表
|
||||
func (r *{{.Abbreviation}}Repo) List(ctx context.Context, req *{{.Package}}Req.{{.StructName}}Search) ([]*{{.Package}}.{{.StructName}}, int64, error) {
|
||||
t := query.{{.StructName}}
|
||||
q := t.WithContext(ctx)
|
||||
|
||||
// 条件搜索 (使用 GORM Gen 类型安全查询)
|
||||
{{- if .GvaModel }}
|
||||
if len(req.CreatedAtRange) == 2 {
|
||||
q = q.Where(t.CreatedAt.Between(req.CreatedAtRange[0], req.CreatedAtRange[1]))
|
||||
}
|
||||
{{- end }}
|
||||
{{ GenerateGormGenSearchConditions .Fields .StructName }}
|
||||
|
||||
// 统计总数
|
||||
total, err := q.Count()
|
||||
if err != nil {
|
||||
return nil, 0, err
|
||||
}
|
||||
|
||||
{{- if .NeedSort}}
|
||||
// 排序 (使用 GORM Gen 类型安全排序)
|
||||
orderMap := make(map[string]func() field.OrderExpr)
|
||||
{{- if .GvaModel }}
|
||||
orderMap["id"] = func() field.OrderExpr { return t.ID }
|
||||
orderMap["created_at"] = func() field.OrderExpr { return t.CreatedAt }
|
||||
{{- end }}
|
||||
{{- range .Fields}}
|
||||
{{- if .Sort}}
|
||||
orderMap["{{.ColumnName}}"] = func() field.OrderExpr { return t.{{.FieldName}} }
|
||||
{{- end}}
|
||||
{{- end}}
|
||||
if orderFn, ok := orderMap[req.Sort]; ok {
|
||||
if req.Order == "descending" {
|
||||
q = q.Order(orderFn().Desc())
|
||||
} else {
|
||||
q = q.Order(orderFn())
|
||||
}
|
||||
}
|
||||
{{- end}}
|
||||
|
||||
// 分页
|
||||
limit := req.PageSize
|
||||
offset := req.PageSize * (req.Page - 1)
|
||||
if limit != 0 {
|
||||
q = q.Limit(limit).Offset(offset)
|
||||
}
|
||||
|
||||
models, err := q.Find()
|
||||
if err != nil {
|
||||
return nil, 0, err
|
||||
}
|
||||
|
||||
// 转换为 biz 类型
|
||||
result := make([]*{{.Package}}.{{.StructName}}, len(models))
|
||||
for i, m := range models {
|
||||
result[i] = toBiz{{.StructName}}(m)
|
||||
}
|
||||
|
||||
return result, total, nil
|
||||
}
|
||||
{{- end }}
|
||||
|
||||
|
||||
{{- if .HasDataSource }}
|
||||
// GetDataSource 获取{{.Description}}数据源
|
||||
func (r *{{.Abbreviation}}Repo) GetDataSource(ctx context.Context) (map[string][]map[string]any, error) {
|
||||
res := make(map[string][]map[string]any)
|
||||
{{range $key, $value := .DataSourceMap}}
|
||||
{{$key}} := make([]map[string]any, 0)
|
||||
if err := r.data.db.WithContext(ctx).Table("{{$value.Table}}"){{- if $value.HasDeletedAt}}.Where("deleted_at IS NULL"){{ end }}.Select("{{$value.Label}} as label, {{$value.Value}} as value").Scan(&{{$key}}).Error; err != nil {
|
||||
return nil, err
|
||||
}
|
||||
res["{{$key}}"] = {{$key}}
|
||||
{{- end }}
|
||||
return res, nil
|
||||
}
|
||||
{{- end }}
|
||||
|
||||
// toModel{{.StructName}} 将 biz 层对象转换为 model 层对象
|
||||
func toModel{{.StructName}}(b *{{.Package}}.{{.StructName}}) *model.{{.StructName}} {
|
||||
if b == nil {
|
||||
return nil
|
||||
}
|
||||
return &model.{{.StructName}}{
|
||||
{{- if .GvaModel }}
|
||||
GVA_MODEL: model.GVA_MODEL{
|
||||
ID: int64(b.ID),
|
||||
CreatedAt: b.CreatedAt,
|
||||
UpdatedAt: b.UpdatedAt,
|
||||
},
|
||||
{{- else }}
|
||||
{{.PrimaryField.FieldName}}: {{- if eq .PrimaryField.FieldType "uint"}} int64(b.{{.PrimaryField.FieldName}}){{- else if eq .PrimaryField.FieldType "int"}} int64(b.{{.PrimaryField.FieldName}}){{- else}} b.{{.PrimaryField.FieldName}}{{- end}},
|
||||
{{- end }}
|
||||
{{- range .Fields}}
|
||||
{{- if not .PrimaryKey}}
|
||||
{{.FieldName}}: b.{{.FieldName}},
|
||||
{{- end}}
|
||||
{{- end}}
|
||||
}
|
||||
}
|
||||
|
||||
// toBiz{{.StructName}} 将 model 层对象转换为 biz 层对象
|
||||
func toBiz{{.StructName}}(m *model.{{.StructName}}) *{{.Package}}.{{.StructName}} {
|
||||
if m == nil {
|
||||
return nil
|
||||
}
|
||||
return &{{.Package}}.{{.StructName}}{
|
||||
{{- if .GvaModel }}
|
||||
ID: uint(m.ID),
|
||||
CreatedAt: m.CreatedAt,
|
||||
UpdatedAt: m.UpdatedAt,
|
||||
{{- else }}
|
||||
{{.PrimaryField.FieldName}}: {{- if eq .PrimaryField.FieldType "uint"}} uint(m.{{.PrimaryField.FieldName}}){{- else if eq .PrimaryField.FieldType "int"}} int(m.{{.PrimaryField.FieldName}}){{- else}} m.{{.PrimaryField.FieldName}}{{- end}},
|
||||
{{- end }}
|
||||
{{- range .Fields}}
|
||||
{{- if not .PrimaryKey}}
|
||||
{{.FieldName}}: m.{{.FieldName}},
|
||||
{{- end}}
|
||||
{{- end}}
|
||||
}
|
||||
}
|
||||
|
||||
{{- if .HasCustomMethods }}
|
||||
{{ GenerateCustomMethodImpl .CustomMethods .StructName .Package .Abbreviation }}
|
||||
{{- end }}
|
||||
|
||||
{{- end}}
|
||||
@@ -0,0 +1,14 @@
|
||||
{{- if .IsAdd }}
|
||||
// 请在 {{.Package}} 包的 enter.go 文件中的 ProviderSet 添加以下内容:
|
||||
// New{{.StructName}}Repo,
|
||||
{{- else }}
|
||||
package {{.Package}}
|
||||
|
||||
import "github.com/google/wire"
|
||||
|
||||
// ProviderSet is {{.Package}} data providers.
|
||||
// 包含 {{.Description}} 相关的数据访问层实现
|
||||
var ProviderSet = wire.NewSet(
|
||||
New{{.StructName}}Repo,
|
||||
)
|
||||
{{- end }}
|
||||
@@ -0,0 +1,86 @@
|
||||
{{- if .IsAdd}}
|
||||
// 在 model.{{.StructName}} 结构体中新增如下字段
|
||||
{{- range .Fields}}
|
||||
{{ GenerateModelField . }}
|
||||
{{- end }}
|
||||
|
||||
{{ else }}
|
||||
// Code generated by KRA AutoCode. DO NOT EDIT.
|
||||
// 自动生成模板 {{.StructName}}
|
||||
package model
|
||||
|
||||
import (
|
||||
{{- if .GvaModel }}
|
||||
"time"
|
||||
|
||||
"gorm.io/gorm"
|
||||
{{- else }}
|
||||
{{- if or .HasTimer }}
|
||||
"time"
|
||||
{{- end }}
|
||||
{{- if .NeedJSON }}
|
||||
"gorm.io/datatypes"
|
||||
{{- end }}
|
||||
{{- end }}
|
||||
)
|
||||
|
||||
const TableName{{.StructName}} = "{{.TableName}}"
|
||||
|
||||
// {{.StructName}} {{.Description}}
|
||||
// mapped from table <{{.TableName}}>
|
||||
type {{.StructName}} struct {
|
||||
{{- if .GvaModel }}
|
||||
ID int64 `gorm:"column:id;type:bigint(20) unsigned;primaryKey;autoIncrement:true" json:"id"`
|
||||
CreatedAt time.Time `gorm:"column:created_at;type:datetime(3)" json:"created_at"`
|
||||
UpdatedAt time.Time `gorm:"column:updated_at;type:datetime(3)" json:"updated_at"`
|
||||
DeletedAt gorm.DeletedAt `gorm:"column:deleted_at;type:datetime(3);index:idx_{{.TableName}}_deleted_at,priority:1" json:"deleted_at"`
|
||||
{{- else }}
|
||||
{{- with .PrimaryField}}
|
||||
{{.FieldName}} int64 `gorm:"column:{{.ColumnName}};type:bigint(20) unsigned;primaryKey;autoIncrement:true" json:"{{.FieldJson}}"` // {{.FieldDesc}}
|
||||
{{- end}}
|
||||
{{- end }}
|
||||
{{- range .Fields}}
|
||||
{{- if not .PrimaryKey}}
|
||||
{{ GenerateModelField . }}
|
||||
{{- end}}
|
||||
{{- end }}
|
||||
{{- if .AutoCreateResource }}
|
||||
CreatedBy int64 `gorm:"column:created_by;type:bigint(20) unsigned;comment:创建者" json:"created_by"` // 创建者
|
||||
UpdatedBy int64 `gorm:"column:updated_by;type:bigint(20) unsigned;comment:更新者" json:"updated_by"` // 更新者
|
||||
DeletedBy int64 `gorm:"column:deleted_by;type:bigint(20) unsigned;comment:删除者" json:"deleted_by"` // 删除者
|
||||
{{- end }}
|
||||
{{- if .IsTree }}
|
||||
ParentID int64 `gorm:"column:parent_id;type:bigint(20);comment:父节点ID" json:"parent_id"` // 父节点ID
|
||||
Children []*{{.StructName}} `gorm:"-" json:"children"` // 子节点
|
||||
{{- end }}
|
||||
}
|
||||
|
||||
// TableName {{.StructName}}'s table name
|
||||
func (*{{.StructName}}) TableName() string {
|
||||
return TableName{{.StructName}}
|
||||
}
|
||||
|
||||
{{- if .IsTree }}
|
||||
|
||||
// GetChildren 实现 TreeNode 接口
|
||||
func (s *{{.StructName}}) GetChildren() []*{{.StructName}} {
|
||||
return s.Children
|
||||
}
|
||||
|
||||
// SetChildren 实现 TreeNode 接口
|
||||
func (s *{{.StructName}}) SetChildren(child *{{.StructName}}) {
|
||||
s.Children = append(s.Children, child)
|
||||
}
|
||||
|
||||
// GetID 实现 TreeNode 接口
|
||||
func (s *{{.StructName}}) GetID() int {
|
||||
return int(s.{{- if .GvaModel}}ID{{- else}}{{.PrimaryField.FieldName}}{{- end}})
|
||||
}
|
||||
|
||||
// GetParentID 实现 TreeNode 接口
|
||||
func (s *{{.StructName}}) GetParentID() int {
|
||||
return int(s.ParentID)
|
||||
}
|
||||
{{- end }}
|
||||
|
||||
{{ end }}
|
||||
Reference in New Issue
Block a user