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