environments.gen.go 10.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378
  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. "strings"
  8. "gorm.io/gorm"
  9. "gorm.io/gorm/clause"
  10. "gorm.io/gorm/schema"
  11. "gorm.io/gen"
  12. "gorm.io/gen/field"
  13. "gorm.io/plugin/dbresolver"
  14. "github.com/0xJacky/Nginx-UI/model"
  15. )
  16. func newEnvironment(db *gorm.DB, opts ...gen.DOOption) environment {
  17. _environment := environment{}
  18. _environment.environmentDo.UseDB(db, opts...)
  19. _environment.environmentDo.UseModel(&model.Environment{})
  20. tableName := _environment.environmentDo.TableName()
  21. _environment.ALL = field.NewAsterisk(tableName)
  22. _environment.ID = field.NewUint64(tableName, "id")
  23. _environment.CreatedAt = field.NewTime(tableName, "created_at")
  24. _environment.UpdatedAt = field.NewTime(tableName, "updated_at")
  25. _environment.DeletedAt = field.NewField(tableName, "deleted_at")
  26. _environment.Name = field.NewString(tableName, "name")
  27. _environment.URL = field.NewString(tableName, "url")
  28. _environment.Token = field.NewString(tableName, "token")
  29. _environment.Enabled = field.NewBool(tableName, "enabled")
  30. _environment.fillFieldMap()
  31. return _environment
  32. }
  33. type environment struct {
  34. environmentDo
  35. ALL field.Asterisk
  36. ID field.Uint64
  37. CreatedAt field.Time
  38. UpdatedAt field.Time
  39. DeletedAt field.Field
  40. Name field.String
  41. URL field.String
  42. Token field.String
  43. Enabled field.Bool
  44. fieldMap map[string]field.Expr
  45. }
  46. func (e environment) Table(newTableName string) *environment {
  47. e.environmentDo.UseTable(newTableName)
  48. return e.updateTableName(newTableName)
  49. }
  50. func (e environment) As(alias string) *environment {
  51. e.environmentDo.DO = *(e.environmentDo.As(alias).(*gen.DO))
  52. return e.updateTableName(alias)
  53. }
  54. func (e *environment) updateTableName(table string) *environment {
  55. e.ALL = field.NewAsterisk(table)
  56. e.ID = field.NewUint64(table, "id")
  57. e.CreatedAt = field.NewTime(table, "created_at")
  58. e.UpdatedAt = field.NewTime(table, "updated_at")
  59. e.DeletedAt = field.NewField(table, "deleted_at")
  60. e.Name = field.NewString(table, "name")
  61. e.URL = field.NewString(table, "url")
  62. e.Token = field.NewString(table, "token")
  63. e.Enabled = field.NewBool(table, "enabled")
  64. e.fillFieldMap()
  65. return e
  66. }
  67. func (e *environment) GetFieldByName(fieldName string) (field.OrderExpr, bool) {
  68. _f, ok := e.fieldMap[fieldName]
  69. if !ok || _f == nil {
  70. return nil, false
  71. }
  72. _oe, ok := _f.(field.OrderExpr)
  73. return _oe, ok
  74. }
  75. func (e *environment) fillFieldMap() {
  76. e.fieldMap = make(map[string]field.Expr, 8)
  77. e.fieldMap["id"] = e.ID
  78. e.fieldMap["created_at"] = e.CreatedAt
  79. e.fieldMap["updated_at"] = e.UpdatedAt
  80. e.fieldMap["deleted_at"] = e.DeletedAt
  81. e.fieldMap["name"] = e.Name
  82. e.fieldMap["url"] = e.URL
  83. e.fieldMap["token"] = e.Token
  84. e.fieldMap["enabled"] = e.Enabled
  85. }
  86. func (e environment) clone(db *gorm.DB) environment {
  87. e.environmentDo.ReplaceConnPool(db.Statement.ConnPool)
  88. return e
  89. }
  90. func (e environment) replaceDB(db *gorm.DB) environment {
  91. e.environmentDo.ReplaceDB(db)
  92. return e
  93. }
  94. type environmentDo struct{ gen.DO }
  95. // FirstByID Where("id=@id")
  96. func (e environmentDo) FirstByID(id uint64) (result *model.Environment, err error) {
  97. var params []interface{}
  98. var generateSQL strings.Builder
  99. params = append(params, id)
  100. generateSQL.WriteString("id=? ")
  101. var executeSQL *gorm.DB
  102. executeSQL = e.UnderlyingDB().Where(generateSQL.String(), params...).Take(&result) // ignore_security_alert
  103. err = executeSQL.Error
  104. return
  105. }
  106. // DeleteByID update @@table set deleted_at=strftime('%Y-%m-%d %H:%M:%S','now') where id=@id
  107. func (e environmentDo) DeleteByID(id uint64) (err error) {
  108. var params []interface{}
  109. var generateSQL strings.Builder
  110. params = append(params, id)
  111. generateSQL.WriteString("update environments set deleted_at=strftime('%Y-%m-%d %H:%M:%S','now') where id=? ")
  112. var executeSQL *gorm.DB
  113. executeSQL = e.UnderlyingDB().Exec(generateSQL.String(), params...) // ignore_security_alert
  114. err = executeSQL.Error
  115. return
  116. }
  117. func (e environmentDo) Debug() *environmentDo {
  118. return e.withDO(e.DO.Debug())
  119. }
  120. func (e environmentDo) WithContext(ctx context.Context) *environmentDo {
  121. return e.withDO(e.DO.WithContext(ctx))
  122. }
  123. func (e environmentDo) ReadDB() *environmentDo {
  124. return e.Clauses(dbresolver.Read)
  125. }
  126. func (e environmentDo) WriteDB() *environmentDo {
  127. return e.Clauses(dbresolver.Write)
  128. }
  129. func (e environmentDo) Session(config *gorm.Session) *environmentDo {
  130. return e.withDO(e.DO.Session(config))
  131. }
  132. func (e environmentDo) Clauses(conds ...clause.Expression) *environmentDo {
  133. return e.withDO(e.DO.Clauses(conds...))
  134. }
  135. func (e environmentDo) Returning(value interface{}, columns ...string) *environmentDo {
  136. return e.withDO(e.DO.Returning(value, columns...))
  137. }
  138. func (e environmentDo) Not(conds ...gen.Condition) *environmentDo {
  139. return e.withDO(e.DO.Not(conds...))
  140. }
  141. func (e environmentDo) Or(conds ...gen.Condition) *environmentDo {
  142. return e.withDO(e.DO.Or(conds...))
  143. }
  144. func (e environmentDo) Select(conds ...field.Expr) *environmentDo {
  145. return e.withDO(e.DO.Select(conds...))
  146. }
  147. func (e environmentDo) Where(conds ...gen.Condition) *environmentDo {
  148. return e.withDO(e.DO.Where(conds...))
  149. }
  150. func (e environmentDo) Order(conds ...field.Expr) *environmentDo {
  151. return e.withDO(e.DO.Order(conds...))
  152. }
  153. func (e environmentDo) Distinct(cols ...field.Expr) *environmentDo {
  154. return e.withDO(e.DO.Distinct(cols...))
  155. }
  156. func (e environmentDo) Omit(cols ...field.Expr) *environmentDo {
  157. return e.withDO(e.DO.Omit(cols...))
  158. }
  159. func (e environmentDo) Join(table schema.Tabler, on ...field.Expr) *environmentDo {
  160. return e.withDO(e.DO.Join(table, on...))
  161. }
  162. func (e environmentDo) LeftJoin(table schema.Tabler, on ...field.Expr) *environmentDo {
  163. return e.withDO(e.DO.LeftJoin(table, on...))
  164. }
  165. func (e environmentDo) RightJoin(table schema.Tabler, on ...field.Expr) *environmentDo {
  166. return e.withDO(e.DO.RightJoin(table, on...))
  167. }
  168. func (e environmentDo) Group(cols ...field.Expr) *environmentDo {
  169. return e.withDO(e.DO.Group(cols...))
  170. }
  171. func (e environmentDo) Having(conds ...gen.Condition) *environmentDo {
  172. return e.withDO(e.DO.Having(conds...))
  173. }
  174. func (e environmentDo) Limit(limit int) *environmentDo {
  175. return e.withDO(e.DO.Limit(limit))
  176. }
  177. func (e environmentDo) Offset(offset int) *environmentDo {
  178. return e.withDO(e.DO.Offset(offset))
  179. }
  180. func (e environmentDo) Scopes(funcs ...func(gen.Dao) gen.Dao) *environmentDo {
  181. return e.withDO(e.DO.Scopes(funcs...))
  182. }
  183. func (e environmentDo) Unscoped() *environmentDo {
  184. return e.withDO(e.DO.Unscoped())
  185. }
  186. func (e environmentDo) Create(values ...*model.Environment) error {
  187. if len(values) == 0 {
  188. return nil
  189. }
  190. return e.DO.Create(values)
  191. }
  192. func (e environmentDo) CreateInBatches(values []*model.Environment, batchSize int) error {
  193. return e.DO.CreateInBatches(values, batchSize)
  194. }
  195. // Save : !!! underlying implementation is different with GORM
  196. // The method is equivalent to executing the statement: db.Clauses(clause.OnConflict{UpdateAll: true}).Create(values)
  197. func (e environmentDo) Save(values ...*model.Environment) error {
  198. if len(values) == 0 {
  199. return nil
  200. }
  201. return e.DO.Save(values)
  202. }
  203. func (e environmentDo) First() (*model.Environment, error) {
  204. if result, err := e.DO.First(); err != nil {
  205. return nil, err
  206. } else {
  207. return result.(*model.Environment), nil
  208. }
  209. }
  210. func (e environmentDo) Take() (*model.Environment, error) {
  211. if result, err := e.DO.Take(); err != nil {
  212. return nil, err
  213. } else {
  214. return result.(*model.Environment), nil
  215. }
  216. }
  217. func (e environmentDo) Last() (*model.Environment, error) {
  218. if result, err := e.DO.Last(); err != nil {
  219. return nil, err
  220. } else {
  221. return result.(*model.Environment), nil
  222. }
  223. }
  224. func (e environmentDo) Find() ([]*model.Environment, error) {
  225. result, err := e.DO.Find()
  226. return result.([]*model.Environment), err
  227. }
  228. func (e environmentDo) FindInBatch(batchSize int, fc func(tx gen.Dao, batch int) error) (results []*model.Environment, err error) {
  229. buf := make([]*model.Environment, 0, batchSize)
  230. err = e.DO.FindInBatches(&buf, batchSize, func(tx gen.Dao, batch int) error {
  231. defer func() { results = append(results, buf...) }()
  232. return fc(tx, batch)
  233. })
  234. return results, err
  235. }
  236. func (e environmentDo) FindInBatches(result *[]*model.Environment, batchSize int, fc func(tx gen.Dao, batch int) error) error {
  237. return e.DO.FindInBatches(result, batchSize, fc)
  238. }
  239. func (e environmentDo) Attrs(attrs ...field.AssignExpr) *environmentDo {
  240. return e.withDO(e.DO.Attrs(attrs...))
  241. }
  242. func (e environmentDo) Assign(attrs ...field.AssignExpr) *environmentDo {
  243. return e.withDO(e.DO.Assign(attrs...))
  244. }
  245. func (e environmentDo) Joins(fields ...field.RelationField) *environmentDo {
  246. for _, _f := range fields {
  247. e = *e.withDO(e.DO.Joins(_f))
  248. }
  249. return &e
  250. }
  251. func (e environmentDo) Preload(fields ...field.RelationField) *environmentDo {
  252. for _, _f := range fields {
  253. e = *e.withDO(e.DO.Preload(_f))
  254. }
  255. return &e
  256. }
  257. func (e environmentDo) FirstOrInit() (*model.Environment, error) {
  258. if result, err := e.DO.FirstOrInit(); err != nil {
  259. return nil, err
  260. } else {
  261. return result.(*model.Environment), nil
  262. }
  263. }
  264. func (e environmentDo) FirstOrCreate() (*model.Environment, error) {
  265. if result, err := e.DO.FirstOrCreate(); err != nil {
  266. return nil, err
  267. } else {
  268. return result.(*model.Environment), nil
  269. }
  270. }
  271. func (e environmentDo) FindByPage(offset int, limit int) (result []*model.Environment, count int64, err error) {
  272. result, err = e.Offset(offset).Limit(limit).Find()
  273. if err != nil {
  274. return
  275. }
  276. if size := len(result); 0 < limit && 0 < size && size < limit {
  277. count = int64(size + offset)
  278. return
  279. }
  280. count, err = e.Offset(-1).Limit(-1).Count()
  281. return
  282. }
  283. func (e environmentDo) ScanByPage(result interface{}, offset int, limit int) (count int64, err error) {
  284. count, err = e.Count()
  285. if err != nil {
  286. return
  287. }
  288. err = e.Offset(offset).Limit(limit).Scan(result)
  289. return
  290. }
  291. func (e environmentDo) Scan(result interface{}) (err error) {
  292. return e.DO.Scan(result)
  293. }
  294. func (e environmentDo) Delete(models ...*model.Environment) (result gen.ResultInfo, err error) {
  295. return e.DO.Delete(models)
  296. }
  297. func (e *environmentDo) withDO(do gen.Dao) *environmentDo {
  298. e.DO = *do.(*gen.DO)
  299. return e
  300. }