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.
78 lines
2.6 KiB
78 lines
2.6 KiB
1 year ago
|
package model
|
||
|
{{if .Imports}}
|
||
|
import (
|
||
|
{{range $val:= .Imports}}
|
||
|
"{{$val}}"{{end}}
|
||
|
)
|
||
|
{{end}}
|
||
|
type {{.Name|UpperType}} struct {
|
||
|
updates map[string]any
|
||
|
{{range $val := .Columns}}{{$val.Name|UpperType}} {{$val.GoType}} `db:"{{$val.Name}}"`{{if $val.Comment}}//{{$val.Comment}}{{end}}
|
||
|
{{end}}
|
||
|
}
|
||
|
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}}) {
|
||
|
s.{{$val.Name|UpperType}} = {{$val.Name|LowerType}}
|
||
|
s.set("{{$val.Name}}", {{$val.Name|LowerType}})
|
||
|
}
|
||
|
{{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
|
||
|
}
|
||
|
|
||
|
|
||
|
|
||
|
type default{{.Name|UpperType}}DAO struct {
|
||
|
db *gorm.DB
|
||
|
}
|
||
|
func New{{.Name|UpperType}}DAO(db *gorm.DB) *default{{.Name|UpperType}}DAO {
|
||
|
return &default{{.Name|UpperType}}DAO{
|
||
|
db: db,
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (s *default{{.Name|UpperType}}DAO) Insert(ctx context.Context, data *{{.Name|UpperType}}) error {
|
||
|
return s.db.Create(data).Error
|
||
|
}
|
||
|
{{ if .Primary}}
|
||
|
func (s *default{{.Name|UpperType}}DAO) 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}}DAO) 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}}DAO) 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
|
||
|
if err!=nil{
|
||
|
return nil,err
|
||
|
}
|
||
|
return &row, nil
|
||
|
}
|
||
|
{{end}}
|
||
|
|
||
|
{{range $val:= .Indexes}}
|
||
|
func (s *default{{$.Name|UpperType}}DAO) Find{{$val.Name|UpperType}}(ctx context.Context {{range $column:=$val.Columns}},{{$column.Name|LowerType}} {{$column.GoType}}{{end}}) error {
|
||
|
return s.db.Where("{{$val.Columns|UpdateColumn}}" {{range $column:=$val.Columns}},{{$column.Name|LowerType}}{{end}}).Delete(&{{$.Name|UpperType}}{}).Error
|
||
|
}
|
||
|
{{end}}
|