retrofit.gen.go 9.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359
  1. // Code generated by gorm.io/gen. DO NOT EDIT.
  2. // Code generated by gorm.io/gen. DO NOT EDIT.
  3. // Code generated by gorm.io/gen. DO NOT EDIT.
  4. package query
  5. import (
  6. "context"
  7. "gorm.io/gorm"
  8. "gorm.io/gorm/clause"
  9. "gorm.io/gorm/schema"
  10. "gorm.io/gen"
  11. "gorm.io/gen/field"
  12. "gorm.io/plugin/dbresolver"
  13. "gadmin/internal/gorm/model"
  14. )
  15. func newRetrofit(db *gorm.DB, opts ...gen.DOOption) retrofit {
  16. _retrofit := retrofit{}
  17. _retrofit.retrofitDo.UseDB(db, opts...)
  18. _retrofit.retrofitDo.UseModel(&model.Retrofit{})
  19. tableName := _retrofit.retrofitDo.TableName()
  20. _retrofit.ALL = field.NewAsterisk(tableName)
  21. _retrofit.ID = field.NewInt64(tableName, "id")
  22. _retrofit.GroupID = field.NewInt64(tableName, "group_id")
  23. _retrofit.Name = field.NewString(tableName, "name")
  24. _retrofit.Type = field.NewInt64(tableName, "type")
  25. _retrofit.Mid = field.NewInt64(tableName, "mid")
  26. _retrofit.Nums = field.NewInt64(tableName, "nums")
  27. _retrofit.Status = field.NewInt32(tableName, "status")
  28. _retrofit.CreatedAt = field.NewTime(tableName, "created_at")
  29. _retrofit.UpdatedAt = field.NewTime(tableName, "updated_at")
  30. _retrofit.DeletedAt = field.NewField(tableName, "deleted_at")
  31. _retrofit.fillFieldMap()
  32. return _retrofit
  33. }
  34. type retrofit struct {
  35. retrofitDo
  36. ALL field.Asterisk
  37. ID field.Int64
  38. GroupID field.Int64 // 模板ID
  39. Name field.String // 别名
  40. Type field.Int64 // 加装类型
  41. Mid field.Int64 // 关联ID
  42. Nums field.Int64 // 数量
  43. Status field.Int32 // 状态
  44. CreatedAt field.Time // 创建时间
  45. UpdatedAt field.Time // 更新时间
  46. DeletedAt field.Field // 删除时间
  47. fieldMap map[string]field.Expr
  48. }
  49. func (r retrofit) Table(newTableName string) *retrofit {
  50. r.retrofitDo.UseTable(newTableName)
  51. return r.updateTableName(newTableName)
  52. }
  53. func (r retrofit) As(alias string) *retrofit {
  54. r.retrofitDo.DO = *(r.retrofitDo.As(alias).(*gen.DO))
  55. return r.updateTableName(alias)
  56. }
  57. func (r *retrofit) updateTableName(table string) *retrofit {
  58. r.ALL = field.NewAsterisk(table)
  59. r.ID = field.NewInt64(table, "id")
  60. r.GroupID = field.NewInt64(table, "group_id")
  61. r.Name = field.NewString(table, "name")
  62. r.Type = field.NewInt64(table, "type")
  63. r.Mid = field.NewInt64(table, "mid")
  64. r.Nums = field.NewInt64(table, "nums")
  65. r.Status = field.NewInt32(table, "status")
  66. r.CreatedAt = field.NewTime(table, "created_at")
  67. r.UpdatedAt = field.NewTime(table, "updated_at")
  68. r.DeletedAt = field.NewField(table, "deleted_at")
  69. r.fillFieldMap()
  70. return r
  71. }
  72. func (r *retrofit) GetFieldByName(fieldName string) (field.OrderExpr, bool) {
  73. _f, ok := r.fieldMap[fieldName]
  74. if !ok || _f == nil {
  75. return nil, false
  76. }
  77. _oe, ok := _f.(field.OrderExpr)
  78. return _oe, ok
  79. }
  80. func (r *retrofit) fillFieldMap() {
  81. r.fieldMap = make(map[string]field.Expr, 10)
  82. r.fieldMap["id"] = r.ID
  83. r.fieldMap["group_id"] = r.GroupID
  84. r.fieldMap["name"] = r.Name
  85. r.fieldMap["type"] = r.Type
  86. r.fieldMap["mid"] = r.Mid
  87. r.fieldMap["nums"] = r.Nums
  88. r.fieldMap["status"] = r.Status
  89. r.fieldMap["created_at"] = r.CreatedAt
  90. r.fieldMap["updated_at"] = r.UpdatedAt
  91. r.fieldMap["deleted_at"] = r.DeletedAt
  92. }
  93. func (r retrofit) clone(db *gorm.DB) retrofit {
  94. r.retrofitDo.ReplaceConnPool(db.Statement.ConnPool)
  95. return r
  96. }
  97. func (r retrofit) replaceDB(db *gorm.DB) retrofit {
  98. r.retrofitDo.ReplaceDB(db)
  99. return r
  100. }
  101. type retrofitDo struct{ gen.DO }
  102. func (r retrofitDo) Debug() *retrofitDo {
  103. return r.withDO(r.DO.Debug())
  104. }
  105. func (r retrofitDo) WithContext(ctx context.Context) *retrofitDo {
  106. return r.withDO(r.DO.WithContext(ctx))
  107. }
  108. func (r retrofitDo) ReadDB() *retrofitDo {
  109. return r.Clauses(dbresolver.Read)
  110. }
  111. func (r retrofitDo) WriteDB() *retrofitDo {
  112. return r.Clauses(dbresolver.Write)
  113. }
  114. func (r retrofitDo) Session(config *gorm.Session) *retrofitDo {
  115. return r.withDO(r.DO.Session(config))
  116. }
  117. func (r retrofitDo) Clauses(conds ...clause.Expression) *retrofitDo {
  118. return r.withDO(r.DO.Clauses(conds...))
  119. }
  120. func (r retrofitDo) Returning(value interface{}, columns ...string) *retrofitDo {
  121. return r.withDO(r.DO.Returning(value, columns...))
  122. }
  123. func (r retrofitDo) Not(conds ...gen.Condition) *retrofitDo {
  124. return r.withDO(r.DO.Not(conds...))
  125. }
  126. func (r retrofitDo) Or(conds ...gen.Condition) *retrofitDo {
  127. return r.withDO(r.DO.Or(conds...))
  128. }
  129. func (r retrofitDo) Select(conds ...field.Expr) *retrofitDo {
  130. return r.withDO(r.DO.Select(conds...))
  131. }
  132. func (r retrofitDo) Where(conds ...gen.Condition) *retrofitDo {
  133. return r.withDO(r.DO.Where(conds...))
  134. }
  135. func (r retrofitDo) Exists(subquery interface{ UnderlyingDB() *gorm.DB }) *retrofitDo {
  136. return r.Where(field.CompareSubQuery(field.ExistsOp, nil, subquery.UnderlyingDB()))
  137. }
  138. func (r retrofitDo) Order(conds ...field.Expr) *retrofitDo {
  139. return r.withDO(r.DO.Order(conds...))
  140. }
  141. func (r retrofitDo) Distinct(cols ...field.Expr) *retrofitDo {
  142. return r.withDO(r.DO.Distinct(cols...))
  143. }
  144. func (r retrofitDo) Omit(cols ...field.Expr) *retrofitDo {
  145. return r.withDO(r.DO.Omit(cols...))
  146. }
  147. func (r retrofitDo) Join(table schema.Tabler, on ...field.Expr) *retrofitDo {
  148. return r.withDO(r.DO.Join(table, on...))
  149. }
  150. func (r retrofitDo) LeftJoin(table schema.Tabler, on ...field.Expr) *retrofitDo {
  151. return r.withDO(r.DO.LeftJoin(table, on...))
  152. }
  153. func (r retrofitDo) RightJoin(table schema.Tabler, on ...field.Expr) *retrofitDo {
  154. return r.withDO(r.DO.RightJoin(table, on...))
  155. }
  156. func (r retrofitDo) Group(cols ...field.Expr) *retrofitDo {
  157. return r.withDO(r.DO.Group(cols...))
  158. }
  159. func (r retrofitDo) Having(conds ...gen.Condition) *retrofitDo {
  160. return r.withDO(r.DO.Having(conds...))
  161. }
  162. func (r retrofitDo) Limit(limit int) *retrofitDo {
  163. return r.withDO(r.DO.Limit(limit))
  164. }
  165. func (r retrofitDo) Offset(offset int) *retrofitDo {
  166. return r.withDO(r.DO.Offset(offset))
  167. }
  168. func (r retrofitDo) Scopes(funcs ...func(gen.Dao) gen.Dao) *retrofitDo {
  169. return r.withDO(r.DO.Scopes(funcs...))
  170. }
  171. func (r retrofitDo) Unscoped() *retrofitDo {
  172. return r.withDO(r.DO.Unscoped())
  173. }
  174. func (r retrofitDo) Create(values ...*model.Retrofit) error {
  175. if len(values) == 0 {
  176. return nil
  177. }
  178. return r.DO.Create(values)
  179. }
  180. func (r retrofitDo) CreateInBatches(values []*model.Retrofit, batchSize int) error {
  181. return r.DO.CreateInBatches(values, batchSize)
  182. }
  183. // Save : !!! underlying implementation is different with GORM
  184. // The method is equivalent to executing the statement: db.Clauses(clause.OnConflict{UpdateAll: true}).Create(values)
  185. func (r retrofitDo) Save(values ...*model.Retrofit) error {
  186. if len(values) == 0 {
  187. return nil
  188. }
  189. return r.DO.Save(values)
  190. }
  191. func (r retrofitDo) First() (*model.Retrofit, error) {
  192. if result, err := r.DO.First(); err != nil {
  193. return nil, err
  194. } else {
  195. return result.(*model.Retrofit), nil
  196. }
  197. }
  198. func (r retrofitDo) Take() (*model.Retrofit, error) {
  199. if result, err := r.DO.Take(); err != nil {
  200. return nil, err
  201. } else {
  202. return result.(*model.Retrofit), nil
  203. }
  204. }
  205. func (r retrofitDo) Last() (*model.Retrofit, error) {
  206. if result, err := r.DO.Last(); err != nil {
  207. return nil, err
  208. } else {
  209. return result.(*model.Retrofit), nil
  210. }
  211. }
  212. func (r retrofitDo) Find() ([]*model.Retrofit, error) {
  213. result, err := r.DO.Find()
  214. return result.([]*model.Retrofit), err
  215. }
  216. func (r retrofitDo) FindInBatch(batchSize int, fc func(tx gen.Dao, batch int) error) (results []*model.Retrofit, err error) {
  217. buf := make([]*model.Retrofit, 0, batchSize)
  218. err = r.DO.FindInBatches(&buf, batchSize, func(tx gen.Dao, batch int) error {
  219. defer func() { results = append(results, buf...) }()
  220. return fc(tx, batch)
  221. })
  222. return results, err
  223. }
  224. func (r retrofitDo) FindInBatches(result *[]*model.Retrofit, batchSize int, fc func(tx gen.Dao, batch int) error) error {
  225. return r.DO.FindInBatches(result, batchSize, fc)
  226. }
  227. func (r retrofitDo) Attrs(attrs ...field.AssignExpr) *retrofitDo {
  228. return r.withDO(r.DO.Attrs(attrs...))
  229. }
  230. func (r retrofitDo) Assign(attrs ...field.AssignExpr) *retrofitDo {
  231. return r.withDO(r.DO.Assign(attrs...))
  232. }
  233. func (r retrofitDo) Joins(fields ...field.RelationField) *retrofitDo {
  234. for _, _f := range fields {
  235. r = *r.withDO(r.DO.Joins(_f))
  236. }
  237. return &r
  238. }
  239. func (r retrofitDo) Preload(fields ...field.RelationField) *retrofitDo {
  240. for _, _f := range fields {
  241. r = *r.withDO(r.DO.Preload(_f))
  242. }
  243. return &r
  244. }
  245. func (r retrofitDo) FirstOrInit() (*model.Retrofit, error) {
  246. if result, err := r.DO.FirstOrInit(); err != nil {
  247. return nil, err
  248. } else {
  249. return result.(*model.Retrofit), nil
  250. }
  251. }
  252. func (r retrofitDo) FirstOrCreate() (*model.Retrofit, error) {
  253. if result, err := r.DO.FirstOrCreate(); err != nil {
  254. return nil, err
  255. } else {
  256. return result.(*model.Retrofit), nil
  257. }
  258. }
  259. func (r retrofitDo) FindByPage(offset int, limit int) (result []*model.Retrofit, count int64, err error) {
  260. result, err = r.Offset(offset).Limit(limit).Find()
  261. if err != nil {
  262. return
  263. }
  264. if size := len(result); 0 < limit && 0 < size && size < limit {
  265. count = int64(size + offset)
  266. return
  267. }
  268. count, err = r.Offset(-1).Limit(-1).Count()
  269. return
  270. }
  271. func (r retrofitDo) ScanByPage(result interface{}, offset int, limit int) (count int64, err error) {
  272. count, err = r.Count()
  273. if err != nil {
  274. return
  275. }
  276. err = r.Offset(offset).Limit(limit).Scan(result)
  277. return
  278. }
  279. func (r retrofitDo) Scan(result interface{}) (err error) {
  280. return r.DO.Scan(result)
  281. }
  282. func (r retrofitDo) Delete(models ...*model.Retrofit) (result gen.ResultInfo, err error) {
  283. return r.DO.Delete(models)
  284. }
  285. func (r *retrofitDo) withDO(do gen.Dao) *retrofitDo {
  286. r.DO = *do.(*gen.DO)
  287. return r
  288. }