environments.gen.go 10 KB

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