service.go.tpl 6.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151
  1. package {{.Package}}
  2. import (
  3. "github.com/flipped-aurora/gin-vue-admin/server/global"
  4. "github.com/flipped-aurora/gin-vue-admin/server/model/{{.Package}}"
  5. {{.Package}}Req "github.com/flipped-aurora/gin-vue-admin/server/model/{{.Package}}/request"
  6. {{- if .AutoCreateResource }}
  7. "gorm.io/gorm"
  8. {{- end}}
  9. )
  10. type {{.StructName}}Service struct {
  11. }
  12. {{- $db := "" }}
  13. {{- if eq .BusinessDB "" }}
  14. {{- $db = "global.GVA_DB" }}
  15. {{- else}}
  16. {{- $db = printf "global.MustGetGlobalDBByDBName(\"%s\")" .BusinessDB }}
  17. {{- end}}
  18. // Create{{.StructName}} 创建{{.Description}}记录
  19. // Author [piexlmax](https://github.com/piexlmax)
  20. func ({{.Abbreviation}}Service *{{.StructName}}Service) Create{{.StructName}}({{.Abbreviation}} *{{.Package}}.{{.StructName}}) (err error) {
  21. err = {{$db}}.Create({{.Abbreviation}}).Error
  22. return err
  23. }
  24. // Delete{{.StructName}} 删除{{.Description}}记录
  25. // Author [piexlmax](https://github.com/piexlmax)
  26. func ({{.Abbreviation}}Service *{{.StructName}}Service)Delete{{.StructName}}({{.PrimaryField.FieldJson}} string{{- if .AutoCreateResource -}},userID uint{{- end -}}) (err error) {
  27. {{- if .AutoCreateResource }}
  28. err = {{$db}}.Transaction(func(tx *gorm.DB) error {
  29. if err := tx.Model(&{{.Package}}.{{.StructName}}{}).Where("{{.PrimaryField.ColumnName}} = ?", {{.PrimaryField.FieldJson}}).Update("deleted_by", userID).Error; err != nil {
  30. return err
  31. }
  32. if err = tx.Delete(&{{.Package}}.{{.StructName}}{},"{{.PrimaryField.ColumnName}} = ?",{{.PrimaryField.FieldJson}}).Error; err != nil {
  33. return err
  34. }
  35. return nil
  36. })
  37. {{- else }}
  38. err = {{$db}}.Delete(&{{.Package}}.{{.StructName}}{},"{{.PrimaryField.ColumnName}} = ?",{{.PrimaryField.FieldJson}}).Error
  39. {{- end }}
  40. return err
  41. }
  42. // Delete{{.StructName}}ByIds 批量删除{{.Description}}记录
  43. // Author [piexlmax](https://github.com/piexlmax)
  44. func ({{.Abbreviation}}Service *{{.StructName}}Service)Delete{{.StructName}}ByIds({{.PrimaryField.FieldJson}}s []string {{- if .AutoCreateResource }},deleted_by uint{{- end}}) (err error) {
  45. {{- if .AutoCreateResource }}
  46. err = {{$db}}.Transaction(func(tx *gorm.DB) error {
  47. if err := tx.Model(&{{.Package}}.{{.StructName}}{}).Where("{{.PrimaryField.ColumnName}} in ?", {{.PrimaryField.FieldJson}}s).Update("deleted_by", deleted_by).Error; err != nil {
  48. return err
  49. }
  50. if err := tx.Where("{{.PrimaryField.ColumnName}} in ?", {{.PrimaryField.FieldJson}}s).Delete(&{{.Package}}.{{.StructName}}{}).Error; err != nil {
  51. return err
  52. }
  53. return nil
  54. })
  55. {{- else}}
  56. err = {{$db}}.Delete(&[]{{.Package}}.{{.StructName}}{},"{{.PrimaryField.ColumnName}} in ?",{{.PrimaryField.FieldJson}}s).Error
  57. {{- end}}
  58. return err
  59. }
  60. // Update{{.StructName}} 更新{{.Description}}记录
  61. // Author [piexlmax](https://github.com/piexlmax)
  62. func ({{.Abbreviation}}Service *{{.StructName}}Service)Update{{.StructName}}({{.Abbreviation}} {{.Package}}.{{.StructName}}) (err error) {
  63. err = {{$db}}.Model(&{{.Package}}.{{.StructName}}{}).Where("{{.PrimaryField.ColumnName}} = ?",{{.Abbreviation}}.{{.PrimaryField.FieldName}}).Updates(&{{.Abbreviation}}).Error
  64. return err
  65. }
  66. // Get{{.StructName}} 根据{{.PrimaryField.FieldJson}}获取{{.Description}}记录
  67. // Author [piexlmax](https://github.com/piexlmax)
  68. func ({{.Abbreviation}}Service *{{.StructName}}Service)Get{{.StructName}}({{.PrimaryField.FieldJson}} string) ({{.Abbreviation}} {{.Package}}.{{.StructName}}, err error) {
  69. err = {{$db}}.Where("{{.PrimaryField.ColumnName}} = ?", {{.PrimaryField.FieldJson}}).First(&{{.Abbreviation}}).Error
  70. return
  71. }
  72. // Get{{.StructName}}InfoList 分页获取{{.Description}}记录
  73. // Author [piexlmax](https://github.com/piexlmax)
  74. func ({{.Abbreviation}}Service *{{.StructName}}Service)Get{{.StructName}}InfoList(info {{.Package}}Req.{{.StructName}}Search) (list []{{.Package}}.{{.StructName}}, total int64, err error) {
  75. limit := info.PageSize
  76. offset := info.PageSize * (info.Page - 1)
  77. // 创建db
  78. db := {{$db}}.Model(&{{.Package}}.{{.StructName}}{})
  79. var {{.Abbreviation}}s []{{.Package}}.{{.StructName}}
  80. // 如果有条件搜索 下方会自动创建搜索语句
  81. {{- if .GvaModel }}
  82. if info.StartCreatedAt !=nil && info.EndCreatedAt !=nil {
  83. db = db.Where("created_at BETWEEN ? AND ?", info.StartCreatedAt, info.EndCreatedAt)
  84. }
  85. {{- end }}
  86. {{- range .Fields}}
  87. {{- if .FieldSearchType}}
  88. {{- if or (eq .FieldType "string") (eq .FieldType "enum") (eq .FieldType "picture") (eq .FieldType "video") (eq .FieldType "richtext") }}
  89. if info.{{.FieldName}} != "" {
  90. db = db.Where("{{.ColumnName}} {{.FieldSearchType}} ?",{{if eq .FieldSearchType "LIKE"}}"%"+ {{ end }}info.{{.FieldName}}{{if eq .FieldSearchType "LIKE"}}+"%"{{ end }})
  91. }
  92. {{- else if eq .FieldSearchType "BETWEEN" "NOT BETWEEN"}}
  93. if info.Start{{.FieldName}} != nil && info.End{{.FieldName}} != nil {
  94. db = db.Where("{{.ColumnName}} {{.FieldSearchType}} ? AND ? ",info.Start{{.FieldName}},info.End{{.FieldName}})
  95. }
  96. {{- else}}
  97. if info.{{.FieldName}} != nil {
  98. db = db.Where("{{.ColumnName}} {{.FieldSearchType}} ?",{{if eq .FieldSearchType "LIKE"}}"%"+{{ end }}info.{{.FieldName}}{{if eq .FieldSearchType "LIKE"}}+"%"{{ end }})
  99. }
  100. {{- end }}
  101. {{- end }}
  102. {{- end }}
  103. err = db.Count(&total).Error
  104. if err!=nil {
  105. return
  106. }
  107. {{- if .NeedSort}}
  108. var OrderStr string
  109. orderMap := make(map[string]bool)
  110. {{- range .Fields}}
  111. {{- if .Sort}}
  112. orderMap["{{.ColumnName}}"] = true
  113. {{- end}}
  114. {{- end}}
  115. if orderMap[info.Sort] {
  116. OrderStr = info.Sort
  117. if info.Order == "descending" {
  118. OrderStr = OrderStr + " desc"
  119. }
  120. db = db.Order(OrderStr)
  121. }
  122. {{- end}}
  123. if limit != 0 {
  124. db = db.Limit(limit).Offset(offset)
  125. }
  126. err = db.Find(&{{.Abbreviation}}s).Error
  127. return {{.Abbreviation}}s, total, err
  128. }
  129. {{- if .HasDataSource }}
  130. func ({{.Abbreviation}}Service *{{.StructName}}Service)Get{{.StructName}}DataSource() (res map[string][]map[string]any, err error) {
  131. res = make(map[string][]map[string]any)
  132. {{range $key, $value := .DataSourceMap}}
  133. {{$key}} := make([]map[string]any, 0)
  134. {{$db}}.Table("{{$value.Table}}").Select("{{$value.Label}} as label,{{$value.Value}} as value").Scan(&{{$key}})
  135. res["{{$key}}"] = {{$key}}
  136. {{- end }}
  137. return
  138. }
  139. {{- end }}