| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151 |
- package {{.Package}}
- import (
- "github.com/flipped-aurora/gin-vue-admin/server/global"
- "github.com/flipped-aurora/gin-vue-admin/server/model/{{.Package}}"
- {{.Package}}Req "github.com/flipped-aurora/gin-vue-admin/server/model/{{.Package}}/request"
- {{- if .AutoCreateResource }}
- "gorm.io/gorm"
- {{- end}}
- )
- type {{.StructName}}Service struct {
- }
- {{- $db := "" }}
- {{- if eq .BusinessDB "" }}
- {{- $db = "global.GVA_DB" }}
- {{- else}}
- {{- $db = printf "global.MustGetGlobalDBByDBName(\"%s\")" .BusinessDB }}
- {{- end}}
- // Create{{.StructName}} 创建{{.Description}}记录
- // Author [piexlmax](https://github.com/piexlmax)
- func ({{.Abbreviation}}Service *{{.StructName}}Service) Create{{.StructName}}({{.Abbreviation}} *{{.Package}}.{{.StructName}}) (err error) {
- err = {{$db}}.Create({{.Abbreviation}}).Error
- return err
- }
- // Delete{{.StructName}} 删除{{.Description}}记录
- // Author [piexlmax](https://github.com/piexlmax)
- func ({{.Abbreviation}}Service *{{.StructName}}Service)Delete{{.StructName}}({{.PrimaryField.FieldJson}} string{{- if .AutoCreateResource -}},userID uint{{- end -}}) (err error) {
- {{- if .AutoCreateResource }}
- err = {{$db}}.Transaction(func(tx *gorm.DB) error {
- if err := tx.Model(&{{.Package}}.{{.StructName}}{}).Where("{{.PrimaryField.ColumnName}} = ?", {{.PrimaryField.FieldJson}}).Update("deleted_by", userID).Error; err != nil {
- return err
- }
- if err = tx.Delete(&{{.Package}}.{{.StructName}}{},"{{.PrimaryField.ColumnName}} = ?",{{.PrimaryField.FieldJson}}).Error; err != nil {
- return err
- }
- return nil
- })
- {{- else }}
- err = {{$db}}.Delete(&{{.Package}}.{{.StructName}}{},"{{.PrimaryField.ColumnName}} = ?",{{.PrimaryField.FieldJson}}).Error
- {{- end }}
- return err
- }
- // Delete{{.StructName}}ByIds 批量删除{{.Description}}记录
- // Author [piexlmax](https://github.com/piexlmax)
- func ({{.Abbreviation}}Service *{{.StructName}}Service)Delete{{.StructName}}ByIds({{.PrimaryField.FieldJson}}s []string {{- if .AutoCreateResource }},deleted_by uint{{- end}}) (err error) {
- {{- if .AutoCreateResource }}
- err = {{$db}}.Transaction(func(tx *gorm.DB) error {
- if err := tx.Model(&{{.Package}}.{{.StructName}}{}).Where("{{.PrimaryField.ColumnName}} in ?", {{.PrimaryField.FieldJson}}s).Update("deleted_by", deleted_by).Error; err != nil {
- return err
- }
- if err := tx.Where("{{.PrimaryField.ColumnName}} in ?", {{.PrimaryField.FieldJson}}s).Delete(&{{.Package}}.{{.StructName}}{}).Error; err != nil {
- return err
- }
- return nil
- })
- {{- else}}
- err = {{$db}}.Delete(&[]{{.Package}}.{{.StructName}}{},"{{.PrimaryField.ColumnName}} in ?",{{.PrimaryField.FieldJson}}s).Error
- {{- end}}
- return err
- }
- // Update{{.StructName}} 更新{{.Description}}记录
- // Author [piexlmax](https://github.com/piexlmax)
- func ({{.Abbreviation}}Service *{{.StructName}}Service)Update{{.StructName}}({{.Abbreviation}} {{.Package}}.{{.StructName}}) (err error) {
- err = {{$db}}.Model(&{{.Package}}.{{.StructName}}{}).Where("{{.PrimaryField.ColumnName}} = ?",{{.Abbreviation}}.{{.PrimaryField.FieldName}}).Updates(&{{.Abbreviation}}).Error
- return err
- }
- // Get{{.StructName}} 根据{{.PrimaryField.FieldJson}}获取{{.Description}}记录
- // Author [piexlmax](https://github.com/piexlmax)
- func ({{.Abbreviation}}Service *{{.StructName}}Service)Get{{.StructName}}({{.PrimaryField.FieldJson}} string) ({{.Abbreviation}} {{.Package}}.{{.StructName}}, err error) {
- err = {{$db}}.Where("{{.PrimaryField.ColumnName}} = ?", {{.PrimaryField.FieldJson}}).First(&{{.Abbreviation}}).Error
- return
- }
- // Get{{.StructName}}InfoList 分页获取{{.Description}}记录
- // Author [piexlmax](https://github.com/piexlmax)
- func ({{.Abbreviation}}Service *{{.StructName}}Service)Get{{.StructName}}InfoList(info {{.Package}}Req.{{.StructName}}Search) (list []{{.Package}}.{{.StructName}}, total int64, err error) {
- limit := info.PageSize
- offset := info.PageSize * (info.Page - 1)
- // 创建db
- db := {{$db}}.Model(&{{.Package}}.{{.StructName}}{})
- var {{.Abbreviation}}s []{{.Package}}.{{.StructName}}
- // 如果有条件搜索 下方会自动创建搜索语句
- {{- if .GvaModel }}
- if info.StartCreatedAt !=nil && info.EndCreatedAt !=nil {
- db = db.Where("created_at BETWEEN ? AND ?", info.StartCreatedAt, info.EndCreatedAt)
- }
- {{- end }}
- {{- range .Fields}}
- {{- if .FieldSearchType}}
- {{- if or (eq .FieldType "string") (eq .FieldType "enum") (eq .FieldType "picture") (eq .FieldType "video") (eq .FieldType "richtext") }}
- if info.{{.FieldName}} != "" {
- db = db.Where("{{.ColumnName}} {{.FieldSearchType}} ?",{{if eq .FieldSearchType "LIKE"}}"%"+ {{ end }}info.{{.FieldName}}{{if eq .FieldSearchType "LIKE"}}+"%"{{ end }})
- }
- {{- else if eq .FieldSearchType "BETWEEN" "NOT BETWEEN"}}
- if info.Start{{.FieldName}} != nil && info.End{{.FieldName}} != nil {
- db = db.Where("{{.ColumnName}} {{.FieldSearchType}} ? AND ? ",info.Start{{.FieldName}},info.End{{.FieldName}})
- }
- {{- else}}
- if info.{{.FieldName}} != nil {
- db = db.Where("{{.ColumnName}} {{.FieldSearchType}} ?",{{if eq .FieldSearchType "LIKE"}}"%"+{{ end }}info.{{.FieldName}}{{if eq .FieldSearchType "LIKE"}}+"%"{{ end }})
- }
- {{- end }}
- {{- end }}
- {{- end }}
- err = db.Count(&total).Error
- if err!=nil {
- return
- }
- {{- if .NeedSort}}
- var OrderStr string
- orderMap := make(map[string]bool)
- {{- range .Fields}}
- {{- if .Sort}}
- orderMap["{{.ColumnName}}"] = true
- {{- end}}
- {{- end}}
- if orderMap[info.Sort] {
- OrderStr = info.Sort
- if info.Order == "descending" {
- OrderStr = OrderStr + " desc"
- }
- db = db.Order(OrderStr)
- }
- {{- end}}
- if limit != 0 {
- db = db.Limit(limit).Offset(offset)
- }
-
- err = db.Find(&{{.Abbreviation}}s).Error
- return {{.Abbreviation}}s, total, err
- }
- {{- if .HasDataSource }}
- func ({{.Abbreviation}}Service *{{.StructName}}Service)Get{{.StructName}}DataSource() (res map[string][]map[string]any, err error) {
- res = make(map[string][]map[string]any)
- {{range $key, $value := .DataSourceMap}}
- {{$key}} := make([]map[string]any, 0)
- {{$db}}.Table("{{$value.Table}}").Select("{{$value.Label}} as label,{{$value.Value}} as value").Scan(&{{$key}})
- res["{{$key}}"] = {{$key}}
- {{- end }}
- return
- }
- {{- end }}
|