You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
87 lines
3.4 KiB
87 lines
3.4 KiB
package model |
|
{{if .Imports}} |
|
import ( |
|
{{range $val:= .Imports}} |
|
"{{$val}}"{{end}} |
|
) |
|
{{end}} |
|
type ( |
|
{{.Name|LowerType}}Model interface { |
|
Insert(ctx context.Context, data *{{.Name|UpperType}}) error |
|
{{ if .Primary}}Update(ctx context.Context, {{.Primary.Name|LowerType}} {{.Primary.GoType}}, updates map[string]any) error |
|
Delete(ctx context.Context, {{.Primary.Name|LowerType}} {{.Primary.GoType}}) error |
|
FindOne(ctx context.Context, {{.Primary.Name|LowerType}} {{.Primary.GoType}}) (*{{.Name|UpperType}}, error){{end}} |
|
{{range $val:= .Indexes}}Find{{$val.Name|UpperType}}(ctx context.Context {{range $column:=$val.Columns}},{{$column.Name|LowerType}} {{$column.GoType}}{{end}}) (*{{$.Name|UpperType}}, error) |
|
{{end}} |
|
} |
|
{{.Name|UpperType}} struct { |
|
updates map[string]any |
|
{{range $val := .Columns}}{{$val.Name|UpperType}} {{$val.GoType}} `gorm:"column:{{$val.Name}}"`{{if $val.Comment}}//{{$val.Comment}}{{end}} |
|
{{end}} |
|
} |
|
default{{.Name|UpperType}}Model struct { |
|
db *gorm.DB |
|
} |
|
) |
|
func (s *{{.Name|UpperType}}) TableName() string { |
|
return "{{.Name}}" |
|
} |
|
|
|
func New{{.Name|UpperType}}() *{{.Name|UpperType}} { |
|
return &{{.Name|UpperType}}{ |
|
updates: make(map[string]any), |
|
{{range $val := .Columns}}{{if $val.Default}}{{$val.Name|UpperType}}:{{if (eq $val.GoType "string")}}"{{$val.Default}}"{{else}}{{$val.Default}}{{end}}, |
|
{{end}}{{end}} |
|
} |
|
} |
|
{{range $val := .Columns}} |
|
func (s *{{$.Name|UpperType}}) Set{{$val.Name|UpperType}}({{$val.Name|LowerType}} {{$val.GoType}}) *{{$.Name|UpperType}}{ |
|
s.{{$val.Name|UpperType}} = {{$val.Name|LowerType}} |
|
s.set("{{$val.Name}}", {{$val.Name|LowerType}}) |
|
return s |
|
} |
|
{{end}} |
|
func (s *{{.Name|UpperType}}) set(key string, val any) { |
|
s.updates[key] = val |
|
} |
|
|
|
func (s *{{.Name|UpperType}}) UpdateColumn() map[string]any { |
|
return s.updates |
|
} |
|
|
|
|
|
|
|
func new{{.Name|UpperType}}DAO(db *gorm.DB) *default{{.Name|UpperType}}Model { |
|
return &default{{.Name|UpperType}}Model{ |
|
db: db, |
|
} |
|
} |
|
|
|
func (s *default{{.Name|UpperType}}Model) Insert(ctx context.Context, data *{{.Name|UpperType}}) error { |
|
return s.db.Create(data).Error |
|
} |
|
{{ if .Primary}} |
|
func (s *default{{.Name|UpperType}}Model) Update(ctx context.Context, {{.Primary.Name|LowerType}} {{.Primary.GoType}}, updates map[string]any) error { |
|
return s.db.Model({{.Name|UpperType}}{}).Where("{{.Primary.Name}}=?", {{.Primary.Name|LowerType}}).Updates(updates).Error |
|
} |
|
|
|
|
|
func (s *default{{.Name|UpperType}}Model) Delete(ctx context.Context, {{.Primary.Name|LowerType}} {{.Primary.GoType}}) error { |
|
return s.db.Where("{{.Primary.Name}}=?", {{.Primary.Name|LowerType}}).Delete(&{{.Name|UpperType}}{}).Error |
|
} |
|
|
|
|
|
func (s *default{{.Name|UpperType}}Model) FindOne(ctx context.Context, {{.Primary.Name|LowerType}} {{.Primary.GoType}}) (*{{.Name|UpperType}}, error) { |
|
row := {{.Name|UpperType}}{} |
|
err := s.db.Where("{{.Primary.Name}}=?", {{.Primary.Name|LowerType}}).Find(&row).Error |
|
return findResultWithError(&row,err) |
|
} |
|
{{end}} |
|
|
|
{{range $val:= .Indexes}} |
|
func (s *default{{$.Name|UpperType}}Model) Find{{$val.Name|UpperType}}(ctx context.Context {{range $column:=$val.Columns}},{{$column.Name|LowerType}} {{$column.GoType}}{{end}}) (*{{$.Name|UpperType}}, error) { |
|
row := {{$.Name|UpperType}}{} |
|
err := s.db.Where("{{$val.Columns|UpdateColumn}}" {{range $column:=$val.Columns}},{{$column.Name|LowerType}}{{end}}).Find(&row).Error |
|
return findResultWithError(&row,err) |
|
} |
|
{{end}}
|
|
|