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/server/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) Exists(subquery interface{ UnderlyingDB() *gorm.DB }) *environmentDo {
  155. return e.Where(field.CompareSubQuery(field.ExistsOp, nil, subquery.UnderlyingDB()))
  156. }
  157. func (e environmentDo) Order(conds ...field.Expr) *environmentDo {
  158. return e.withDO(e.DO.Order(conds...))
  159. }
  160. func (e environmentDo) Distinct(cols ...field.Expr) *environmentDo {
  161. return e.withDO(e.DO.Distinct(cols...))
  162. }
  163. func (e environmentDo) Omit(cols ...field.Expr) *environmentDo {
  164. return e.withDO(e.DO.Omit(cols...))
  165. }
  166. func (e environmentDo) Join(table schema.Tabler, on ...field.Expr) *environmentDo {
  167. return e.withDO(e.DO.Join(table, on...))
  168. }
  169. func (e environmentDo) LeftJoin(table schema.Tabler, on ...field.Expr) *environmentDo {
  170. return e.withDO(e.DO.LeftJoin(table, on...))
  171. }
  172. func (e environmentDo) RightJoin(table schema.Tabler, on ...field.Expr) *environmentDo {
  173. return e.withDO(e.DO.RightJoin(table, on...))
  174. }
  175. func (e environmentDo) Group(cols ...field.Expr) *environmentDo {
  176. return e.withDO(e.DO.Group(cols...))
  177. }
  178. func (e environmentDo) Having(conds ...gen.Condition) *environmentDo {
  179. return e.withDO(e.DO.Having(conds...))
  180. }
  181. func (e environmentDo) Limit(limit int) *environmentDo {
  182. return e.withDO(e.DO.Limit(limit))
  183. }
  184. func (e environmentDo) Offset(offset int) *environmentDo {
  185. return e.withDO(e.DO.Offset(offset))
  186. }
  187. func (e environmentDo) Scopes(funcs ...func(gen.Dao) gen.Dao) *environmentDo {
  188. return e.withDO(e.DO.Scopes(funcs...))
  189. }
  190. func (e environmentDo) Unscoped() *environmentDo {
  191. return e.withDO(e.DO.Unscoped())
  192. }
  193. func (e environmentDo) Create(values ...*model.Environment) error {
  194. if len(values) == 0 {
  195. return nil
  196. }
  197. return e.DO.Create(values)
  198. }
  199. func (e environmentDo) CreateInBatches(values []*model.Environment, batchSize int) error {
  200. return e.DO.CreateInBatches(values, batchSize)
  201. }
  202. // Save : !!! underlying implementation is different with GORM
  203. // The method is equivalent to executing the statement: db.Clauses(clause.OnConflict{UpdateAll: true}).Create(values)
  204. func (e environmentDo) Save(values ...*model.Environment) error {
  205. if len(values) == 0 {
  206. return nil
  207. }
  208. return e.DO.Save(values)
  209. }
  210. func (e environmentDo) First() (*model.Environment, error) {
  211. if result, err := e.DO.First(); err != nil {
  212. return nil, err
  213. } else {
  214. return result.(*model.Environment), nil
  215. }
  216. }
  217. func (e environmentDo) Take() (*model.Environment, error) {
  218. if result, err := e.DO.Take(); err != nil {
  219. return nil, err
  220. } else {
  221. return result.(*model.Environment), nil
  222. }
  223. }
  224. func (e environmentDo) Last() (*model.Environment, error) {
  225. if result, err := e.DO.Last(); err != nil {
  226. return nil, err
  227. } else {
  228. return result.(*model.Environment), nil
  229. }
  230. }
  231. func (e environmentDo) Find() ([]*model.Environment, error) {
  232. result, err := e.DO.Find()
  233. return result.([]*model.Environment), err
  234. }
  235. func (e environmentDo) FindInBatch(batchSize int, fc func(tx gen.Dao, batch int) error) (results []*model.Environment, err error) {
  236. buf := make([]*model.Environment, 0, batchSize)
  237. err = e.DO.FindInBatches(&buf, batchSize, func(tx gen.Dao, batch int) error {
  238. defer func() { results = append(results, buf...) }()
  239. return fc(tx, batch)
  240. })
  241. return results, err
  242. }
  243. func (e environmentDo) FindInBatches(result *[]*model.Environment, batchSize int, fc func(tx gen.Dao, batch int) error) error {
  244. return e.DO.FindInBatches(result, batchSize, fc)
  245. }
  246. func (e environmentDo) Attrs(attrs ...field.AssignExpr) *environmentDo {
  247. return e.withDO(e.DO.Attrs(attrs...))
  248. }
  249. func (e environmentDo) Assign(attrs ...field.AssignExpr) *environmentDo {
  250. return e.withDO(e.DO.Assign(attrs...))
  251. }
  252. func (e environmentDo) Joins(fields ...field.RelationField) *environmentDo {
  253. for _, _f := range fields {
  254. e = *e.withDO(e.DO.Joins(_f))
  255. }
  256. return &e
  257. }
  258. func (e environmentDo) Preload(fields ...field.RelationField) *environmentDo {
  259. for _, _f := range fields {
  260. e = *e.withDO(e.DO.Preload(_f))
  261. }
  262. return &e
  263. }
  264. func (e environmentDo) FirstOrInit() (*model.Environment, error) {
  265. if result, err := e.DO.FirstOrInit(); err != nil {
  266. return nil, err
  267. } else {
  268. return result.(*model.Environment), nil
  269. }
  270. }
  271. func (e environmentDo) FirstOrCreate() (*model.Environment, error) {
  272. if result, err := e.DO.FirstOrCreate(); err != nil {
  273. return nil, err
  274. } else {
  275. return result.(*model.Environment), nil
  276. }
  277. }
  278. func (e environmentDo) FindByPage(offset int, limit int) (result []*model.Environment, count int64, err error) {
  279. result, err = e.Offset(offset).Limit(limit).Find()
  280. if err != nil {
  281. return
  282. }
  283. if size := len(result); 0 < limit && 0 < size && size < limit {
  284. count = int64(size + offset)
  285. return
  286. }
  287. count, err = e.Offset(-1).Limit(-1).Count()
  288. return
  289. }
  290. func (e environmentDo) ScanByPage(result interface{}, offset int, limit int) (count int64, err error) {
  291. count, err = e.Count()
  292. if err != nil {
  293. return
  294. }
  295. err = e.Offset(offset).Limit(limit).Scan(result)
  296. return
  297. }
  298. func (e environmentDo) Scan(result interface{}) (err error) {
  299. return e.DO.Scan(result)
  300. }
  301. func (e environmentDo) Delete(models ...*model.Environment) (result gen.ResultInfo, err error) {
  302. return e.DO.Delete(models)
  303. }
  304. func (e *environmentDo) withDO(do gen.Dao) *environmentDo {
  305. e.DO = *do.(*gen.DO)
  306. return e
  307. }