environments.gen.go 10 KB

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